English

A Nearly Optimal and Low-Switching Algorithm for Reinforcement Learning with General Function Approximation

Machine Learning 2025-10-06 v2 Optimization and Control Machine Learning

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 O~(dHK)\tilde{O}(d\sqrt{HK}) when KK is sufficiently large and near-optimal policy switching cost of O~(dH)\tilde{O}(dH), with dd being the eluder dimension of the function class, HH being the planning horizon, and KK being the number of episodes. Our work sheds light on designing provably sample-efficient and deployment-efficient Q-learning with nonlinear function approximation.

Keywords

Cite

@article{arxiv.2311.15238,
  title  = {A Nearly Optimal and Low-Switching Algorithm for Reinforcement Learning with General Function Approximation},
  author = {Heyang Zhao and Jiafan He and Quanquan Gu},
  journal= {arXiv preprint arXiv:2311.15238},
  year   = {2025}
}

Comments

46 pages, 1 table

R2 v1 2026-06-28T13:31:41.447Z