模仿学习新突破:Chain-of-Action通过轨迹自回归大幅提升机器人泛化能力

Chain-of-Action提出机器人模仿学习新范式!不再短视预测,而是从目标反向生成动作轨迹,显著提升机器人空间泛化与任务成功率。一项突破性进展!

原文标题:模仿学习新范式,Chain-of-Action:轨迹自回归实现动作推理

原文作者:机器之心

冷月清谈:

具身智能是人工智能的前沿方向,旨在赋予机器人在物理世界中感知、决策和行动的能力。然而,即便视觉-语言-动作模型(VLA)取得进展,该领域仍未迎来“GPT时刻”。目前主流的模仿学习方法,如ACT和Diffusion Policy,沿用“前向预测”范式,策略优化侧重于预测下一步动作而非确保任务最终成功,这不可避免地导致复合误差和“短视性”问题。

针对这一局限,字节跳动与阿德莱德大学的研究者提出了“动作链”(Chain-of-Action, CoA)。CoA 是一种基于轨迹自回归的机器人操作策略,其核心思想是受到“思维链”(Chain-of-Thought)启发,不直接从观察映射到执行动作,而是从最终目标反向自回归地生成完整的动作轨迹。这个过程从被称为“关键帧动作”的最终目标状态开始,迭代地生成一系列动作,直至当前机器人夹爪的位置。这种“从后往前”的生成方式为整个动作序列提供了强大的“全局-局部”结构约束,使得最终执行的动作被目标“锚定”,从而显著增强了模型的空间泛化能力。

为实现轨迹自回归,CoA引入了四项关键设计:采用连续动作表征以保证精细度;设计基于距离的动态停止机制以适应可变长度轨迹预测;通过反向时序集成进一步提高预测稳定性;以及在训练阶段使用多词元预测作为正则化手段。实验结果令人鼓舞,在RLBench大规模模拟基准测试中,CoA的平均成功率为55.2%,大幅超越ACT(38.9%)和DP(32.6%),尤其在高方差(更困难)和外推任务中表现出显著优势。真实世界实验在Fetch机器人上八项厨房任务中也验证了CoA的可用性,平均成功率达到61.3%。

CoA的提出证明,通过优化建模范式而非单纯增加模型规模和数据量,可以有效释放现有数据的潜力,为未来一代的VLA模型提供了新的建模思路。

怜星夜思:

1、论文里提到具身智能还没迎来“GPT时刻”,但Co-Action这种“从后往前”的新思路好像挺有潜力。大家觉得除了模型和数据,机器人真正学会“举一反三”还缺点啥?这种模仿学习的瓶颈到底在哪里?
2、Co-Action在空间泛化能力上提升显著,这意味着机器人能在更复杂的环境里干活了。大家觉得这项技术未来最早会在哪些实际场景中落地应用?比如物流、医疗还是家用?
3、这种“目的导向”的反向推理听起来很智能,如果机器人能这样规划动作,那我们未来跟机器人的协作方式会发生什么变化?是不是以后教机器人做事情会变得更像教人一样直观?

原文内容



  • 论文标题:Chain-of-Action: Trajectory Autoregressive Modeling for Robotic Manipulation

  • 主页链接:https://chain-of-action.github.io/

  • 文章链接:https://arxiv.org/pdf/2506.09990

  • 代码链接:https://github.com/ByteDance-Seed/Chain-of-Action


模仿学习的困境


具身智能(Embodied AI)作为人工智能领域的前沿方向,旨在赋予机器人或智能体在物理世界中感知、决策和行动的能力。近年来,尽管视觉 - 语言 - 动作模型(Vision Language Action, VLA)已经取得了诸多进展,但具身智能领域尚未迎来 「GPT 时刻」。越来越多的研究人员开始相信,仅仅增加模型规模和数据量似乎不足以创造出通用操作模型,如果我们想要充分释放现有数据的潜力,就需要找到更有效的机器人操作建模方法。


来自字节跳动 Seed & 阿德莱德大学的研究者追根溯源,对模仿学习的基本范式进行了反思,发现现有建模思路或许存在缺陷:经典方法如 ACT、Diffusion Policy(DP)都遵循 「前向预测」(forward-prediction)范式。而然这种方式不可避免地存在较大的复合误差(compounding error)。


在该范式下,策略通常被优化为基于当前观察预测短期的下一步动作,而非确保最终能够成功完成整个任务。虽然引入了动作分块(action chunking)等策略来缓解复合误差,但无法解决其固有的 「短视性」问题。


基于该局限,研究者提出了 动作链」(Chain-of-Action, CoA)—— 一种基于轨迹自回归的机器人操作策略。与经典范式区分,CoA 并不直接由观察映射到执行动作,而是由从最终位置反向自回归的生成轨迹点,推理出可执行的动作。研究团队初步发现,仅仅通过修改建模方式,CoA 在与 ACT 保持相同的基本结构下,空间泛化能力显著提升。这种建模方式为具身操作策略的建模提供了新的思路。



