A Nearly Optimal and Low-Switching Algorithm for Reinforcement Learning with General Function Approximation
Heyang Zhao, Jiafan He, Quanquan Gu
Abstract
The exploration-exploitation dilemma has been a central challenge in reinforcement learning (RL) with complex model classes. In this paper, we propose a new algorithm, Monotonic Q-Learning with Upper Confidence Bound (MQL-UCB) for RL with general function approximation. Our key algorithmic design includes (1) a general deterministic policy-switching strategy that achieves low switching cost, (2) a monotonic value function structure with carefully controlled function class complexity, and (3) a variance-weighted regression scheme that exploits historical trajectories with high data efficiency. MQL-UCB achieves minimax optimal regret of when is sufficiently large and near-optimal policy switching cost of , with being the eluder dimension of the function class, being the planning horizon, and being the number of episodes. Our work sheds light on designing provably sample-efficient and deployment-efficient Q-learning with nonlinear function approximation.
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.
Cited by top-tier papers10
- Sharp Analysis for KL-Regularized Contextual Bandits and RLHFHeyang Zhao, Chenlu Ye, Quanquan Gu, Tong ZhangNeurIPS 2025 · 33 citations
- Towards Robust Model-Based Reinforcement Learning Against Adversarial CorruptionChenlu Ye, Jiafan He, Quanquan Gu, Tong ZhangICML 2024 · 10 citations
- Uncertainty-Aware Reward-Free Exploration with General Function ApproximationJunkai Zhang, Weitong Zhang, Dongruo Zhou, Quanquan GuICML 2024 · 7 citations
- Near-Optimal Reinforcement Learning with Self-Play under Adaptivity ConstraintsDan Qiao, Yu-Xiang WangICML 2024 · 5 citations
- Breaking the Total Variance Barrier: Sharp Sample Complexity for Linear Heteroscedastic Bandits with Fixed Action SetHeyang Zhao, Tianyuan Jin, Weixin Wang, Vincent Y. F. Tan et al.ICLR 2026 · 1 citation
Builds on2
Related papers
- Reinforcement Learning with General Value Function Approximation: Provably Efficient Approach via Bounded Eluder DimensionRuosong Wang, Ruslan Salakhutdinov, Lin F. YangNeurIPS 2020 · 168 citations
- Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision ProcessesJiafan He, Heyang Zhao, Dongruo Zhou, Quanquan GuICML 2023 · 68 citations
- Near-Optimal Deployment Efficiency in Reward-Free Reinforcement Learning with Linear Function ApproximationDan Qiao, Yu-Xiang WangICLR 2023
- Model-Based Reinforcement Learning with Multinomial Logistic Function ApproximationTaehyun Hwang, Min-hwan OhAAAI 2023 · 13 citations
- Reinforcement Learning with Logarithmic Regret and Policy SwitchesGrigoris Velegkas, Zhuoran Yang, Amin KarbasiNeurIPS 2022 · 7 citations
