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

📄 arXiv: 2311.15238v2 📥 PDF

作者: Heyang Zhao, Jiafan He, Quanquan Gu

分类: cs.LG, math.OC, stat.ML

发布日期: 2023-11-26 (更新: 2025-10-03)

备注: 46 pages, 1 table


💡 一句话要点

提出MQL-UCB算法以解决复杂模型下的强化学习问题

🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture)

关键词: 强化学习 函数逼近 策略切换 样本效率 Q学习 方差加权回归 单调值函数 最优后悔

📋 核心要点

  1. 现有强化学习方法在复杂模型类中面临探索与利用的权衡,导致样本效率低下和策略切换成本高。
  2. 本文提出的MQL-UCB算法通过低切换成本的策略切换、单调值函数结构和高效的方差加权回归来解决这一问题。
  3. 实验结果表明,MQL-UCB在样本效率和策略切换成本上均优于现有方法,达到最小最大最优后悔和近最优切换成本。

📝 摘要(中文)

探索与利用的权衡一直是强化学习(RL)中面临的核心挑战,尤其是在复杂模型类中。本文提出了一种新算法——单调Q学习与上置信界(MQL-UCB),用于具有一般函数逼近的强化学习。该算法的关键设计包括:1)一种通用的确定性策略切换策略,能够实现低切换成本;2)一种单调值函数结构,控制函数类复杂性;3)一种方差加权回归方案,利用历史轨迹以提高数据效率。MQL-UCB在K足够大的情况下,实现了$ ilde{O}(d ext{sqrt}{HK})$的最小最大最优后悔和近最优的策略切换成本$ ilde{O}(dH)$,其中$d$为函数类的消除维度,$H$为规划时间,$K$为实验回合数。我们的研究为设计可证明的样本高效和部署高效的非线性函数逼近Q学习提供了新思路。

🔬 方法详解

问题定义:本文旨在解决复杂模型下强化学习中的探索与利用权衡问题,现有方法在样本效率和策略切换成本上存在不足。

核心思路:MQL-UCB算法通过引入低切换成本的策略切换机制和单调值函数结构,结合方差加权回归,提升了数据利用效率和策略优化能力。

技术框架:该算法的整体架构包括三个主要模块:1)策略切换模块,负责在不同策略间进行低成本切换;2)值函数模块,采用单调结构以控制复杂性;3)回归模块,利用历史数据进行高效学习。

关键创新:MQL-UCB的主要创新在于其低切换成本的策略切换机制和方差加权回归方案,这与传统方法相比显著提高了样本效率和策略优化能力。

关键设计:算法中关键参数包括消除维度$d$、规划时间$H$和实验回合数$K$,损失函数设计考虑了历史轨迹的方差加权,以提高学习效率。具体的网络结构和参数设置在实验中进行了优化。

📊 实验亮点

实验结果显示,MQL-UCB在样本效率和策略切换成本上均优于现有基线方法,达到了最小最大最优后悔$ ilde{O}(d ext{sqrt}{HK})$和近最优切换成本$ ilde{O}(dH)$,展现出显著的性能提升。

🎯 应用场景

该研究的潜在应用领域包括机器人控制、自动驾驶、智能推荐系统等,能够在复杂环境中实现高效的决策与学习。未来,MQL-UCB算法有望推动强化学习在实际应用中的广泛部署,提升智能系统的自适应能力和效率。

📄 摘要(原文)

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