English

Is Q-learning Provably Efficient?

Machine Learning 2018-07-11 v1 Artificial Intelligence Optimization and Control Machine Learning

Abstract

Model-free reinforcement learning (RL) algorithms, such as Q-learning, directly parameterize and update value functions or policies without explicitly modeling the environment. They are typically simpler, more flexible to use, and thus more prevalent in modern deep RL than model-based approaches. However, empirical work has suggested that model-free algorithms may require more samples to learn [Deisenroth and Rasmussen 2011, Schulman et al. 2015]. The theoretical question of "whether model-free algorithms can be made sample efficient" is one of the most fundamental questions in RL, and remains unsolved even in the basic scenario with finitely many states and actions. We prove that, in an episodic MDP setting, Q-learning with UCB exploration achieves regret O~(H3SAT)\tilde{O}(\sqrt{H^3 SAT}), where SS and AA are the numbers of states and actions, HH is the number of steps per episode, and TT is the total number of steps. This sample efficiency matches the optimal regret that can be achieved by any model-based approach, up to a single H\sqrt{H} factor. To the best of our knowledge, this is the first analysis in the model-free setting that establishes T\sqrt{T} regret without requiring access to a "simulator."

Keywords

Cite

@article{arxiv.1807.03765,
  title  = {Is Q-learning Provably Efficient?},
  author = {Chi Jin and Zeyuan Allen-Zhu and Sebastien Bubeck and Michael I. Jordan},
  journal= {arXiv preprint arXiv:1807.03765},
  year   = {2018}
}

Comments

Best paper in ICML 2018 workshop "Exploration in RL"