精读笔记
Problem Setting
论文标题：LAGO Policy: Latency-Aware Asynchronous Diffusion Policies with Goal-Directed Collision-Free Planning for Smooth Manipulation（arXiv preprint / 2026）
这篇论文实际处理的是 diffusion-based visuomotor policy 从 offline imitation 到真机异步执行时暴露出的部署问题，而不是提出一个更强的 imitation learner。核心场景是：policy 每次生成一个 action chunk，但推理时间不可忽略；为了不中断执行，下一 chunk 在上一 chunk 执行过程中异步生成。这样会产生一个很具体的矛盾：policy conditioning 的 observation 来自过去，但 action chunk 在未来执行；同时上一 chunk 的未执行部分和当前 chunk 的开头在时间上应当重合，却可能由于延迟和 tracking error 发生错位。
真正困难点在于，chunked diffusion policy 的高层优势——多步、多模态、闭环重规划——恰好会在真实系统里变成不连续源。每个 chunk 在局部看都合理，但跨 chunk 拼接后可能 jerk 很大，尤其在抓取保持、倒水、插入、抽屉等对微小速度/夹爪变化敏感的任务中会放大为失败。
第二个问题是 safety。普通 GCP 学到的是 demonstration manifold 上的动作分布，并不学习 collision-free feasible set。遇到 demonstration 中没有出现的障碍时，policy 仍倾向于执行直达目标或模仿轨迹；局部 safety filter 虽然能避免碰撞，但经常把状态推到 demonstration 之外，后续 policy 不知道如何恢复。关键矛盾是：安全修正必须足够全局，才能不破坏任务意图；但又不能完全替代 learned policy，否则接触交互和细操作会丢失。
Motivation
已有 asynchronous diffusion policy 的主要补丁是 future-action conditioning：把上一轮已经预测但尚未执行的动作作为条件，让当前 chunk 与它对齐。问题是多数实现默认这个条件在训练和部署时是时间对齐的，尤其像 SAIL 那样把 future action 直接拼到 observation conditioning 里，会把一个本应表示 overlap constraint 的信号与视觉/状态历史强绑定。一旦真实部署里因为推理延迟、执行滞后或控制频率波动导致 future-action condition 有若干 step shift，模型收到的是“语义上类似但时间戳错误”的条件，反而会在 chunk boundary 制造 discontinuity。
安全方面，作者不满足于 diffusion denoising 过程中的 collision cost guidance 或安全层。核心原因是局部约束只知道当前 chunk 哪里会撞，不知道任务最终应该回到哪里；这种修正对短视避障有效，但对长距离绕障、绕开障碍后继续完成任务并不可靠。作者的关键缺口判断是：learned policy 缺少一个可供 planner 使用的 task-level attractor。没有目标，planner 只能做 reactive avoidance；有目标，planner 可以把 robot 拉回一个对 policy 后续接管友好的区域。
因此这篇的动机不是“让 diffusion model 自己学会 planning”，而是承认 diffusion policy 在几何可行性和低 jerk 执行上不擅长，把这些交给 classical trajectory optimization；同时保留 diffusion policy 对任务意图、视觉条件和 contact behavior 的建模能力。
Core Idea
这篇的核心思想可以概括为：把异步 diffusion policy 的上一轮未执行动作视作一种短期 temporal memory，但以 latency-aware 的方式使用；把 demonstration 中隐含的交互目标显式化，作为 test-time planner 的中程恢复目标。前者解决“连续性”，后者解决“安全绕障后还能回到任务”。
本质变化在于信息流重组。prior future-action conditioning 通常把未来动作当作 observation 的一部分，和视觉状态历史一起进入 policy；LAGO 把它独立为 CFG 条件，使其更像一个 soft constraint，而不是 state representation 的组成部分。再通过训练时随机 temporal shift，模型不再只学习严格对齐的 overlap matching，而是学习在一段可能延迟范围内维持 chunk consistency。这是一个很明确的 inductive bias：部署时条件时间戳不可信，因此训练时不要让模型过度依赖精确对齐。
另一个本质变化是安全规划的目标化。它没有试图让 diffusion policy 直接生成 collision-free long-horizon trajectory，也没有只在输出端做局部 clipping，而是让 policy 预测 task-relevant goal，再由 trajectory optimizer 在几何空间中生成到该 goal 的平滑避障 motion。这个设计可扩展性的来源不是模型更大，而是职责分解更合理：生成模型负责“意图/目标/局部交互”，优化器负责“几何/平滑/约束”。
Method
1. Latency-aware CFG：它解决的是异步 chunk 边界不连续。未来动作条件来自上一轮尚未执行的 action subsequence，用于约束当前 chunk 的开头。关键不是 CFG 公式本身，而是 future-action condition 不再拼进 observation history，而是作为独立条件参与 conditional/unconditional denoising difference。这样可以减少 misaligned condition 对视觉状态表征的污染。
2. Delay randomization：它解决训练-部署时间戳 mismatch。训练时从 demonstration 中以随机 offset 采样 future-action condition，让模型见过 shifted overlap。机制上这类似 domain randomization / latency augmentation，不是更强的 diffusion objective。它的价值在于把真实系统中不可控的 delay/tracking error 显式变成训练分布的一部分。
3. Goal prediction head：它解决 planner 缺少任务意图的问题。goal 从 U-Net bottleneck 表征回归，作为 task-relevant interaction target。这个 goal 不是完整 plan，也不是 symbolic subgoal，更像一个 demonstration-derived waypoint/attractor。它让避障从“远离障碍”变成“绕障后去一个 policy 熟悉的交互区域”。
4. Goal-directed collision-free trajectory generation：它解决 unseen obstacle 下 policy 输出不可行的问题。若直线路径有障碍或距离较远，就用 B-spline trajectory optimization、A* guiding path 和 collision penalty 生成 end-effector task-space trajectory。关键机制是用 classical planner 提供几何全局性，而不是依赖 diffusion sampling 的局部 cost。
5. Spatial-temporal trajectory optimization：它解决 discrete chunk 到实际执行轨迹之间的低 jerk 和限制满足问题。无论动作来自 policy 还是 planner，最终都经过连续时间优化。这个模块很可能贡献了相当一部分 smoothness gain，因此不能把所有低 jerk 结果归因给 LA-CFG。
Key Insight / Why It Works
最核心的 insight 是：异步 deployment 中的主要 failure mode 是 temporal alignment error，而不是单纯 action prediction 不准。future-action conditioning 有效，是因为它复用了上一 chunk 的 commitment，使当前 chunk 不会随视觉噪声或多模态动作分布突然切换；但如果这个 memory 的时间戳错了，强条件反而有害。LAGO 的 LA-CFG 相当于把 commitment signal 变成一个可调强度的 soft prior，并通过 shift augmentation 让模型对小范围错位有鲁棒性。这部分是论文最实质的贡献之一，属于 better inductive bias + memory reuse，而不是 scaling。
第二个有效原因是 planner 的介入点选得比较聪明。它没有在所有时刻替代 policy，而是在长距离或障碍阻挡时把机器人带到 task-relevant goal 附近，然后再交还给 policy。这避免了 local safety filter 最常见的问题：修正完虽然安全，但状态不再是 demonstration 中 policy 会处理的状态。这里的 goal 是连接 learned policy 和 planner 的 interface；没有这个 goal，planner 只是安全模块，有了它才变成 task-preserving recovery module。
第三，smoothness gain 很可能来自两个来源叠加：LA-CFG 降低 chunk boundary disagreement，spatial-temporal optimization 直接压低 jerk。文中结果显示两者都有作用，但归因不是完全干净。尤其 ISJ 的大幅下降很可能相当程度来自后处理优化，而非 diffusion policy 自身学到了更平滑的动作分布。这里需要警惕把 engineering-level trajectory smoothing 包装成 policy-level improvement。
第四，goal-directed planning 的“泛化”更像 test-time compute + classical geometry 的泛化，而不是 neural policy 对 unseen obstacle 的语义泛化。障碍不在 demonstration 中，policy 不会自动绕；是外部规划器在 task-space 中绕。这个判断并不贬低方法，反而说明其实际价值：把神经 policy 不擅长的 OOD geometry 从学习问题转成规划问题。
最可能的核心贡献排序：第一是 latency-aware future-action CFG；第二是用 demonstration-derived goal 作为 planner 接口；第三是系统层面的 trajectory optimization integration。后两者中有不少已有思想重组，但组合在 asynchronous diffusion policy deployment 中是有意义的。
Relation To Prior Work
它最接近三条线：Diffusion Policy 的 receding-horizon visuomotor control、SAIL/RTC/VLASH 这类 asynchronous/action-conditioned execution、以及 VLSA/RAIL/cost-guided diffusion 等 learned policy safety filtering。
相对 Diffusion Policy，LAGO 不是改变 generative policy 的基本建模，而是修补 deployment interface：chunk 如何连续执行、遇到障碍如何恢复、离散动作如何变成低 jerk motion。它是 diffusion policy 的系统化实用化扩展，而不是新的 action distribution model。
相对 SAIL，实质差异在 future-action condition 的语义处理。SAIL 把 future action 作为 observation conditioning 的一部分，隐含假设时间严格对齐；LAGO 把它独立为 CFG 条件，并训练时随机延迟。这不是简单换接口，而是改变了模型对 future action 的依赖方式：从状态表征的一部分变成可调 soft consistency prior。
相对 VLASH，VLASH 更关注把 robot state roll forward 到执行时刻，减少 state staleness；LAGO 更直接约束 chunk overlap consistency。两者处理的是异步误差的不同切面：一个修 observation timestamp，一个修 action commitment mismatch。
相对 collision-cost guidance / safety filter，LAGO 的新增信息是 task goal。local safety 只有 constraint，没有 task attractor；goal-directed planner 同时有 constraint 和 terminal intention。这一点是安全模块从 reactive correction 变成 task-preserving recovery 的关键。
但也要明确：B-spline optimization、A* guidance、MINCO、CFG dropout、goal regression head 都不是新思想。论文的新意主要在于把这些机制组织成一个面向 asynchronous diffusion policy 真机部署的 information-flow design。
Dataset / Evaluation
评估是强于很多纯仿真或单任务论文的：八个真实世界 manipulation tasks，覆盖 pick-place、插入、倒水、杯子转移、折叠、整理、挂胶带、螺丝分类；使用两类机械臂和多相机，确实验证了方法在真实系统中的 deployment value。任务包含 contact-rich 和 smoothness-sensitive 场景，因此对 jerk 和 chunk discontinuity 的评估是相关的。
对核心 claim 的支持大体成立：future-action shift 实验直接对应 LA-CFG 的动机；真实任务中的 CON/ISJ 支持 smoothness；加未见障碍的任务支持 goal-directed planning 比 local filter 更能保任务成功。
但 evaluation 仍有明显限制。首先，每个任务 demonstration 只有 50 条，但任务和场景似乎是 task-specific training，不是跨任务统一模型，也不是真正大规模泛化。其次，障碍实验每任务 rollout 数较少，障碍形态是否足够多样、是否存在 adversarial obstruction、是否动态变化，文中未充分说明。第三，goal extraction 的监督来源没有被充分展开；如果 goal label 带有人工设计成分，那么所谓从 demonstration 中学习 task goal 的泛化性需要打折。
此外，实验没有充分拆分 LA-CFG、goal planning、spatial-temporal optimization 对 SR、CON、ISJ 的独立贡献。尤其 smoothness 指标容易被后端优化直接改善，因此它并不能单独证明 policy 的 temporal consistency 学得更好。整体来说，实验证明这是一个有效系统，但还不足以证明每个 claimed mechanism 的贡献都干净可分。
Limitation
最主要限制是 goal 的前提很强。方法假设每个任务在当前阶段存在一个可回归的 task-relevant goal，且朝这个 goal 的 task-space 运动是合理的。这对 pick/place、插入前接近、绕障到容器附近有效；但对多阶段、强接触、需要力控探索、目标随物体交互变化的任务，单一 goal attractor 可能不够。
第二，collision model 很弱。文中使用 end-effector sphere 和 task-space obstacle avoidance，作者也承认未来需要 whole-body collision-aware planning。真实机械臂中 forearm/elbow collision、夹爪姿态、被持物体几何、线缆/环境接触都会破坏这个近似。当前方法的安全性不能理解为 full robot safety。
第三，planner 可能把问题从 learning 转移到 perception/geometry pipeline。障碍物如何检测、表示、更新，A* grid/resolution 如何设置，障碍不确定性如何处理，文中未充分说明。若感知不稳定，goal-directed planner 可能输出看似平滑但实际不可执行的轨迹。
第四，增益归因不清。LA-CFG、delay randomization、trajectory smoothing、goal planner 同时存在，很多成功率提升可能来自 test-time compute 和 classical optimization，而不是 learned policy 能力增强。尤其 ISJ 改善中，spatial-temporal optimization 是直接优化目标，属于 engineering/trajectory post-processing 成分较重。
第五，所谓 long-range planning 不等于长期状态建模。planner 只是向预测 goal 生成几何轨迹，并没有在多步交互中建模未来接触状态、对象动力学或 failure recovery。这里的 planning 更接近 waypoint reaching，不应被解读成 learned reasoning。
第六，泛化边界不清。任务是 task-specific 训练，障碍是测试时加入但仍围绕已知任务目标；如果目标外观、物体布局、任务阶段分布变化更大，goal head 是否仍能稳定给出正确 attractor，文中没有证明。核心能力可能仍主要来自 demonstration coverage + hand-designed planning interface。
Takeaway
- 1. 对 chunked generative control，异步部署的关键不是更快采样本身，而是显式处理 condition timestamp mismatch。
- future action 是很有价值的 temporal memory，但必须以 latency-aware 的 soft constraint 形式使用。
- 2. 安全 learned manipulation 不应只做 local safety projection。
- 没有 task attractor 的 safety layer 很容易把状态推离 demonstration manifold；goal-conditioned recovery 是更合理的 interface。
一句话总结
LAGO Policy 是 diffusion visuomotor policy 走向真实异步部署的一篇系统型工作，其真正贡献不是新的生成模型，而是用 latency-aware temporal memory 和 goal-conditioned classical planning 重新划分 learned policy 与优化器的职责边界。