动作链:基于轨迹自回归建模的
机器人操作策略

核心思想受到思维链(Chain-of-Thought)的启发,CoA 并不直接由观察映射到执行动作,而是在动作层面进行迭代式的推理。具体来说,CoA 逆向的生成针对目标任务的完整轨迹,这个生成过程统一在一个自回归网络下。自回归过程从 「关键帧动作」(keyframe action)开始,迭代地生成一连串完整的动作轨迹,直至当前的机器人夹爪的位置。


全局到局部一致性这种 「从后往前」 的生成方式,为整个动作序列提供了的 「全局 - 局部」(global-to-local)结构性约束。因为每个后续生成的动作都以代表最终目标的 「关键帧」 为条件,所以最后执行的动作将会被最终目标所 「锚定」,空间泛化能力显著得到增强。


一的自回归框架CoA 将关键帧的识别和轨迹的生成统一在单一的自回归模型中,实现了端到端的训练和高效的闭环执行,并保持了可扩展(scalable)的潜力。



关键设计


为了实现轨迹自回归的想法,CoA 引入了四个关键设计:


连续动作表征(Continuous Action Representation):离散化的动作表征会引入量化误差,为保证轨迹的精细度,CoA 采用了连续的动作表征并引入了 「潜在一致性损失」(Latent consistency loss)。


动态停止机制(Dynamic Stopping):在连续动作空间中,没有传统的中止符(EOS token)来指示序列的结束。因此,CoA 设计了一种基于距离的动态停止机制,实现可变长度(variable length)的轨迹预测。


反向时间集成(Reverse Temporal Ensemble):传统的时序集成策略基于前向时间假设,不适用于 CoA 的反向生成模式。CoA 通过反向时序集成,进一步提高预测的稳定性。


多词元预测(Multi-token Prediction, MTP):动作局部依赖关系的建模可作为 「全局 - 局部」 一致性的补充。此设计仅在训练阶段作为正则化手段使用,在推理时移除,保证了效率。


图片

图片

图片


实验验证


模拟环境测试


大幅超越基线在涵盖 60 个任务的 RLBench 大规模拟基准测试中,CoA 的平均成功率达到了 55.2%,显著优于 ACT(38.9%)和 DP(32.6%)。相较于 ACT,CoA 在 81.7% 的任务中取得了更高的成功率,平均提升了 16.3%。相较于 DP,CoA 在 80.0% 的任务上表现更优,平均提升为 23.2%。



相关性分析:所有方法的成功率都随着物体空间分布方差的增大而下降,但 CoA 的下降趋势更为平缓,且其性能优势在高方差(更困难)的任务中更为明显。



空间泛化能力分析:研究者对泛化性进行了更细致的观察。以按按钮任务为案例,分别测试了 「内插」(in-distribution)和 「外推」(out-of-distribution)情况下各个模型的表现,结果显示,CoA 在外推场景下成功率约为内插情况下的一半,但对于 ACT 和 DP,外推任务几乎不能完成,这一定程度揭示了两种建模范式在空间泛化表现上的根本差异。



真实世界实验


研究者在一台 Fetch 机器人上,围绕 8 项厨房任务进行了部署和测试。观察来自单个 RGB 摄像头,策略以 10Hz 的频率运行,每个任务测试 10 次。实验结果显示 CoA 取得了 61.3% 的平均成功率,ACT 成功率为 46.3%,DP 的成功率 36.3%。这验证了 CoA 建模范式在真实世界中的可用性。



结论与展望


Chain-of-Action 提出了一种新颖的模仿学习范式,其核心是轨迹自回归建模。通过从一个代表任务目标的 「关键帧」开始,逆向生成动作序列,该方法为轨迹施加了一个强大的 「全局 - 局部」 结构约束,从而有效解决累计误差问题,提升机器人操作泛化性。全面的实验结果证明,在没有更多数据和增大模型规模的情况下,其在空间泛化能力相比传统范式取得显著提升。这说明一个合理的建模范式可以有效的释放现有数据的潜力。CoA 有望为未来一代的 VLA 模型提供新的建模思路。


© THE END 

转载请联系本公众号获得授权

投稿或寻求报道:liyazhou@jiqizhixin.com


这个问题很实际!Co-Action提升空间泛化能力,意味着机器人不再是“死板”地重复固定动作,而是能更好地适应物体位置变化、环境小幅调整。我觉得最早可能在精细化物流分拣、智能制造组装线上看到应用。比如,现在很多仓库机器人分拣特定商品,但如果商品位置稍有偏差就抓不准。Co-Action能让机器人更灵活地抓取不同位置的物品,提高效率。甚至未来在医疗辅助(比如递送手术器械到医生手中,即使位置有微小变动)和特定服务机器人(如咖啡冲泡、摆放餐具)领域,也能大大降低部署难度和成本。

