提出自适应扩散策略优化方法,提升机器人操作中扩散策略的训练速度与稳定性。
Adaptive Diffusion Policy Optimization for Robotic Manipulation
- 采用自适应梯度下降法优化扩散策略,实现快速稳定训练。
- 在标准机器人任务上优于或相当其他六种扩散策略基准方法。
- 分析超参数敏感性,为实际应用提供指导,适合机器人控制研究者。
近期研究表明,扩散模型在强化学习中具有巨大潜力,能建模复杂策略、表达高度多模态特性,并高效处理高维连续控制任务。然而,目前关于如何快速且稳定地优化基于扩散的策略(如扩散策略)的研究仍有限。本文提出一种基于Adam的扩散策略优化方法(ADPO),该算法框架结合了强化学习中的自适应梯度下降法,适用于机器人控制任务中对扩散策略的微调。自适应梯度方法在强化学习训练中研究较少,更未在扩散策略中系统应用。我们通过大量实验验证了ADPO在标准机器人控制任务上的有效性,其中包含六种主流扩散策略作为基线对比。结果表明,ADPO在性能上优于或等同于基线方法。此外,我们系统分析了多个超参数在典型机器人任务中的敏感性,为后续实践应用提供参考。视频演示已发布于https://github.com/Timeless-lab/ADPO.git。
原文摘要 · Abstract (English)
Recent studies have shown the great potential of diffusion models in improving reinforcement learning (RL) by modeling complex policies, expressing a high degree of multi-modality, and efficiently handling high-dimensional continuous control tasks. However, there is currently limited research on how to optimize diffusion-based polices (e.g., Diffusion Policy) fast and stably. In this paper, we propose an Adam-based Diffusion Policy Optimization (ADPO), a fast algorithmic framework containing best practices for fine-tuning diffusion-based polices in robotic control tasks using the adaptive gradient descent method in RL. Adaptive gradient method is less studied in training RL, let alone diffusion-based policies. We confirm that ADPO outperforms other diffusion-based RL methods in terms of overall effectiveness for fine-tuning on standard robotic tasks. Concretely, we conduct extensive experiments on standard robotic control tasks to test ADPO, where, particularly, six popular diffusion-based RL methods are provided as benchmark methods. Experimental results show that ADPO acquires better or comparable performance than the baseline methods. Finally, we systematically analyze the sensitivity of multiple hyperparameters in standard robotics tasks, providing guidance for subsequent practical applications. Our video demonstrations are released in https://github.com/Timeless-lab/ADPO.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。