About Time: Model-free Reinforcement Learning with Timed Reward Machines

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Roy, Rajarshi, Majumdar, Anirban, Raha, Ritam, Parker, David, Kwiatkowska, Marta
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917485800849408
author Roy, Rajarshi
Majumdar, Anirban
Raha, Ritam
Parker, David
Kwiatkowska, Marta
author_facet Roy, Rajarshi
Majumdar, Anirban
Raha, Ritam
Parker, David
Kwiatkowska, Marta
contents Reward specification plays a central role in reinforcement learning (RL), guiding the agent's behavior. To express non-Markovian rewards, formalisms such as reward machines have been introduced to capture dependencies on histories. However, traditional reward machines lack the ability to model precise timing constraints, limiting their use in time-sensitive applications. In this paper, we propose timed reward machines (TRMs), which are an extension of reward machines that incorporate timing constraints into the reward structure. TRMs enable more expressive specifications with tunable reward logic, for example, imposing costs for delays and granting rewards for timely actions. We study model-free RL frameworks (i.e., tabular Q-learning) for learning optimal policies with TRMs under digital and real-time semantics. Our algorithms integrate the TRM into learning via abstractions of timed automata, and employ counterfactual-imagining heuristics that exploit the structure of the TRM to improve the search. Experimentally, we demonstrate that our algorithm learns policies that achieve high rewards while satisfying the timing constraints specified by the TRM on popular RL benchmarks. Moreover, we conduct comparative studies of performance under different TRM semantics, along with ablations that highlight the benefits of counterfactual-imagining.
format Preprint
id arxiv_https___arxiv_org_abs_2512_17637
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle About Time: Model-free Reinforcement Learning with Timed Reward Machines
Roy, Rajarshi
Majumdar, Anirban
Raha, Ritam
Parker, David
Kwiatkowska, Marta
Artificial Intelligence
Formal Languages and Automata Theory
Logic in Computer Science
Reward specification plays a central role in reinforcement learning (RL), guiding the agent's behavior. To express non-Markovian rewards, formalisms such as reward machines have been introduced to capture dependencies on histories. However, traditional reward machines lack the ability to model precise timing constraints, limiting their use in time-sensitive applications. In this paper, we propose timed reward machines (TRMs), which are an extension of reward machines that incorporate timing constraints into the reward structure. TRMs enable more expressive specifications with tunable reward logic, for example, imposing costs for delays and granting rewards for timely actions. We study model-free RL frameworks (i.e., tabular Q-learning) for learning optimal policies with TRMs under digital and real-time semantics. Our algorithms integrate the TRM into learning via abstractions of timed automata, and employ counterfactual-imagining heuristics that exploit the structure of the TRM to improve the search. Experimentally, we demonstrate that our algorithm learns policies that achieve high rewards while satisfying the timing constraints specified by the TRM on popular RL benchmarks. Moreover, we conduct comparative studies of performance under different TRM semantics, along with ablations that highlight the benefits of counterfactual-imagining.
title About Time: Model-free Reinforcement Learning with Timed Reward Machines
topic Artificial Intelligence
Formal Languages and Automata Theory
Logic in Computer Science
url https://arxiv.org/abs/2512.17637