Average Reward Reinforcement Learning for Omega-Regular and Mean-Payoff Objectives
Abstract
Recent advances in reinforcement learning (RL) have renewed interest in reward design for shaping agent behavior, but manually crafting reward functions is tedious and error-prone. A principled alternative is to specify behavioral requirements in a formal, unambiguous language and automatically compile them into learning objectives. -regular languages are a natural fit, given their role in formal verification and synthesis. However, most existing -regular RL approaches operate in an episodic, discounted setting with periodic resets, which is misaligned with -regular semantics over infinite traces. For continuing tasks, where the agent interacts with the environment over a single uninterrupted lifetime, the average-reward criterion is more appropriate. We focus on absolute liveness specifications, a subclass of -regular languages that cannot be violated by any finite prefix and thus aligns naturally with continuing interaction. We present the first model-free RL framework that translates absolute liveness specifications into average-reward objectives and enables learning in unknown communicating Markov decision processes (MDPs) without episodic resetting. We also introduce a reward structure for lexicographic multi-objective optimization: among policies that maximize the satisfaction probability of an absolute liveness specification, the agent maximizes an external average-reward objective. Our method guarantees convergence in unknown communicating MDPs and supports on-the-fly reductions that do not require full environment knowledge, enabling model-free learning. Experiments across several benchmarks show that the continuing, average-reward approach outperforms competing discount-based methods.
Cite
@article{arxiv.2505.15693,
title = {Average Reward Reinforcement Learning for Omega-Regular and Mean-Payoff Objectives},
author = {Milad Kazemi and Mateo Perez and Fabio Somenzi and Sadegh Soudjani and Ashutosh Trivedi and Alvaro Velasquez},
journal= {arXiv preprint arXiv:2505.15693},
year = {2026}
}
Comments
29 pages