Reward Machines for Deep RL in Noisy and Uncertain Environments

Andrew C. Li, Zizhao Chen, Toryn Q. Klassen, Pashootan Vaezipoor, Rodrigo Toro Icarte +1 more
2/3/2026

No official code found. Best implementation: andrewli77/reward-machines-noisy-environments (6 stars, MIT, last commit 2024-06-17).

Abstract

Reward Machines provide an automaton-inspired structure for specifying instructions, safety constraints, and other temporally extended reward-worthy behaviour. By exposing the underlying structure of a reward function, they enable the decomposition of an RL task, leading to impressive gains in sample efficiency. Although Reward Machines and similar formal specifications have a rich history of application towards sequential decision-making problems, they critically rely on a ground-truth interpretation of the domain-specific vocabulary that forms the building blocks of the reward function--such ground-truth interpretations are elusive in the real world due in part to partial observability and noisy sensing. In this work, we explore the use of Reward Machines for Deep RL in noisy and uncertain environments. We characterize this problem as a POMDP and propose a suite of RL algorithms that exploit task structure under uncertain interpretation of the domain-specific vocabulary. Through theory and experiments, we expose pitfalls in naive approaches to this problem while simultaneously demonstrating how task structure can be successfully leveraged under noisy interpretations of the vocabulary.

View on arXivView PDF

Code Implementations(1)

MIT

A Deep RL framework for Reward Machines under an imperfect labelling function.

61Shell, PythonJun 4, 20242 years agoMIT

Similar Papers

Cite this paper

@article{li2026reward,
  title  = {Reward Machines for Deep RL in Noisy and Uncertain Environments},
  author = {Andrew C. Li and Zizhao Chen and Toryn Q. Klassen and Pashootan Vaezipoor and Rodrigo Toro Icarte and Sheila A. McIlraith},
  year   = {2026},
  eprint = {2406.00120},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2406.00120},
  url    = {https://arxiv.org/abs/2406.00120},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.00120.svg)](https://codeofpaper.com/paper/2406.00120)

Discussion