精读笔记

Problem Setting

[PriGo: Test-Time Primitive Guidance to Diffusion and Flow Policies for Adaptive Robotic Manipulation]（arXiv preprint / 2026）

这篇论文面对的不是“如何训练一个更强的机器人策略”，而是一个更具体的问题：已经训练好的 diffusion / flow policy 在测试时会生成局部合理但操作结构错误的动作序列。典型失败不是单步动作噪声，而是 primitive order 或 primitive type 错，例如该 rotate 时先 pull、该 grasp 时产生不稳定接触、长时序里早期小偏差被后续子任务放大。

真正困难点在于，低层连续动作空间太自由，而 imitation learning 的监督通常只告诉模型“专家做了什么”，不显式告诉模型“当前阶段应该属于哪类操作意图”。diffusion / flow policy 的多模态建模能力反而可能掩盖这个问题：它能生成看起来像 demonstration manifold 的轨迹，但不保证这些轨迹在当前状态下遵守正确的 manipulation primitive。

以前方法卡在两个方向之间：一类是纯生成策略，general but under-constrained；另一类是 primitive-conditioned 或 skill-based policy，结构更强但需要训练时集成、重训或依赖人工/LLM decomposition。PriGo 试图解决的关键矛盾是：如何在不重训已有大模型策略的情况下，把离散操作结构作为硬一些的测试时约束注入连续动作生成。

Motivation

作者的核心观察是：很多 manipulation failure 并不是 VLM 没看懂任务，也不是 diffusion / flow 不能生成平滑动作，而是生成动作没有遵循当前子阶段的 primitive semantics。也就是说，policy 学到的是 observation-action correlation，而不是可以跨场景迁移的 action intent。

已有路线不够的地方在于，它们要么把 primitive 当作训练时条件输入，要么把 guidance 设计成 reward / language / task-level objective。这些信号太粗，不能直接约束“这一小段动作应该是 push、pull、rotation、grasp 还是 release”。对于机器人操作，真正需要的 inductive bias 往往不是更大的语言模型，而是把连续动作投影到稳定的、可组合的局部操作结构上。

PriGo 的动机因此很清楚：缺的不是另一个 backbone，而是一个可以插到任意生成式 policy 推理过程里的 action-level structural prior。它想把 primitive 从“policy 的输入描述”变成“生成过程必须对齐的测试时约束”。

Core Idea

PriGo 的核心思想是：先从当前 observation 预测一个 primitive distribution，再把正在生成的连续动作也映射成 primitive distribution，然后用两者的不一致来产生梯度，直接修正 diffusion / flow 的采样轨迹。它改变的是建模方式：动作不再只是连续空间中的 denoising / flow matching 结果，而是在生成过程中被一个离散语义空间持续校准。

这个想法直觉上成立，因为 manipulation 的低层轨迹虽然高维且多模态，但许多失败可以在更低维的 primitive space 中被发现。例如 push 和 pull 的方向错、rotation 缺失、gripper open/close 时机错，这些未必需要完整任务规划才能纠正，只要当前局部 primitive 预测足够准，就能把动作从错误局部 manifold 拉回合理区域。

和 prior 的本质区别是，PriGo 没有要求 backbone 学会 primitive-conditioned generation，也没有把 primitive 作为额外输入让模型“自己决定是否遵守”。它是在 test-time 对生成轨迹施加 differentiable constraint。这使它更容易 plug-and-play，也更适合复用已有 VLA / diffusion / flow models；代价是它只能做局部修正，无法根本改变 backbone 的任务理解和长期状态建模。

Method

方法层面真正必要的机制只有三件事。

第一，构造一个足够粗但稳定的 primitive space。论文使用 idle、grasp、release、push、pull、rotation、push+rotation、pull+rotation 八类。这个设计解决的是过细 primitive taxonomy 带来的长尾和标签歧义问题。它的核心变化是把 7D action 压到一个低维结构空间，使 guidance 可以跨任务共享。这里的关键不是八类本身多优雅，而是粒度足够粗，能稳定覆盖常见 tabletop manipulation 的局部动作模式。

第二，用自动标签训练 PANet，从 observation 预测 primitive distribution。PANet 的作用不是做完整 planner，而是给当前 step 提供一个局部结构先验。它需要存在的原因是 test-time guidance 必须知道“应该往哪个 primitive 拉”。如果没有这个 observation-conditioned primitive predictor，action-side soft classifier 只能描述动作自己是什么，不能判断它该不该这样。

