通过交替融合模仿与强化学习,提升机器人策略微调的稳定性和效率。
IN-RIL: Interleaved Reinforcement and Imitation Learning for Policy Fine-Tuning
- 交替执行强化学习与模仿学习更新,结合两者优势。
- 在Robomimic Transport任务中成功率从12%提升至88%,样本效率提高6.3倍。
- 适用于各类强化学习算法,尤其适合稀疏奖励和长时序任务。
模仿学习(IL)和强化学习(RL)在机器人策略学习中各有优势:IL能从示范数据中稳定学习,而RL通过探索促进泛化。现有方法通常先用IL预训练,再以RL微调,但后者常因不稳定和样本效率低而受限。本文提出IN-RIL(Interleaved Reinforcement and Imitation Learning),在多个RL更新后周期性注入IL更新,从而在全程微调中受益于IL的稳定性与专家数据的引导。针对两者优化目标差异,设计梯度分离机制,将可能冲突的梯度投影至正交子空间,避免破坏性干扰。在3个基准平台(FurnitureBench、OpenAI Gym、Robomimic)上的14项机器人操控与行走任务实验表明,IN-RIL显著提升样本效率,并缓解长/短时程任务中稀疏或密集奖励下的性能崩溃。该方法作为通用插件,可兼容多种先进RL算法,在Robomimic Transport任务中成功率从12%提升至88%,样本效率提升6.3倍。
原文摘要 · Abstract (English)
Imitation learning (IL) and reinforcement learning (RL) each offer distinct advantages for robotics policy learning: IL provides stable learning from demonstrations, and RL promotes generalization through exploration. While existing robot learning approaches using IL-based pre-training followed by RL-based fine-tuning are promising, this two-step learning paradigm often suffers from instability and poor sample efficiency during the RL fine-tuning phase. In this work, we introduce IN-RIL, INterleaved Reinforcement learning and Imitation Learning, for policy fine-tuning, which periodically injects IL updates after multiple RL updates and hence can benefit from the stability of IL and the guidance of expert data for more efficient exploration throughout the entire fine-tuning process. Since IL and RL involve different optimization objectives, we develop gradient separation mechanisms to prevent destructive interference during \ABBR fine-tuning, by separating possibly conflicting gradient updates in orthogonal subspaces. Furthermore, we conduct rigorous analysis, and our findings shed light on why interleaving IL with RL stabilizes learning and improves sample-efficiency. Extensive experiments on 14 robot manipulation and locomotion tasks across 3 benchmarks, including FurnitureBench, OpenAI Gym, and Robomimic, demonstrate that \ABBR can significantly improve sample efficiency and mitigate performance collapse during online finetuning in both long- and short-horizon tasks with either sparse or dense rewards. IN-RIL, as a general plug-in compatible with various state-of-the-art RL algorithms, can significantly improve RL fine-tuning, e.g., from 12\% to 88\% with 6.3x improvement in the success rate on Robomimic Transport. Project page: https://github.com/ucd-dare/IN-RIL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。