Breaking the Computational Barrier: Provably Efficient Actor–Critic for Low-Rank MDPs
Ruiquan Huang, Donghao Li, Yingbin LIANG, Jing Yang
Abstract
Reinforcement learning (RL) is a fundamental framework for sequential decision-making, in which an agent learns an optimal policy through interactions with an unknown environment. In settings with function approximation, many existing RL algorithms achieve favorable sample complexity, but often rely on computationally intractable oracles. In this paper, we use supervised learning as a computational proxy to establish a clear hierarchy of commonly adopted RL oracles under low-rank Markov Decision Processes (MDPs). This hierarchy shows that policy evaluation is the most computationally efficient oracle, provided that supervised learning can be efficiently solved. Motivated by this observation, we propose a novel optimistic actor-critic algorithm that relies solely on the policy evaluation oracle. We prove that our algorithm outperforms the existing sample complexity guarantees for low-rank MDPs while avoiding computationally expensive planning or optimization oracles commonly assumed in prior works. We further extend our theoretical results to approximately low-rank MDPs and demonstrate that this setting captures a broad class of realworld environments. Finally, we validate our theoretical results with experiments on several standard Gym environments.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext fe32ca4c-230b-4de9-82a9-352f291cc778Builds on14
- FLAMBE: Structural Complexity and Representation Learning of Low Rank MDPsAlekh Agarwal, Sham M. Kakade, Akshay Krishnamurthy, Wen SunNeurIPS 2020 · 271 citations
- Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient AlgorithmsChi Jin, Qinghua Liu, Sobhan MiryoosefiNeurIPS 2021 · 264 citations
- Bilinear Classes: A Structural Framework for Provable Generalization in RLSimon S. Du, Sham M. Kakade, Jason D. Lee, Shachar Lovett et al.ICML 2021 · 207 citations
- Representation Learning for Online and Offline RL in Low-rank MDPsMasatoshi Uehara, Xuezhou Zhang, Wen SunICLR 2022 · 138 citations
- On Reward-Free Reinforcement Learning with Linear Function ApproximationRuosong Wang, Simon S. Du, Lin F. Yang, Ruslan SalakhutdinovNeurIPS 2020 · 121 citations
Related papers
- Ranking Policy GradientKaixiang Lin, Jiayu ZhouICLR 2020 · 8 citations
- Quantum algorithms for reinforcement learning with a generative modelDaochen Wang, Aarthi Sundaram, Robin Kothari, Ashish Kapoor et al.ICML 2021 · 38 citations
- PC-MLP: Model-based Reinforcement Learning with Policy Cover Guided ExplorationYuda Song, Wen SunICML 2021 · 23 citations
- Agnostic Reinforcement Learning with Low-Rank MDPs and Rich ObservationsAyush Sekhari, Christoph Dann, Mehryar Mohri, Yishay Mansour et al.NeurIPS 2021 · 15 citations
- Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RLAndrew Wagenmaker, Kevin Huang, Liyiming Ke, Kevin Jamieson et al.NeurIPS 2024 · 45 citations
