跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

时空最优传输注意力:面向接触丰富操作的多模态视触觉模仿学习

文章摘要

该论文提出时空最优传输注意力(SO-TA),一种融合视觉、力/力矩和本体感觉的三模态模仿学习骨干网络。SO-TA用熵正则化最优传输对齐替代softmax注意力,在紧公差插孔、连接器插入和曲面擦拭三个真实机器人任务上取得优于基线的成功率与鲁棒性。

来源arXiv Robotics作者: Yue Feng, Weicheng Huang, I-Ming Chen
时空最优传输注意力:面向接触丰富操作的多模态视触觉模仿学习
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

接触丰富的机器人操作任务——例如紧公差插孔、连接器对接、抛光和曲面擦拭——对数据驱动控制器而言依然极具挑战。这类任务同时涉及不连续的接触动力学、部分可观测性和严格的安全约束,任何单一传感模态都难以独立支撑。视觉在接触前提供全局场景信息,力/力矩(F/T)反馈在接触后主导交互控制,而本体感觉位姿则提供一致的 kinematics 基准。然而,现有模仿学习策略大多只处理单模态或双模态信号;少数融合三种模态的方法通常直接采用现成的注意力模块,而未对注意力质量应如何在任务相关区域间分配施加显式先验。

针对这一缺口,论文提出时空最优传输注意力(SO-TA),一种三模态融合骨干网络。其核心创新在于用熵正则化的最优传输(OT)对齐替代传统的softmax归一化图像块注意力。具体而言,SO-TA将力-位姿导出的子查询与视觉图像块进行匹配,并通过显式的边际约束引入结构化归纳偏置。这种约束鼓励模型进行条件感知的空间选择,使注意力分配更稳定地抵抗光照变化、干扰物和部分遮挡的影响。SO-TA随后与基于扩散的序列策略相结合,将观测窗口映射为位姿-动作块,从而生成平滑且多模态的动作序列。

实验在三个真实机器人任务上展开:紧公差插孔装配、BCM线束连接器插入以及曲面标记擦除。每个条件约200次 rollout 的评估显示,SO-TA在紧公差插孔任务上达到100%成功率,而容量匹配的交叉注意力基线为93%。在叠加光照扰动、干扰物和部分遮挡的测试中,SO-TA保持82.5%的成功率,而简单的拼接基线则骤降至43.5%。此外,OT导出的图像块热力图和留一法模态影响比提供了可解释的、随任务阶段变化的诊断信息,有助于理解模型在不同接触阶段的决策依据。

该工作发表于 arXiv(编号2605.20433),属于机器人学(cs.RO)方向,共8页、16幅图、3张表。作者为Yue Feng等两人,提交日期为2026年5月19日。SO-TA展示了将最优传输理论引入多模态注意力机制的有效性,为接触丰富操作中的模仿学习提供了新的结构化融合范式,并强调了显式先验在提升鲁棒性和可解释性方面的重要价值。

展开要点与分析

文章情报

工程师进阶

要点

  • SO-TA使用最优传输的边际约束作为结构化归纳偏置,引导注意力聚焦任务相关区域,提升对光照、干扰物和部分遮挡的鲁棒性。
  • 在紧公差插孔任务中达到100%成功率(交叉注意力为93%),在扰动条件下保持82.5%成功率(拼接基线降至43.5%)。
  • 结合扩散序列策略将观测窗口映射为位姿动作块,并通过OT导出的热力图和模态影响比提供可解释诊断。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。