用梯度追踪替代硬更新,让DQN自动调节目标网络
Deep Q-Learning with Gradient Target Tracking
- 用梯度下降实现目标网络的连续更新,取代固定周期的硬拷贝
- 在多个游戏上性能优于标准DQN,且无需调参更新周期
- 理论证明可收敛,适合追求稳定训练的强化学习研究者
本文提出一种基于梯度目标追踪的Q-learning框架,以连续、结构化的梯度更新替代传统DQN中的硬更新机制。标准DQN通过周期性复制在线网络权重来维护目标网络,但其更新周期需人工精细调优。为此,本文提出两种梯度驱动方法:AGT2-DQN与SGT2-DQN,利用梯度下降实现目标网络的平滑更新,彻底消除对人工调参的依赖。理论分析证明其在表格设定下的收敛性。实验表明,该方法在多个Atari游戏上显著优于标准DQN基线,验证了梯度目标更新作为有效替代方案的可行性。
原文摘要 · Abstract (English)
This paper introduces Q-learning with gradient target tracking, a novel reinforcement learning framework that provides a learned continuous target update mechanism as an alternative to the conventional hard update paradigm. In the standard deep Q-network (DQN), the target network is a copy of the online network's weights, held fixed for a number of iterations before being periodically replaced via a hard update. While this stabilizes training by providing consistent targets, it introduces a new challenge: the hard update period must be carefully tuned to achieve optimal performance. To address this issue, we propose two gradient-based target update methods: DQN with asymmetric gradient target tracking (AGT2-DQN) and DQN with symmetric gradient target tracking (SGT2-DQN). These methods replace the conventional hard target updates with continuous and structured updates using gradient descent, which effectively eliminates the need for manual tuning. We provide a theoretical analysis proving the convergence of these methods in tabular settings. Additionally, empirical evaluations demonstrate their advantages over standard DQN baselines, which suggest that gradient-based target updates can serve as an effective alternative to conventional target update mechanisms in Q-learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。