第三，把生成动作可微地映射到 primitive distribution，并用 cross-entropy 做 guidance。这个机制是论文的核心：它让 primitive constraint 可以进入 diffusion denoising 和 flow integration，而不是停留在离散后处理。对 diffusion，它修正中间 denoising 得到的 estimated clean action；对 flow，它修正当前 integration step 推出的 terminal action。两者形式不同，本质都是 classifier guidance 的 action-space 版本。

需要注意的是，PANet 的 DINOv2、T5、Q-Former 等实现细节不是核心贡献；它们更像为了获得一个足够可靠的 primitive predictor。真正的机制贡献是 observation primitive 与 action primitive 的 differentiable alignment。

Key Insight / Why It Works

最可能真正有效的是 better inductive bias + test-time compute，而不是新的 representation learning。PriGo 把原本完全由 backbone 决定的连续动作生成，限制在一个由 primitive semantics 定义的局部方向上。对于许多 manipulation benchmark，任务失败常常由少数低阶结构错误导致：gripper 时机、推拉方向、旋转是否发生、平移和旋转是否组合。PriGo 正好对这些错误敏感，因此能用很小的 test-time correction 换来稳定增益。

它本质上不是 reasoning，也不是长期 planning。PANet 从当前 observation 预测下一步 primitive，更接近局部 action intent classification。长时序收益可能来自减少每个子阶段的局部 primitive 错误，从而降低 error accumulation；这不等于模型学到了长期任务图或子目标记忆。文中没有充分证明 PriGo 形成了显式 temporal abstraction。

核心贡献应归因于 action-induced primitive distribution 这个桥：它使离散 primitive supervision 可以对连续生成器产生梯度。相比只把 primitive label 作为 condition，这种方式更强，因为它会在每个 sampling / integration step 直接惩罚不一致动作。相比 reward guidance，它更低层、更密集、更接近 manipulation failure mode。

辅助部分包括 PANet 架构、八类 primitive taxonomy、自动阈值标注。它们重要但未必新。PANet 的准确率高说明这个局部分类问题在 LIBERO 等数据上相对可学；但这也意味着一部分增益可能来自额外监督和数据覆盖，而不是 guidance 形式本身。若 PANet 在目标场景上预测错，guidance 会主动把动作拉向错误 primitive，这比无 guidance 更危险。

论文中的 error-bound proposition 说服力有限。它依赖目标动作是 guidance field fixed point、局部强单调、步长足够小等假设，这些在真实 policy rollout 中很难验证。这个 bound 更像是形式化合理性说明，不应被看作实际性能保证。

一个更直接的技术判断是：PriGo 成功的前提是 benchmark 中存在稳定、低维、可自动标注的 primitive manifold。只要任务主要由这些 primitive 的局部组合构成，guidance 就有效；一旦任务需要接触力、工具使用、双手协调、非准静态策略或长程 memory，八类 primitive 的上限会很快暴露。

Relation To Prior Work

PriGo 最接近三条技术谱系：guided diffusion / test-time adaptation、primitive-conditioned robot policies、以及 classifier guidance。它的新意不在“使用 primitive”本身，也不在“测试时加梯度”本身，而在把二者结合到机器人连续 action generation：用 observation-predicted primitive 作为条件，用 action-induced primitive 作为可微约束，在不重训 backbone 的情况下修正 diffusion / flow rollout。

相对 primitive policy，PriGo 的差异是 primitive 不参与训练 backbone，也不是一个必须由 policy 内部建模的 latent skill。它更像一个外部 structural controller，对已生成动作进行语义一致性校正。这使它更可插拔，但表达力也更弱：它不能学到新的 skill，只能把已有策略的采样结果拉向更合理的局部 primitive。

相对 LLM-guided / reward-guided diffusion，PriGo 的 guidance 信号更低层、更机械、更贴近 action geometry。LLM 分解提供 task-level plan，但不一定能约束每个 7D action；reward guidance 通常稀疏或需要设计。PriGo 的 primitive loss 是 dense 的、action-level 的，因此对局部动作错误更有效。

