Provably Efficient Exploration for Reinforcement Learning Using Unsupervised Learning
Fei Feng, Ruosong Wang, Wotao Yin, Simon Du, Lin Yang
Spotlight presentation: Orals & Spotlights Track 04: Reinforcement Learning
on 2020-12-07T19:50:00-08:00 - 2020-12-07T20:00:00-08:00
on 2020-12-07T19:50:00-08:00 - 2020-12-07T20:00:00-08:00
Toggle Abstract Paper (in Proceedings / .pdf)
Abstract: Motivated by the prevailing paradigm of using unsupervised learning for efficient exploration in reinforcement learning (RL) problems [tang2017exploration,bellemare2016unifying], we investigate when this paradigm is provably efficient. We study episodic Markov decision processes with rich observations generated from a small number of latent states. We present a general algorithmic framework that is built upon two components: an unsupervised learning algorithm and a no-regret tabular RL algorithm. Theoretically, we prove that as long as the unsupervised learning algorithm enjoys a polynomial sample complexity guarantee, we can find a near-optimal policy with sample complexity polynomial in the number of latent states, which is significantly smaller than the number of observations. Empirically, we instantiate our framework on a class of hard exploration problems to demonstrate the practicality of our theory.