English

From Relative Entropy to Minimax: A Unified Framework for Coverage in MDPs

Machine Learning 2026-01-21 v1

Abstract

Targeted and deliberate exploration of state--action pairs is essential in reward-free Markov Decision Problems (MDPs). More precisely, different state-action pairs exhibit different degree of importance or difficulty which must be actively and explicitly built into a controlled exploration strategy. To this end, we propose a weighted and parameterized family of concave coverage objectives, denoted by UρU_\rho, defined directly over state--action occupancy measures. This family unifies several widely studied objectives within a single framework, including divergence-based marginal matching, weighted average coverage, and worst-case (minimax) coverage. While the concavity of UρU_\rho captures the diminishing return associated with over-exploration, the simple closed form of the gradient of UρU_\rho enables an explicit control to prioritize under-explored state--action pairs. Leveraging this structure, we develop a gradient-based algorithm that actively steers the induced occupancy toward a desired coverage pattern. Moreover, we show that as ρ\rho increases, the resulting exploration strategy increasingly emphasizes the least-explored state--action pairs, recovering worst-case coverage behavior in the limit.

Keywords

Cite

@article{arxiv.2601.11890,
  title  = {From Relative Entropy to Minimax: A Unified Framework for Coverage in MDPs},
  author = {Xihe Gu and Urbashi Mitra and Tara Javidi},
  journal= {arXiv preprint arXiv:2601.11890},
  year   = {2026}
}