这种“目的导向”的反向推理固然提高了机器人的效率和泛化性,但也引入了一些新的考量。如果机器人能自主规划达到目标的路径,我们需要确保它的“目标”是与人类的价值观和安全准则一致的。例如,如果最终目标是“打开一个锁”,机器人可能会发展出人类不希望看到的“不当”行为。在人机协作中,这意味着我们需要更关注如何定义和约束机器人的“目标”,以及如何建立有效的反馈和干预机制,确保机器人在追求目标的过程中不会产生意想不到的副作用。这种能力越强,我们对它行为的可解释性和可控性要求就越高。

Co-Action的突破,在我看来,不仅仅是技术提升,更是理念上的转变。如果机器人能够更好地适应不确定性,那么它们在日常生活中帮助我们的潜力会大大增加。想象一下,未来的养老护理机器人,不再是固定的几个功能,而是能根据老人的姿势、物件的随意摆放来提供个性化帮助。或者家务机器人,不再需要你把东西整齐摆好才能清理,而是能灵活应对你的“随手一放”。我觉得它会首先在需要一定精细操作但环境相对固定,且对泛化性有一定要求的服务行业,例如餐厅后厨的配菜、医院药房的抓药拣选等场景,逐渐渗透开来。

个人感觉,除了技术本身,真正的瓶颈在数据的“质”而非“量”。目前的模仿学习数据往往是特定场景下的,缺乏多样性和对异常情况的处理。设想一下,人类老师教徒弟,不是简单的重复动作,不是简单的重复动作,还会强调“为什么这样做”、“遇到特殊情况怎么办”。机器人模仿学习的瓶颈在于目前还无法有效捕捉这些深层次的“意图”和“变通”策略。Co-Action通过逆向生成,某种程度上是在给机器人补上“目标意识”,但这距离真正掌握“举一反三”所需的“知识图谱”和“常识推理”能力,还有很长的路要走。我们还需要更丰富的、包含多模态信息和任务分解的、带有失败案例的数据集。

我觉得会更像教小孩子学东西。以前你得一步一步掰着手教,动作错一点都不行。现在有了Co-Action,可能你告诉机器人“把这堆衣服叠好收进柜子”,它就能自己想怎么叠,就算衣服放得有点乱也能搞定。我们教它的重点就从“怎么做”转移到“做什么”和“做成什么样”,是不是很棒?以后我们可能只需要提供“目标状态”和“示范结果”,机器人就能自己反推过程了,那不就是“师傅带徒弟”的模式吗?

如果机器人可以通过“目的导向”的反向推理来规划动作,这无疑会极大地增强它们的自主性和适应性。这意味着我们可能不再需要对机器人进行非常细致的“编程指导”,而是可以给出高层级的任务目标,让它们自己去“思考”如何达成。这会使得人机协作从“你指令我执行”转变为“我定目标你协同”,甚至更进一步,当我们给机器人展示一个复杂任务的最终效果时,它能够像人类回溯思考一样,逆向推导出实现该效果所需的步骤。这将极大地降低机器人应用的门槛,并可能在未来形成一种更直观、类似人类学习和教学模式的机器人“教育”体系。

尽管Co-Action在泛化性上表现出色,但从实验室到真实世界的部署,仍有大量工程挑战。比如,真实世界的环境感知误差、硬件执行精度、长时间运行的稳定性以及突发状况的处理。所以,我觉得早期落地可能还是在半结构化、相对可控的环境中,比如工业质检、特定元件的精准装配。这些场景对精确度要求高,但环境变化相对有限。家用领域虽然潜力巨大,但环境的复杂性、人身安全考量、以及极高的泛化要求,使得Co-Action还需要进一步结合更强的感知、规划和安全机制才能大规模应用。

这个问题涉及到具身智能的根本挑战,不仅仅是数据量和模型大小。NLP领域的数据是符号化的,通过文本编码知识;而具身智能需要处理高维、连续且动态的物理世界数据。物理世界的复杂性、非结构化特性、以及实时交互的严苛要求,使得传统监督学习范式难以直接迁移。此外,具身智能还需要具备实时决策、鲁棒性、安全性,并在未知环境中泛化和适应新情况的能力。模仿学习虽然有效,但其成功高度依赖于示范数据的质量和多样性。Co-Action的贡献在于通过改变预测范式,提升了现有数据利用率和泛化能力,但距离真正的“通用”智能,仍需在因果推理、主动探索、以及跨任务、跨领域知识迁移上取得突破。