Adapting to Stochastic and Adversarial Losses in Episodic MDPs with Aggregate Bandit Feedback
Shinji Ito, Kevin G. Jamieson, Haipeng Luo, Arnab Maiti, Taira Tsuchiya
Abstract
We study online learning in finite-horizon episodic Markov decision processes (MDPs) under the challenging aggregate bandit feedback model, where the learner observes only the cumulative loss incurred in each episode, rather than individual losses at each state-action pair. While prior work in this setting has focused exclusively on worst-case analysis, we initiate the study of best-of-both-worlds (BOBW) algorithms that achieve low regret in both stochastic and adversarial environments. We propose the first BOBW algorithms for episodic tabular MDPs with aggregate bandit feedback. In the case of known transitions, our algorithms achieve regret in stochastic settings and regret in adversarial ones. Importantly, we also establish matching lower bounds, showing the optimality of our algorithms in this setting. We further extend our approach to unknown-transition settings by incorporating confidence-based techniques. Our results rely on a combination of FTRL over occupancy measures, self-bounding techniques, and new loss estimators inspired by recent advances in online shortest path problems. Along the way, we also provide the first individual-gap-dependent lower bounds and demonstrate near-optimal BOBW algorithms for shortest path problems with bandit feedback.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 8fdad313-a1c2-4abb-96de-616317c43991Cited by top-tier papers2
- Best-of-Both-Worlds for Heavy-Tailed Markov Decision ProcessesYu Chen, Yuhao Liu, Jiatai Huang, Yihan Du et al.ICML 2026 · 1 citation
- Data- and Variance-dependent Regret Bounds for Online Tabular MDPsMingyi Li, Taira Tsuchiya, Kenji YamanishiICML 2026
Related papers
- Simultaneously Learning Stochastic and Adversarial Episodic MDPs with Known TransitionTiancheng Jin, Haipeng LuoNeurIPS 2020 · 62 citations
- The best of both worlds: stochastic and adversarial episodic MDPs with unknown transitionTiancheng Jin, Longbo Huang, Haipeng LuoNeurIPS 2021 · 51 citations
- Learning Adversarial Markov Decision Processes with Bandit Feedback and Unknown TransitionChi Jin, Tiancheng Jin, Haipeng Luo, Suvrit Sra et al.ICML 2020 · 117 citations
- Dynamic Regret of Online Markov Decision ProcessesPeng Zhao, Longfei Li, Zhi-Hua ZhouICML 2022 · 22 citations
- Policy Optimization for CMDPs with Bandit Feedback: Learning Stochastic and Adversarial ConstraintsFrancesco Emanuele Stradi, Anna Lunghi, Matteo Castiglioni, Alberto Marchesi et al.ICML 2025
