arXiv:2501.19133cs.LGcs.AI2025-01

通过层间去相关加速深度强化学习,提升训练效率

Decorrelated Soft Actor-Critic for Efficient Deep Reinforcement Learning

  • 在SAC算法中引入网络级去相关机制,优化信用分配
  • 7个Atari游戏中5次更快收敛,2次奖励提升,耗时减少约50%
  • 适合追求高效训练的深度强化学习研究者和应用开发者

深度强化学习在处理高维数据时,信用分配的有效性受深度神经网络表征学习的影响,进而影响算法的样本效率。输入去相关已被证明能加速神经网络优化,并有效提升深度强化学习的表征能力。本文提出一种基于去相关反向传播的在线去相关方法,将去相关过程无缝集成到强化学习训练流程中。在每一层添加去相关矩阵,通过独立的去相关学习规则最小化全网去相关损失,与常规强化学习损失并行优化。该方法与软演员-评论家(SAC)结合,称为去相关软演员-评论家(DSAC)。在Atari 100k基准上的实验表明,相较于标准SAC,在7个游戏中有5个训练速度更快,2个游戏奖励性能提升,平均耗时减少约50%,其余游戏性能保持稳定。结果验证了网络级去相关对提升深度强化学习样本效率的积极作用。

原文摘要 · Abstract (English)

The effectiveness of credit assignment in reinforcement learning (RL) when dealing with high-dimensional data is influenced by the success of representation learning via deep neural networks, and has implications for the sample efficiency of deep RL algorithms. Input decorrelation has been previously introduced as a method to speed up optimization in neural networks, and has proven impactful in both efficient deep learning and as a method for effective representation learning for deep RL algorithms. We propose a novel approach to online decorrelation in deep RL based on the decorrelated backpropagation algorithm that seamlessly integrates the decorrelation process into the RL training pipeline. Decorrelation matrices are added to each layer, which are updated using a separate decorrelation learning rule that minimizes the total decorrelation loss across all layers, in parallel to minimizing the usual RL loss. We used our approach in combination with the soft actor-critic (SAC) method, which we refer to as decorrelated soft actor-critic (DSAC). Experiments on the Atari 100k benchmark with DSAC shows, compared to the regular SAC baseline, faster training in five out of the seven games tested and improved reward performance in two games with around 50% reduction in wall-clock time, while maintaining performance levels on the other games. These results demonstrate the positive impact of network-wide decorrelation in deep RL for speeding up its sample efficiency through more effective credit assignment.

强化学习去相关SAC样本效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。