用模型控制+强化学习,让机器人零样本完成毫米级插拔任务
Integrating Model-based Control and RL for Sim2Real Transfer of Tight Insertion Policies
- 结合模型控制与残差强化学习,提升插拔精度
- 仿真训练后直接部署真实机器人,无需调参,成功率超90%
- 适合对装配精度要求高的工业场景,如精密零件组装
在小于1毫米公差下的物体插入是一项重要但极具挑战性的装配任务,微小误差即可能导致不良接触。现有方法多依赖强化学习(RL),通常需精心设计密集奖励函数。本文提出一种融合模型基控制与强化学习的策略,实现高精度插入。整个策略仅在仿真中训练,零样本迁移到真实系统。首先利用基于势场的控制器,在完全可观测条件下生成模型基插入策略;再通过仅接收稀疏目标达成奖励的残差强化学习进行优化,并采用观测噪声和动作幅度的课程学习方案训练。两个模块均以插头与插座的SE(3)位姿为输入,输出插头的位姿变换,由机械臂执行。集成策略在真实系统中部署时仅需视觉位姿追踪器,无需额外训练。在多种对象与条件下进行仿真与真实验证,结果表明该方法优于当前主流的基于强化学习的方法及早期混合策略。消融实验验证了各组件的有效性。
原文摘要 · Abstract (English)
Object insertion under tight tolerances ($< \hspace{-.02in} 1mm$) is an important but challenging assembly task as even small errors can result in undesirable contacts. Recent efforts focused on Reinforcement Learning (RL), which often depends on careful definition of dense reward functions. This work proposes an effective strategy for such tasks that integrates traditional model-based control with RL to achieve improved insertion accuracy. The policy is trained exclusively in simulation and is zero-shot transferred to the real system. It employs a potential field-based controller to acquire a model-based policy for inserting a plug into a socket given full observability in simulation. This policy is then integrated with residual RL, which is trained in simulation given only a sparse, goal-reaching reward. A curriculum scheme over observation noise and action magnitude is used for training the residual RL policy. Both policy components use as input the SE(3) poses of both the plug and the socket and return the plug's SE(3) pose transform, which is executed by a robotic arm using a controller. The integrated policy is deployed on the real system without further training or fine-tuning, given a visual SE(3) object tracker. The proposed solution and alternatives are evaluated across a variety of objects and conditions in simulation and reality. The proposed approach outperforms recent RL-based methods in this domain and prior efforts with hybrid policies. Ablations highlight the impact of each component of the approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。