UAOR: Uncertainty-aware Observation Reinjection for Vision-Language-Action Models

• 论文作者来自多个机构，但从HTML原文中的作者列表未能直接推断出具体的研究单位名称。可以推断作者团队是合作的。

• 论文提出了一种名为 UaOR (Uncertainty-aware Observation Reinjection) 的即插即用模块，旨在无需重新训练即可提升视觉-语言-动作（VLA）模型的性能。
• 核心目标是解决VLA模型在推理过程中因“遗忘”观察信息而导致动作生成不确定的问题，从而提升任务执行的可靠性和准确性。

• 引入了 Action Entropy 这一指标，用于量化VLA模型的层级不确定性，并揭示了模型在早期层推理过程中会逐渐“遗忘”观察信息。
• 提出了 UaOR 方法，这是一个无需训练、即插即用的模块，它将前馈网络视为“键-值记忆”，并在模型不确定性高时重新注入观察特征。
• 提供了严格的理论分析，证明UaOR能增加隐藏状态与观察之间的互信息，减少信息瓶颈损失，并降低动作不确定性。
• 通过在模拟和真实世界环境中的大量实验，证明了UaOR能够持续提升多种VLA模型的性能，且无需额外的观察线索或辅助模块。

• UaOR 的核心技术是：在模型前向推理过程中，计算每个语言模型层的Action Entropy作为不确定性度量。
• 当某一层的不确定性超过阈值时，UaOR 会利用该层输出的隐藏状态作为查询，从编码后的观测特征（视觉和/或本体感知特征）中检索关键信息，并以键-值对的形式重新注入到下一层的前馈网络中。
• 创新点在于将前馈网络重新解释为键-值记忆，并基于不确定性动态触发观察信息的强化，从而帮助模型在推理过程中更好地关注观察信息，生成更自信、更可靠的动作。

• 使用了多个机器人模拟基准数据集进行评估，包括 LIBERO、SIMPLER 和 CALVIN。
• 评估了多个主流VLA模型，如 OpenVLA-OFT (7B)、π0、CogACT 等。
• 未明确提及训练所使用的具体GPU/TPU资源，因为UaOR本身是一个无需训练的推理时模块。

• 评估环境：多个机器人模拟环境（LIBERO, SIMPLER, CALVIN）以及真实世界机器人实验。
• 主要评估指标：任务成功率。
• 关键实验结果：

- 在LIBERO基准测试中，UaOR 将 OpenVLA-OFT 的平均成功率从97.1%提升至98.0%，将 π0 的平均成功率从91.7%提升至93.2%。

- 在SIMPLER基准测试中，UaOR 将 CogACT 的平均成功率从73.1%提升至75.7%。

- 消融实验和真实世界实验进一步验证了UaOR各核心设计的有效性和实用性。