arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.12817cs.LGstat.ML
收藏

使用随机递归梯度的深度Q学习的方差缩减

Variance Reduction for Deep Q-Learning using Stochastic Recursive Gradient

Gaoling School of Artificial Intelligence, Renmin University of China(中国人民大学高瓴人工智能学院);Baidu Inc.(百度公司);Huawei Technologies(华为技术有限公司)

Haonan Jia, Xiao Zhang, Jun Xu, Wei Zeng, Hao Jiang, Xiaohui Yan, Ji-Rong Wen

摘要:

深度Q-学习算法经常遭受具有过大方差的差的梯度估计,导致不稳定的训练和差的采样效率。已应用SVRG等随机方差缩减梯度方法来降低估计方差(Zhao等人,2019)。然而,由于强化学习的在线实例生成特性,直接将SVRG应用于深度Q学习面临着锚点估计不准确的问题,这极大地限制了SVRG的潜力。为了解决这个问题,并受到递归梯度方差缩减算法SARAH(Nguyen et al. 2017)的启发,本文提出引入递归框架来更新深度Q学习中的随机梯度估计,实现一种名为SRG-DQN的新算法。与基于SVRG的算法不同,SRG-DQN设计了随机梯度估计的递归更新。参数更新是利用过去的随机梯度信息沿着累积方向进行的,因此可以摆脱作为锚的全梯度的估计。此外,SRG-DQN涉及Adam过程以进一步加速训练过程。理论分析和著名的强化学习任务的实验结果表明,所提出的SRG-DQN算法的效率和有效性。

英文摘要:

Deep Q-learning algorithms often suffer from poor gradient estimations with an excessive variance, resulting in unstable training and poor sampling efficiency. Stochastic variance-reduced gradient methods such as SVRG have been applied to reduce the estimation variance (Zhao et al. 2019). However, due to the online instance generation nature of reinforcement learning, directly applying SVRG to deep Q-learning is facing the problem of the inaccurate estimation of the anchor points, which dramatically limits the potentials of SVRG. To address this issue and inspired by the recursive gradient variance reduction algorithm SARAH (Nguyen et al. 2017), this paper proposes to introduce the recursive framework for updating the stochastic gradient estimates in deep Q-learning, achieving a novel algorithm called SRG-DQN. Unlike the SVRG-based algorithms, SRG-DQN designs a recursive update of the stochastic gradient estimate. The parameter update is along an accumulated direction using the past stochastic gradient information, and therefore can get rid of the estimation of the full gradients as the anchors. Additionally, SRG-DQN involves the Adam process for further accelerating the training process. Theoretical analysis and the experimental results on well-known reinforcement learning tasks demonstrate the efficiency and effectiveness of the proposed SRG-DQN algorithm.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