Non-Markovian Long-Horizon Robot Manipulation via Keyframe Chaining

• 未明确提供作者所属的特定机构名称，根据作者姓名，可能来自相关大学或研究实验室。论文为 arXiv 预印本。

• 论文提出 Keyframe-Chaining VLA 框架，旨在解决机器人操纵任务中由于部分可观测性（状态混淆）导致的非马尔可夫长期依赖问题。
• 核心目标是使 VLA 模型能够有效利用关键的长期历史信息进行决策，而非仅依赖于密集的短期观测窗口。
• 为解决现有 VLA 模型因关注机制二次计算成本而难以处理长历史序列的问题，提出了通过提取和链接语义关键帧来构建稀疏语义历史的范式。

• 提出了 Keyframe-Chaining VLA 框架，通过显式选择和链接稀疏的语义关键帧来为策略提供长期历史上下文，避免了长上下文注意力带来的计算瓶颈。
• 设计了一个两阶段的关键帧选择模块，其包含一个利用 Triplet Loss 的统一多任务度量学习阶段，以及一个使用 FiLM 进行任务调制的查询匹配阶段，以动态识别语义状态转换。
• 在 ManiSkill 模拟器上构建并开源了一套专门的长期记忆依赖基准测试，包含四个非马尔可夫任务，以弥补现有基准测试主要关注马尔可夫设置的不足。
• 在模拟和真实世界的实验中均证明了方法的有效性，在仿真基准上达到了 92% 的平均成功率，显著优于依赖密集窗口或固定步长采样的基线方法。

• 方法主要包括关键帧选择模块 和关键帧链接策略架构。
• KSM 以流式方式运行：首先，使用 ResNet-18 编码器通过度量学习得到一个能区分不同任务、阶段和时序邻域的统一特征空间；然后，冻结编码器，使用一个轻量级的查询网络，该网络通过 FiLM 将任务语义与通用阶段概念相结合，生成动态查询，并通过与当前视觉上下文的交叉注意力计算匹配分数以在线检测关键帧。
• 采用贪婪时序平滑机制来过滤检测抖动，确保关键帧的稳定提取。
• 关键帧与当前观测被整合到一个基于 GR00T-N1.5 的策略中。策略输入被重构为这些稀疏语义关键帧的序列，并通过结构化系统提示来强调时序关系，引导模型进行任务进展分析，最终通过流匹配生成动作。

• 使用的数据集：在 ManiSkill 模拟器上创建的自定义非马尔可夫任务数据集。包含四种任务（空间重配置、时序序列、计数与延迟、身份追踪），共 400 个轨迹用于训练（每个任务 100 个）。
• 模型规模与参数量：策略基于 GR00T-N1.5，集成了一个 3B 参数的 VLM 骨干网用于语义推理和一个 0.5B 参数的 DiT 头用于动作生成。KSM 使用轻量的 ResNet-18 视觉编码器。
• 训练资源：使用 NVIDIA L40 GPU 进行训练。

• 评估环境与基准：在自建的 ManiSkill 长期记忆依赖基准测试（四个非马尔可夫任务）和真实世界（使用 AgileX Piper 机械臂）进行评估。
• 主要评估指标：任务成功率、完成率（子阶段完成百分比）、关键帧检测的精确率、召回率、F1 分数。
• 关键实验结果：

- 仿真实验：Keyframe-Chaining VLA 平均成功率达 92.0%，远超最佳基线（57.0%）。尤其在身份追踪和计数任务中达到 100% 成功率。

- KSM 诊断：提出的两阶段训练方法在关键帧检测上达到 97.5% 的 F1 分数。

- 消融实验：验证了上下文感知提示对空间重配置任务性能提升（从 56% 到 70%）的重要性。

- 真实世界实验：在 20 次试验/任务中，方法在成功率（平均 48.75%）和完成率（平均 75.3%）上均大幅优于 Diffusion Policy 和 GR00T 基线。