Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Policy Optimization Achieves Data-Dependent Regret Bounds in MDPs with Unknown Transitions

About

We study policy optimization for online episodic tabular Markov decision processes with unknown transition kernels, aiming for best-of-both-worlds guarantees together with data-dependent regret bounds. Recent work (Dann et al., 2023; Li et al., 2026) has shown that policy optimization can adapt to both adversarial and stochastic losses with first-order, second-order, and path-length bounds, but only under known transitions, leaving open whether such data-dependent guarantees are achievable by policy optimization when the transition kernel is unknown. We resolve this by developing a new algorithm based on optimistic follow-the-regularized-leader that attains these guarantees under unknown transitions. The key ingredient is a new design of optimistic $Q$-function estimators together with a data-dependent transition bonus that controls estimator bias through the loss-prediction error. Our analysis further identifies an unavoidable transition-dependent complexity term that captures the intrinsic cost of estimating the transition kernel. As a result, we obtain first-order, second-order, and path-length bounds with the transition-dependent complexity term while simultaneously achieving gap-dependent $\mathrm{polylog}(T)$ regret in the stochastic regime.

Mingyi Li, Taira Tsuchiya, Kenji Yamanishi• 2026

Related benchmarks

TaskDatasetResultRank
Regret MinimizationEpisodic Tabular MDPs Adversarial Regime
Regret Upper Bound2
6
Regret MinimizationEpisodic Tabular MDPs Stochastic Regime with Adversarial Corruption
Regret Upper Bound2
5
Showing 2 of 2 rows

Other info

Follow for update