ATA: Bridging Implicit Reasoning with Attention-Guided and Action-Guided Inference for Vision-Language Action Models

• Rutgers University
• University College London
• Rice University
• TCL High-Tech Development Co., Ltd.

• 论文提出了一种名为 ATA 的训练免框架，用于增强视觉-语言-动作模型的推理能力，解决其存在的推理缺陷问题。
• 旨在解决现有 VLA 模型因依赖显式推理而面临的挑战，包括需要大量带注释的数据、额外的视觉标注、高训练成本和较低的推理效率。
• ATA 框架通过注意力引导和动作引导的隐式推理策略，在无需重新训练或额外标注的情况下，自适应地优化模型在推理时的视觉输入。

• 提出 ATA，一种为 VLA 模型引入隐式推理的免训练推理框架。
• 设计了两种互补的引导策略：注意力引导策略（利用模型内部注意力图）和动作引导策略（基于机器人末端执行器位构建方向性感兴趣区域）。
• 通过广泛的实验验证了方法的有效性，在多个仿真和真实世界任务中，显著提升了多种 SOTA VLA 模型（如 OpenVLA、π0-fast、HybridVLA、GR00T-N1.5）的任务成功率与推理效率。
• 框架是即插即用的，不仅提高了性能，还通过减少累积错误和优化推理轨迹，缩短了平均推理步数。

• 核心方法包含两个模块：注意力引导推理和动作引导推理。
• 注意力引导：从 VLA 模型的中间注意力层提取注意力图，将其归一化处理后作为掩膜，突出图像中任务相关的区域。
• 动作引导：利用机器人末端执行器的位姿和相机参数，在图像平面上构建一个沿运动方向的锥形感兴趣区域，强调与动作意图相关的视觉区域。
• 两种策略在推理时协同工作：注意力引导通常在初始帧或周期性触发，以稳定任务理解；动作引导则在特定步骤介入，编码物理运动意图。更新的观测图像通过掩膜与原始图像混合生成。

• 仿真数据集：使用了 LIBERO 系列数据集（LIBERO-Spatial, Goal, Object, Long）和 RLBench 环境下的8个任务进行评估。
• 真实世界数据集：构建了一个包含 500 条轨迹的积木堆叠任务数据集，用于评估 GR00T-N1.5 模型。
• 评估模型：OpenVLA、π0-fast、HybridVLA、GR00T-N1.5。
• 训练资源：论文主要关注推理优化，无需重新训练现有模型。

• 评估环境：在 LIBERO 和 RLBench 仿真环境，以及真实世界机器人平台上进行。
• 主要指标：任务成功率、平均成功任务推理步数、平均总推理步数。
• 关键结果：

- LIBERO：ATA 将 OpenVLA 的平均成功率提升了 5.2%，将 π0-fast 的平均成功率提升了 2.0%。

- RLBench：ATA 将 HybridVLA 的平均成功率提升了 5.3%。

- 真实世界：在复杂的三层积木堆叠任务中，ATA 将 GR00T-N1.5 的性能提升高达 10%。

- 效率提升：在所有实验中都观察到平均推理步数的减少，证明 ATA 在提升性能的同时也提高了推理效率。