arXiv:2409.04971cs.LGcs.AI2024-09被引 1

用隐式重参数化让SAC支持贝塔分布策略,提升连续控制效率。

Soft Actor-Critic with Beta Policy via Implicit Reparameterization Gradients

  • 引入隐式重参数化梯度,扩展SAC对贝塔分布策略的支持
  • 在机器人运动仿真中,贝塔策略性能优于正态策略,接近挤压正态策略
  • 适合需要高样本效率的复杂连续控制任务研究者

深度强化学习虽在复杂任务中取得显著进展,但样本效率低下仍是实际应用的主要障碍。软演员-评论家(SAC)通过结合随机策略优化与离线学习缓解该问题,但其适用性受限于可使用重参数化技巧计算梯度的分布。这一限制排除了贝塔分布等重要分布,而贝塔分布因具有有界支撑,在高维连续控制中已被证明可加速算法收敛。为解决此问题,本文研究使用隐式重参数化技术,该技术可扩展可重参数化分布的范围。我们采用隐式重参数化梯度训练基于贝塔分布策略的SAC,在模拟机器人运动环境中进行测试,并与常见基线方法对比。实验结果表明,贝塔策略是可行替代方案:其性能优于正态策略,且与挤压正态策略相当,后者是SAC的常用选择。代码已开源:https://github.com/lucadellalib/sac-beta。

原文摘要 · Abstract (English)

Recent advances in deep reinforcement learning have achieved impressive results in a wide range of complex tasks, but poor sample efficiency remains a major obstacle to real-world deployment. Soft actor-critic (SAC) mitigates this problem by combining stochastic policy optimization and off-policy learning, but its applicability is restricted to distributions whose gradients can be computed through the reparameterization trick. This limitation excludes several important examples such as the beta distribution, which was shown to improve the convergence rate of actor-critic algorithms in high-dimensional continuous control problems thanks to its bounded support. To address this issue, we investigate the use of implicit reparameterization, a powerful technique that extends the class of reparameterizable distributions. In particular, we use implicit reparameterization gradients to train SAC with the beta policy on simulated robot locomotion environments and compare its performance with common baselines. Experimental results show that the beta policy is a viable alternative, as it outperforms the normal policy and is on par with the squashed normal policy, which is the go-to choice for SAC. The code is available at https://github.com/lucadellalib/sac-beta.

强化学习策略优化贝塔分布SAC

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