Minimax-Optimal Learning for Robust Average-Reward MDPs.

Yuepeng Yang, Yuxin Chen, Yuejie Chi· August 10, 2026 View original

Key takeaways

  • Learning robust policies in average-reward MDPs has specific sample complexity requirements.
  • A perturbation scale differentiates high- and low-tolerance regimes for robustness.
  • Minimax-optimal learning rates are achieved using reduction-based plug-in procedures.
  • The sample complexity includes both nominal and robustness-specific terms.

Who benefits

RoboticsAutonomous SystemsFinanceLogisticsManufacturing

Summary

This paper determines the necessary and sufficient sample complexity for learning an epsilon-optimal robust policy in average-reward Markov Decision Processes (MDPs) under model uncertainty. It identifies a perturbation scale separating high and low-tolerance regimes and achieves minimax rates using reduction-based plug-in procedures.

Decision-making systems often operate under uncertainty about the exact dynamics of their environment. Distributionally robust Markov Decision Processes (MDPs) provide a framework to handle this by considering a set of possible transition kernels rather than a single nominal one. This research focuses on the average-reward criterion, a common objective in continuous decision-making, and investigates how many samples are truly needed to learn a policy that is robustly optimal. The study establishes precise upper and lower bounds for the minimax total sample complexity required to achieve an epsilon-optimal robust policy. A key finding is the identification of a "perturbation scale" that distinguishes between high- and low-tolerance regimes, where the sample complexity behaves differently. The total sample complexity is shown to comprise a term similar to nominal MDP results and an additional robustness-specific term that emerges in the low-tolerance regime. These optimal rates are achieved through novel reduction-based plug-in procedures. These procedures intelligently select between nominal and robust reductions, along with their discount factors, either by using known span parameters or by calibrating these choices directly from data. This work provides fundamental theoretical insights into the sample efficiency of learning robust policies in uncertain environments.

Why it matters

For professionals designing AI agents or control systems that must operate reliably under model uncertainty, this research offers theoretical guarantees and efficient learning strategies for robust decision-making.

How to implement this in your domain

  1. 1Assess the level of model uncertainty in your sequential decision-making applications.
  2. 2Consider using distributionally robust MDPs for applications requiring high reliability under uncertainty.
  3. 3Explore implementing reduction-based plug-in procedures for learning robust policies.
  4. 4Factor in the identified sample complexity requirements when planning data collection for robust AI systems.

Original post by Yuepeng Yang, Yuxin Chen, Yuejie Chi

"arXiv:2608.06545v1 Announce Type: new Abstract: Distributionally robust Markov decision processes provide a principled framework for sequential decision making under model uncertainty. We study how many samples are necessary and sufficient to learn an $\varepsilon$-optimal robust…"

View on X

Originally posted by Yuepeng Yang, Yuxin Chen, Yuejie Chi on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses