精读笔记
Problem Setting
【Where Should Action Generation Begin? A Learnable Source Prior for Generative Robot Policies】(arXiv preprint / 2026)
这篇论文真正解决的不是一个新的 manipulation policy 架构问题,而是生成式机器人策略里长期被默认掉的 q0 选择问题:动作生成到底应该从哪里开始。对于 diffusion / flow policy,source distribution 决定了生成轨迹的初始位置;标准做法从 N(0,I) 开始,相当于把“从无语义噪声到当前状态可行动作”的全部 transport 交给 generator。
困难点在于机器人动作不是无条件数据分布。action chunk 与 robot proprioception 有很强的局部几何关系:关节状态、末端位姿、夹爪状态会强约束下一段动作的合理区域。标准高斯 source 完全不利用这点,导致 generator 同时承担两个任务:先把样本搬到一个状态相关的可行动作区域,再做视觉/任务条件下的精修。以前方法的问题是,要么仍默认无信息噪声,要么把 observation-derived source 做成 deterministic point estimate,绕开了 source uncertainty。
因此这篇的关键矛盾是:生成式策略需要 stochasticity 和 iterative refinement,但机器人控制又强烈受当前状态约束;如何让随机起点既有状态信息,又不退化成确定性 policy。
Motivation
已有路线不够的地方在于,它们几乎都把注意力放在 generator dynamics 上:diffusion 还是 flow,UNet 还是 Transformer,solver 多少步,conditioning 怎么注入。但 q0 通常只是一个固定标准高斯。这个默认在图像生成里可以接受,因为模型学的是从通用噪声到数据 manifold 的大规模映射;在机器人控制里则显得低效,因为每个 control step 的 action manifold 已经被 proprioception 大幅收窄。
作者的核心观察是:source distribution 是一个独立设计轴,不应该被 generator formulation 隐含固定。A2A、VITA、BridgePolicy 已经说明 observation-derived source 有价值,但它们大多是点估计或强指定初始化,没有回答 source 本身是否应该是 distributional,也没有让 variance 随状态变化。
真正缺的是一个轻量、可插拔、仍保留 stochasticity 的 source prior:它不需要重写 generator,也不应该直接替代 generator,而是把起点放到更合理的局部区域。
Core Idea
LeaP 的核心思想是把 action generation 的源分布从 observation-independent Gaussian 改成 proprioception-conditioned diagonal Gaussian。它预测每个动作维度的均值和方差,然后从这个状态条件化分布中采样 source action chunk,再交给 flow / bridge generator 继续 refinement。
这改变的是建模分解方式。传统 flow policy 的分解是:无信息随机变量 + 强条件生成器 = 动作。LeaP 改成:本体状态给出局部 action prior + 生成器做条件修正 = 动作。这个分解更符合控制问题的结构:proprioception 决定当前动作空间的可达局部区域,视觉决定具体目标和接触细节。
和 deterministic prior 的本质区别是,LeaP 没有把 source 当成一个预测动作的点,而是当成一个带不确定性的局部 proposal distribution。它保留 generative policy 的 stochastic computation,同时减少从标准噪声到动作 manifold 的长距离 transport。这个 inductive bias 很干净,也比重做 generator 更 scalable:只改 q0,不改 solver、不改 velocity field 的接口。
Method
方法层面真正重要的机制只有几个。
第一,proprioception-conditioned source。prior 只吃 state feature,而不是完整视觉-状态融合特征。这个选择不是实现细节,而是信息瓶颈:让 prior 只负责“动作局部锚定”,避免它直接学习一个视觉条件 policy。视觉信息留给 generator,有助于维持 prior / generator 的分工。
第二,学习 mean + state-adaptive variance。mean 负责把 source 放到接近 expert action 的局部区域;variance 负责给不同状态下的随机半径。固定噪声只是 stochasticity injection,不能表达某些状态更确定、某些状态更模糊。文中的消融显示 fixed variance 明显不如 adaptive variance,这说明关键不是“加噪声”,而是“学 uncertainty”。
第三,prior 有显式监督。NLL 把 expert action chunk 看成来自该 Gaussian 的样本,直接约束 source distribution 的密度位置;contrastive alignment 进一步约束 sampled source 与 paired action 的实例级对应;flow loss 则让 source 对后续 transport 友好。这里的机制含义是:prior 不是仅通过 downstream generator 间接学出来的 latent,而是被显式拉到 action manifold 附近。
第四,generator 保持不变。这点很重要,因为它把贡献归因集中在 source distribution 上,而不是又引入一个更大的 backbone 或 solver trick。
Key Insight / Why It Works
最可能的核心贡献是:把生成式机器人策略中的“全局生成”问题局部化。标准高斯到动作分布的 transport 在机器人控制中往往是不必要的难题,因为当前 state 已经提供了强先验。LeaP 等价于给 generator 一个 learned warm start,使 iterative computation 主要用于 residual refinement,而不是粗定位。
这更像 better inductive bias,而不是 scaling。prior head 只有很小参数量,性能提升主要来自结构性分解:state-conditioned proposal + conditional generator。它也有一点 curriculum 的味道:训练时 generator 看到的 source 已经更接近 action,velocity regression 的目标更短、更局部,优化更容易,推理时少量 NFE 也更够用。
它不是 retrieval,但有 implicit local memory 的性质:NLL-supervised prior 会学习 demonstration 中“某类 proprioceptive state 对应的动作局部分布”。在 50 demos/task 的设置下,这种局部动作先验可能非常有效。需要警惕的是,这也意味着一部分性能可能来自 supervised action prior 的记忆/拟合,而不完全是 generative dynamics 更强。尤其在 RoboTwin 的任务分布较窄、每个任务 demo 有限但状态-action 规律强时,prior 很可能学到的是局部行为模板。
contrastive alignment 是否是本质贡献不够清楚。消融显示去掉会掉点,但它可能主要是 regularization 或 batch-level pairing signal,不一定是必要的理论机制。更关键的是 NLL + adaptive variance。full covariance / GMM 不提升也说明问题不在 prior expressivity,而在正确的 inductive bias:低容量、状态条件化、可校准随机半径。
state-only prior 优于 vision-conditioned prior 是这篇最有迁移价值的 insight。直觉上,视觉输入进 prior 会让 source 承担过多语义/场景解释责任,容易把初始分布拉偏;而 proprioception 给出的约束更稳定、更低噪。换句话说,source prior 应该编码 control geometry,而不是 scene semantics。
Relation To Prior Work
这篇最接近的谱系是 diffusion / flow visuomotor policy 中的 source initialization 设计,而不是一个新的 policy backbone。和 Diffusion Policy、FlowPolicy、π0 类方法相比,它没有改变生成动力学,而是挑战 q0=N(0,I) 这个默认项。
和 A2A / VITA 的差异在于,后者把 observation-derived source 当成确定性 latent 或点估计,LeaP 显式建模 source distribution,并且推理时保持随机性。这是实质区别:deterministic source 本质上是 warm-start point,LeaP 是 learnable proposal distribution。
和 BridgePolicy 的差异在于,BridgePolicy 把 source 与 diffusion bridge dynamics 耦合在一起,方法贡献难拆;LeaP 刻意把 source prior 做成 plug-in,使其成为独立设计轴。这个归因更干净。
从更宽的生成模型视角看,learned noise / semantic initialization 在图像生成里并不新;这篇的新意在于把它放到 closed-loop robot policy 中,并发现 proprioception-conditioned adaptive Gaussian 这个低复杂度形式已经足够有效。看似新的部分有一部分是已有思想重组,但“source distribution as reusable policy design axis”在机器人生成策略里是实质推进。
Dataset / Evaluation
评估主要在 RoboTwin 2.0 的 15 个桌面双臂 manipulation 任务上,另有三个 Franka 真机任务。任务覆盖 pick-place、工具使用、articulated object、双臂协作等,足以验证短 horizon visuomotor manipulation 中 source prior 的有效性。最有说服力的是 architecture-matched NoPrior 对照,因为它基本隔离了 q0 的影响。
不过 evaluation 支持的是一个相对限定的 claim:在 tabletop manipulation、action chunk imitation、有限 NFE flow/bridge generator 下,learned proprioceptive source prior 能提升效率和成功率。它还不能证明该机制对 VLA、大规模多任务、语言条件、长程规划或强视觉语义任务同样成立。
真机实验有价值,但规模较小,只能说明趋势没有完全停留在仿真。BridgePolicy 无官方代码,重实现可能影响公平性;不过主要结论并不依赖它,因为 NoPrior/A2A/VITA 的对照已经足够说明 learned distributional source 的价值。
Limitation
核心前提是 proprioception 对下一段 action chunk 有足够强的解释力。这个前提在桌面低层控制中成立,但在视觉决定动作模式的任务中可能失效。例如多个可选目标、语言指定对象、隐含任务阶段切换、遮挡下的语义推断,state-only prior 可能只给出一个平均的局部动作倾向,甚至错误收缩多模态性。
方法也可能把难题从 generator 转移到 prior:通过 NLL 直接拟合 expert actions,本质上引入了一个额外的 supervised action density head。文中说 direct mean execution 为 0%,说明 prior 不能单独控制;但这不排除 prior 已经承担了大量低频动作结构,generator 只做 residual correction。这个分工是优点,也是归因上的不清晰。
diagonal Gaussian 是强限制。论文中 full covariance / GMM 没提升,但这不代表相关结构不重要,更可能是数据量、优化、参数化或监督方式不适合。对于 dexterous hand 或高维 humanoid action,独立维度方差可能过于粗糙。
泛化能力仍未被真正验证。RoboTwin 的 domain randomization 和 50 demos/task 可以测试一定鲁棒性,但不能排除 benchmark 内 state-action coverage 足够,使得 prior 学到近似局部模板。所谓更快收敛也可能部分来自更强的 supervised warm start,而不是生成模型本身获得了更强泛化。
此外,variance calibration 文中未充分说明。成功率提升不等于不确定性估计准确;state-adaptive variance 可能只是学到了有利于训练的噪声尺度,而非真实 epistemic / aleatoric uncertainty。
Takeaway
- 1. 生成式机器人策略里 q0 是被低估的设计轴;与其一味加大 generator,不如先让生成从更合理的 state-conditioned region 开始。
- 2. 对低层控制,proprioception 可能比视觉更适合作为 source prior 的条件信号;视觉应更多进入 refinement / disambiguation,而不是决定初始噪声分布。
- 3. 随机性本身不够,关键是 state-adaptive uncertainty。
- fixed noise 往往只是扰动,learned variance 才能成为结构化 proposal。
一句话总结
这篇论文把生成式机器人策略的改进点从“设计更强 generator”转向“学习更合理的动作生成起点”,其真正贡献是将 proprioception-conditioned stochastic source prior 确立为一个低成本但有效的独立建模轴。
