English

Sharp Gap-Dependent Variance-Aware Regret Bounds for Tabular MDPs

Machine Learning 2025-06-10 v1 Machine Learning

Abstract

We consider the gap-dependent regret bounds for episodic MDPs. We show that the Monotonic Value Propagation (MVP) algorithm achieves a variance-aware gap-dependent regret bound of O~((Δh(s,a)>0H2logKVarmaxcΔh(s,a)+Δh(s,a)=0H2VarmaxcΔmin+SAH4(SH))logK),\tilde{O}\left(\left(\sum_{\Delta_h(s,a)>0} \frac{H^2 \log K \land \mathtt{Var}_{\max}^{\text{c}}}{\Delta_h(s,a)} +\sum_{\Delta_h(s,a)=0}\frac{ H^2 \land \mathtt{Var}_{\max}^{\text{c}}}{\Delta_{\mathrm{min}}} + SAH^4 (S \lor H) \right) \log K\right), where HH is the planning horizon, SS is the number of states, AA is the number of actions, and KK is the number of episodes. Here, Δh(s,a)=Vh(a)Qh(s,a)\Delta_h(s,a) =V_h^* (a) - Q_h^* (s, a) represents the suboptimality gap and Δmin:=minΔh(s,a)>0Δh(s,a)\Delta_{\mathrm{min}} := \min_{\Delta_h (s,a) > 0} \Delta_h(s,a). The term Varmaxc\mathtt{Var}_{\max}^{\text{c}} denotes the maximum conditional total variance, calculated as the maximum over all (π,h,s)(\pi, h, s) tuples of the expected total variance under policy π\pi conditioned on trajectories visiting state ss at step hh. Varmaxc\mathtt{Var}_{\max}^{\text{c}} characterizes the maximum randomness encountered when learning any (h,s)(h, s) pair. Our result stems from a novel analysis of the weighted sum of the suboptimality gap and can be potentially adapted for other algorithms. To complement the study, we establish a lower bound of Ω(Δh(s,a)>0H2VarmaxcΔh(s,a)logK),\Omega \left( \sum_{\Delta_h(s,a)>0} \frac{H^2 \land \mathtt{Var}_{\max}^{\text{c}}}{\Delta_h(s,a)}\cdot \log K\right), demonstrating the necessity of dependence on Varmaxc\mathtt{Var}_{\max}^{\text{c}} even when the maximum unconditional total variance (without conditioning on (h,s)(h, s)) approaches zero.

Keywords

Cite

@article{arxiv.2506.06521,
  title  = {Sharp Gap-Dependent Variance-Aware Regret Bounds for Tabular MDPs},
  author = {Shulun Chen and Runlong Zhou and Zihan Zhang and Maryam Fazel and Simon S. Du},
  journal= {arXiv preprint arXiv:2506.06521},
  year   = {2025}
}

Comments

30 pages

R2 v1 2026-07-01T03:04:25.981Z