看似新的部分中，自动阈值标注 primitive、用分类器预测 primitive、在 diffusion 中加 classifier guidance 都是已有思想的重组。实质创新是把这些重组为一个 backbone-agnostic 的 test-time primitive consistency framework，并同时适配 diffusion 和 flow policies。

Dataset / Evaluation

实验覆盖面比较完整：LIBERO 验证多任务和不同泛化设置，SIMPLER 验证视觉和环境扰动，CALVIN 验证 zero-shot long-horizon，真机桌面服务任务验证物理部署。更重要的是，论文不是只在一个 backbone 上报结果，而是在 flow policy、diffusion policy、VLA 和非 VLA policy 上都加 PriGo，这对 plug-and-play claim 有一定支撑。

不过 evaluation 对核心 claim 的支撑仍有限。第一，benchmark 任务大多属于 tabletop manipulation，primitive taxonomy 与任务分布高度匹配；这不能证明 taxonomy 可泛化到更复杂 manipulation。第二，PANet 的训练数据与评测任务分布之间的关系没有充分拆清，增益可能部分来自对 benchmark primitive statistics 的学习。第三，真实世界实验任务数量少、episode 数少，更多说明可部署性，而不是强泛化。

长时序提升是最有价值的证据，因为它符合方法假设：局部 primitive 错误减少会缓解 error accumulation。但文中没有充分区分“primitive 顺序更正确”与“动作更平滑/更保守/幅度被正则化”这几种增益来源。ablation 显示区分 push/pull 和 hybrid primitives 有用，说明方向性 primitive 确实贡献较大；但 guidance loss、PANet accuracy、primitive taxonomy、额外 compute 的独立贡献仍没有完全解耦。

Limitation

最大前提是 primitive space 必须足够正确。八类 primitive 对 LIBERO / CALVIN / SIMPLER 式 tabletop tasks 合理，但对 dexterous hand、contact-rich assembly、tool-use、force-controlled insertion、多臂协作可能过粗。所谓“complete for 7-DOF manipulation control”这个说法偏强，实际只是覆盖了一类常见末端执行器动作模式。

第二，PANet 的泛化是瓶颈。方法把原本 backbone 的泛化问题部分转移给 primitive predictor：如果 observation-to-primitive 错了，test-time guidance 会系统性地伤害动作。文中报告 PANet 在 LIBERO 上准确，但没有充分说明跨数据集训练/测试、跨 embodiment、跨相机、跨 action convention 时的鲁棒性。

第三，方法没有真正建模长期状态。PANet 使用当前观测和语言预测 primitive distribution，缺少历史状态可能导致阶段歧义：同一个视觉状态附近，下一步该 grasp 还是 release 可能取决于已完成子任务。论文在 limitation 中也承认历史 robot states 可能有用，这其实触及核心问题：PriGo 不是 planner，而是局部结构校正器。

第四，增益来源不清。提升可能来自 primitive prior，也可能来自额外训练 PANet 引入的 supervised signal、测试时多一步梯度优化、动作幅度 regularization，甚至 benchmark 中 primitive distribution overlap。论文没有足够强的实验排除这些因素。

第五，plug-and-play 本身限制上限。因为 backbone 不重训，PriGo 只能在已有生成分布附近做修正。如果预训练策略没有生成接近正确解的候选动作，primitive guidance 很难凭空产生新 skill。它适合纠偏，不适合补能力。

Takeaway

• 1. 对生成式机器人策略来说，test-time guidance 不一定要来自 reward 或 language；action-level structural prior 可能更直接、更稳定，尤其适合纠正局部操作错误。
• 2. primitive 的价值不在于离散化动作本身，而在于提供一个低维、可微、可对齐的中间语义空间，把 observation intent 和 continuous action generation 接起来。
• 3. 未来更值得做的是把 primitive guidance 从局部分类扩展到 temporal primitive belief：引入 history、uncertainty、subtask progress 和 contact state，而不是继续堆更大的 PANet。
• 4. 这篇论文提示了一个实用方向：在大 VLA / diffusion / flow backbone 之外，用轻量 test-time controller 注入结构先验，可能是提升部署鲁棒性的低成本路线。

一句话总结

PriGo 是一类把 classifier-guided generation 迁移到机器人 primitive space 的 test-time structural adaptation 方法，真正贡献是在不重训 diffusion / flow policy 的情况下，用可微 primitive consistency 给连续动作生成加上局部操作结构约束。