arXiv:2409.07914cs.ROcs.AI2024-09CoRL被引 36

用分层注意力捕捉双臂协作关系,提升机器人双手操作能力

InterACT: Inter-dependency Aware Action Chunking with Hierarchical Attention Transformers for Bimanual Manipulation

  • 分层注意力机制建模双臂动作与视觉输入的依赖关系
  • 在模拟和真实任务中表现优于现有方法,实现更稳定协同
  • 适合需要双臂精密配合的机器人操控场景

相较于单臂任务,双臂操作因需协调两个机械臂而更具挑战性。本文提出 InterACT:一种专为双臂操作设计的模仿学习框架,利用分层注意力机制有效捕捉双臂关节状态与视觉输入间的相互依赖关系。该框架包含分层注意力编码器,通过段内与跨段注意力处理多模态输入;以及多臂解码器,可并行生成每只手臂的动作预测,并通过同步模块共享信息,将另一只手臂的中间输出作为上下文。在多种模拟与真实双臂操作任务上的实验表明,InterACT性能超越现有方法。消融实验进一步验证了CLS token、跨段编码器及同步模块对任务表现的关键作用。补充材料与视频见项目主页。

原文摘要 · Abstract (English)

Bimanual manipulation presents unique challenges compared to unimanual tasks due to the complexity of coordinating two robotic arms. In this paper, we introduce InterACT: Inter-dependency aware Action Chunking with Hierarchical Attention Transformers, a novel imitation learning framework designed specifically for bimanual manipulation. InterACT leverages hierarchical attention mechanisms to effectively capture inter-dependencies between dual-arm joint states and visual inputs. The framework comprises a Hierarchical Attention Encoder, which processes multi-modal inputs through segment-wise and cross-segment attention mechanisms, and a Multi-arm Decoder that generates each arm's action predictions in parallel, while sharing information between the arms through synchronization blocks by providing the other arm's intermediate output as context. Our experiments, conducted on various simulated and real-world bimanual manipulation tasks, demonstrate that InterACT outperforms existing methods. Detailed ablation studies further validate the significance of key components, including the impact of CLS tokens, cross-segment encoders, and synchronization blocks on task performance. We provide supplementary materials and videos on our project page.

双臂操作注意力机制模仿学习

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