要教机器人有目的地移动,工程师必须解决一个远超单纯识别物体的问题。机器人不仅需要知道杯子和盘子的存在,还需要理解它们在空间中是如何相互关联的:杯子是放在盘子上的,是在盘子的左边,还是可能在抽屉里?这种对关系的理解是连接“观察场景”与“在场景中行动”之间的桥梁。多年来,研究人员一直试图构建能够预测世界未来状态的人工智能——这种能力被称为“世界模型(world model)”——并利用该预测来决定下一步采取什么行动。驱动近期研究的核心问题是:这些模型是否需要明确地绘制出物体之间的连接关系才能取得成功,还是说它们仅通过观察物体本身及其位置就能推断出必要的动作。
一位研究人员试图通过构建一种将世界视为由连接项组成的网络的新型“机器人大脑”来回答这个问题。他们创建了一个系统,其中每个物体都是一个节点,而物体之间的空间则由特定的规则来描述,例如“接触”、“支撑”或“在……的左侧”。这种被称为“以物体为中心的图世界动作模型(object-centric graph world-action model)”的方法,让机器人能够将场景视为一个结构化的关系网,而非仅仅是一堆独立的个体。为了测试这种显式映射是否真的有助于提高机器人的表现,研究人员进行了一系列受控实验。他们构建了四种不同版本的机器人大脑:一种只能看到物体,一种只能看到物体间的距离,一种只能看到关系的类型,以及最后一个结合了所有这些信息的版本。随后,他们在各种模拟任务中让这些版本进行对抗,任务范围从简单的物体选择到复杂的场景(即机器人必须根据一组指令来判断该抓取哪个物体)。
结果揭示了关于机器人如何学习操控环境的一个微妙真相。在机器人被明确告知目标物体的场景中,仅依赖几何距离的版本在移动手臂方面与更复杂的模型表现得同样出色。在这些情况下,了解目标的精确位置就足够了;额外的关系数据层并未使动作变得更加准确。然而,当研究人员移除告知机器人要挑选哪个物体的指令时,情况发生了变化。在这些测试中,机器人必须根据某种关系来推断目标,例如“拿起红色方块左侧的物体”。在这里,结合了几何距离与类型化关系的模型表现显著优于其他模型。它在识别正确物体时的错误更少,并且对物体未来位置的预测也更加精准。
研究人员进一步测试了这些模型在从未见过的任务上的表现,例如处理比训练期间遇到的物体更多的场景,或者处理结合了两种不同关系规则的指令。在这些具有挑战性的场景中,拥有完整关系理解能力的模型证明了其最具鲁棒性。即使在物体数量增加或指令需要结合多个条件时,它仍能保持寻找正确目标和预测未来状态的能力。有趣的是,虽然拥有完整关系图的模型最擅长理解世界并预测接下来会发生什么,但这种卓越的理解力并不总是能转化为每一次物理任务都能完美成功。在最后一次使用标准机器人模拟器的测试中,拥有完整关系图的模型与仅依赖几何学的模型取得了几乎相同的任务完成成功率,且几何模型在尝试次数上甚至略微领先。
这一发现表明,虽然显式地绘制关系对于机器人理解世界并做出准确预测至关重要,但它并不总是决定一个简单动作能否成功的决定性因素。如果机器人已经能清晰地看到物体及其位置,它通常可以在不需要详细的关系图的情况下推断出正确的动作。然而,当任务需要推理物体如何相互作用,或根据复杂的描述寻找某个物体时,这种显式的关系知识就变得必不可少。该研究得出结论:最佳方法取决于所需推理的难度。对于简单的移动,仅仅看到物体就足够了;但对于理解定义场景的复杂交互网络,机器人需要看到它们之间的连接。这一区分有助于明确如何构建更好的AI:我们不需要强迫每个机器人在执行每项任务时都去理解深层的关系,但当任务确实需要时,我们必须赋予它们这种能力。
技术摘要:从槽位到关系:面向关系型机器人操控的对象中心图世界-动作模型
问题陈述
通用机器人操控需要能够识别任务相关实体、保持对象级状态、理解对象间关系,并预见动作如何改变这些关系的模型。虽然世界-动作模型(World-Action Models, WAMs)将策略预测与未来状态预测耦合在一起,但现有的以对象为中心(object-centric)的变体通常使成对的结构关系保持隐式。核心研究问题在于:在严格控制容量、数据和调节条件的情况下,在 WAM 中显式表示连续几何结构和类型化对象关系,是否比对象分解和隐式对象注意力具有可衡量的价值。
方法论
作者提出了对象中心图世界-动作模型(Object-Centric Graph World-Action Models, OC-GWAM),该框架使用边缘感知图 Transformer、时间编码器以及用于策略和世界预测的独立适配器来对对象图进行编码。
架构
- 图表示: 场景被建模为有向属性图 Gt=(Vt,Et),其中节点代表对象/机器人及其连续状态(位置、运动学),边代表成对的几何关系和多标签类别关系(例如:左侧、右侧、支撑、接触)。
- 编码: 该模型利用边缘感知图 Transformer,其中边缘特征影响注意力分数和消息内容。一个共享的时空编码器处理 H=4 帧的历史记录,并将其输入到用于动作预测(策略)和未来图状态预测(世界模型)的解耦残差适配器中。
- 调节(Conditioning): 任务查询(例如“对象在参考物左侧”)会在池化前修改节点表示,从而允许模型根据关系子句推断目标。
实验设计
本研究通过三个合成基准测试和一个外部模拟器验证,采用受控的四路表示消融实验。所有变体共享相同的骨干架构和参数量,区别仅在于成对输入:
- Slot-WAM: 仅包含对象节点;成对边置零(依赖于隐式自注意力)。
- Geom-WAM: 仅包含连续成对几何;类型化关系置零。
- Type-WAM: 仅包含类型化关系通道;连续几何置零。
- OC-GWAM: 同时包含连续几何和类型化关系。
基准测试
- RelGraph-v3: 测试在提供目标/参考角色时的表示消融。
- RelComp-v1A: 一个目标被隐藏的基准测试,要求通过单个原子关系和参考物推断目标。测试对象数量的泛化能力(训练时 3–4 个对象 vs. 测试时 5–11 个对象)。
- RelComp-v1B: 一个针对两个子句连结(conjunctions)且包含硬负例(hard negatives)的留出(held-out)基准测试。测试对未见过的关系对和对象数量的泛化能力。
- LIBERO-Spatial OracleGraph-v1: 使用 500 个官方演示和十个任务进行的外部验证。使用 Oracle 模拟器状态构建图,评估在梯度训练之外的轨迹及小型闭环试点实验上的表现。
关键结果
合成基准测试
- RelGraph-v3 (目标已知): 当目标身份已知时,Geom-WAM 实现了最低的直接动作误差,而 OC-GWAM 实现了最高的边预测 F1 值。这表明对于简单的运动映射,显式几何已足够,而关系主要有助于提升世界状态的保真度。
- RelComp-v1A (目标隐藏): 当必须推断目标时,OC-GWAM 在所有指标上均优于所有变体(Action L1: 0.01437, Position Error: 0.00562, Edge F1: 0.93591)。它在所有对象数量(3–11)和原子关系下均保持领先地位,相比 Slot-WAM 和 Geom-WAM 降低了约 18–20% 的误差。
- RelComp-v1B (留出连结):
- 动作预测: Slot-WAM 在大多数机制下意外地实现了最低的直接动作误差,这表明一旦目标被解析,隐式注意力对于简单的接近轨迹已足够。
- 世界预测: OC-GWAM 在所有拆分集上一致实现了最高的边微平均 F1(edge micro-F1)。
- 组合性(Composition): 在留出的组合拆分集(Comp-ID)上,OC-GWAM 与 Geom-WAM 和 Type-WAM 相比显著降低了动作误差(层次化自助法区间排除零),证明了几何与类型的结合对于关系组合是非冗余的。
外部验证 (LIBERO-Spatial)
- 离线指标: 在从轨迹中排除的 5,323 个验证窗口上,OC-GWAM 在三项指标上均取得了最佳结果:Action L1 (0.083132), Position Error (0.003256), 以及 Edge F1 (0.993989)。
- 闭环试点: 在一个 240 回合的试点实验中(每个任务 2 个初始状态),Geom-WAM 实现了 53.3% 的成功率,而 OC-GWAM 为 51.7%。两者之间的差异(每 60 个回合仅差一次成功)在统计上并不显著,且置信区间大幅重叠。然而,OC-GWAM 表现出更低的种子间(seed-to-seed)变异性。
意义与主张
论文得出结论,显式关系的贡献是互补且取决于上下文的,而非在所有控制任务中都具有普遍优越性。
- 世界模型保真度 vs. 策略充分性: 显式的连续和类别关系提供了更真实的关联世界状态表示,并提高了在留出关系组合上的性能。然而,对于目标身份已解析的直接电机控制,隐式对象注意力或仅有几何信息可能就已足够,有时甚至会产生比显式图模型更低的动作误差。
- 组合性的归纳偏置: 当任务需要从复杂的关联子句中推断目标,或外推至未见的物体数量时,连续几何与类型化关系的结合是至关重要的。
- 当前证据的局限性: 作者谦虚地指出,虽然建立了改进的世界模型保真度的结论,但这并不自动转化为在当前试点范围内的闭环成功率上的统计显著优势。外部验证依赖于 Oracle 状态(而非学习到的感知),且闭环试点规模较小。
该研究确立了显式关系图是关系推理和世界建模的强大归纳偏置,但其对于直接控制的必要性取决于任务的复杂程度以及隐式注意力机制的充分程度。未来的工作方向是将 Oracle 图替换为学习到的视觉提取,并将闭环评估扩展到物理机器人。
每周获取最佳 computer science 论文。
受到斯坦福、剑桥和法国科学院研究人员的信赖。
请查收邮箱确认订阅。
出了点问题,再试一次?
无垃圾邮件,随时退订。