287
v1v2 (latest)

Fine-Grained Gap-Dependent Bounds for Tabular MDPs via Adaptive Multi-Step Bootstrap

Annual Conference Computational Learning Theory (COLT), 2021
Abstract

This paper presents a new model-free algorithm for episodic finite-horizon Markov Decision Processes (MDP), Adaptive Multi-step Bootstrap (AMB), which enjoys a stronger gap-dependent regret bound. The first innovation is to estimate the optimal QQ-function by combining an optimistic bootstrap with an adaptive multi-step Monte Carlo rollout. The second innovation is to select the action with the largest confidence interval length among admissible actions that are not dominated by any other actions. We show when each state has a unique optimal action, AMB achieves a gap-dependent regret bound that only scales with the sum of the inverse of the sub-optimality gaps. In contrast, Simchowitz and Jamieson (2019) showed all upper-confidence-bound (UCB) algorithms suffer an additional Ω(SΔmin)\Omega\left(\frac{S}{\Delta_{min}}\right) regret due to over-exploration where Δmin\Delta_{min} is the minimum sub-optimality gap and SS is the number of states. We further show that for general MDPs, AMB suffers an additional ZmulΔmin\frac{|Z_{mul}|}{\Delta_{min}} regret, where ZmulZ_{mul} is the set of state-action pairs (s,a)(s,a)'s satisfying aa is a non-unique optimal action for ss. We complement our upper bound with a lower bound showing the dependency on ZmulΔmin\frac{|Z_{mul}|}{\Delta_{min}} is unavoidable for any consistent algorithm. This lower bound also implies a separation between reinforcement learning and contextual bandits.

View on arXiv
Comments on this paper