arXiv:2412.02291cs.LGcs.AI2024-12被引 10

用相对论原理设计新优化器,让强化学习更稳定高效。

Conformal Symplectic Optimization for Stable Reinforcement Learning

  • 将神经网络训练建模为保角辛系统,引入相对论动能控制参数更新速度。
  • 在12个环境中超越9种基线优化器,Atari游戏性能最高提升155.1%。
  • 适合追求训练稳定性的强化学习研究者,尤其适用于非凸高维优化场景。

深度强化学习训练面临高度不稳定、非凸且随机的优化挑战。为此,我们提出一种受物理启发的优化算法——相对论自适应梯度下降(RAD),显著提升长期训练稳定性。通过将神经网络训练视为保角哈密顿系统的演化,我们构建了一个通用框架,将保角辛积分器的长期稳定性转移至迭代更新规则中,其中动能选择决定优化算法的动力学特性。采用相对论动能使参数更新速度受限于有限值,有效抑制异常梯度影响。此外,RAD将神经网络优化建模为多粒子系统,每个可训练参数作为独立粒子,拥有自适应学习率。我们在一般非凸设定下证明了RAD的次线性收敛性,较小的梯度方差和较大的批量有助于更紧的收敛界。值得注意的是,当速度系数为1且辛因子为小正数时,RAD退化为经典的自适应矩估计(ADAM)算法。实验表明,在12个环境(包括标准基准与挑战性场景)中,结合五种强化学习算法,RAD优于九种基线优化器,尤其在Atari游戏中性能较ADAM最高提升155.1%,充分验证其在稳定与加速强化学习训练方面的有效性。

原文摘要 · Abstract (English)

Training deep reinforcement learning (RL) agents necessitates overcoming the highly unstable nonconvex stochastic optimization inherent in the trial-and-error mechanism. To tackle this challenge, we propose a physics-inspired optimization algorithm called relativistic adaptive gradient descent (RAD), which enhances long-term training stability. By conceptualizing neural network (NN) training as the evolution of a conformal Hamiltonian system, we present a universal framework for transferring long-term stability from conformal symplectic integrators to iterative NN updating rules, where the choice of kinetic energy governs the dynamical properties of resulting optimization algorithms. By utilizing relativistic kinetic energy, RAD incorporates principles from special relativity and limits parameter updates below a finite speed, effectively mitigating abnormal gradient influences. Additionally, RAD models NN optimization as the evolution of a multi-particle system where each trainable parameter acts as an independent particle with an individual adaptive learning rate. We prove RAD's sublinear convergence under general nonconvex settings, where smaller gradient variance and larger batch sizes contribute to tighter convergence. Notably, RAD degrades to the well-known adaptive moment estimation (ADAM) algorithm when its speed coefficient is chosen as one and symplectic factor as a small positive value. Experimental results show RAD outperforming nine baseline optimizers with five RL algorithms across twelve environments, including standard benchmarks and challenging scenarios. Notably, RAD achieves up to a 155.1% performance improvement over ADAM in Atari games, showcasing its efficacy in stabilizing and accelerating RL training.

强化学习优化算法稳定性深度学习

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