精读笔记

Problem Setting

Spatial Attention: Adapting Execution Horizons for Diffusion Policies via Observation Sensitivity（arXiv preprint / 2026）

这篇论文实际处理的是 action chunking generative policy 的 test-time scheduling 问题：给定一个会一次生成多步动作的 policy，什么时候应该停止执行当前 chunk、重新看观测并重新采样？真正困难点不在于生成动作，而在于部署时的闭环频率分配。固定 execution horizon 把所有任务阶段都用同一个响应速度处理，导致精细操作、接触前后、目标被扰动等阶段明显欠响应，而低风险阶段又浪费推理预算。

以前方法卡在两个方向：一类是固定 horizon，简单但粗糙；另一类是 adaptive replanning，往往需要未来状态/观测 rollout 或 ensemble uncertainty，对高维视觉和实时机器人不友好。这里的关键矛盾是：diffusion/action-chunk policy 的计算瓶颈要求少采样，但机器人控制的误差积累又要求频繁闭环。论文试图给这个矛盾一个可计算的局部判据，而不是调一个全局 Ta。

Motivation

作者的核心动机是：任务中的“需要注意”的时刻不是均匀分布的。抓取、插入、接触建立、目标移动等阶段对观测误差高度敏感；而自由空间移动或冗余动作阶段，长时间 open-loop 通常可接受。固定 horizon 没有表达这种相位差异。

已有 adaptive planning 的缺口在于它们通常要预测未来状态或显式评估未来 trajectory likelihood，这与 action-only generative policy 的部署形态冲突。MoH 这类 policy-side horizon 方法虽然避免未来观测生成，但更多是动作预测分歧 heuristic，缺少一个和扰动鲁棒性直接相连的目标。本文想补的缺口是：能否从 policy 本身的局部敏感性推出一个“何时应该重新观测”的原则。

Core Idea

核心思想是把 execution horizon 不再看成固定超参数，而看成固定采样预算下的资源分配问题。每次重采样相当于花一次昂贵 inference 换取新的观测反馈；这个预算应该优先花在 policy 的动作分布对观测变化最敏感的地方。Spatial Attention 定义为 E_a[||∇_o log π(a|o)||²]，本质上是 policy 条件分布相对于观测的 Fisher-like sensitivity。

直觉上，如果一个阶段中小的观测扰动会显著改变动作分布，那么继续执行旧 chunk 会快速让当前动作变成低 likelihood 行为；反过来，如果动作分布对观测不敏感，旧 chunk 的前缀更可能仍然有效。和 prior 的本质区别是，本文不是用外部 planner 的未来 rollout，也不是用多 horizon 输出之间的分歧，而是把 horizon 调度绑定到 policy density 对 observation 的局部几何。它引入的 inductive bias 是“闭环频率应与 observation sensitivity 成正相关”。

Method

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

第一，作者把扰动下的性能退化写成 action log-likelihood drop。假设观测扰动随 open-loop 时间增长，并对 log π(a|o+δo) 做二阶展开，可得到 drop 近似正比于 Spatial Attention 和 elapsed time 的幂。这一步把“长 chunk 容易失效”变成可优化量，而不是经验判断。

第二，在固定总采样次数 N 的约束下优化时间连续的 resampling rate ρ(t)。推导结果是 ρ(t) ∝ Att(o(t))^{1/(2γ+1)}，也就是 attention 越高，重采样率越高、execution horizon 越短。这里真正改变的是 test-time compute allocation，而不是 policy architecture。

第三，实际部署时不能直接知道未来 chunk 内每一步的 observation attention，因此训练一个 forecaster，根据当前观测和已采样 action chunk 预测未来 Spatial Attention 序列；执行到累积 attention 超过阈值时重新采样。score networks 和 Bayes 分解用于估计 ∇_o log π(a|o)，视觉输入则转到 VAE latent 中估计。它们主要解决可计算性问题，不是核心思想本身。

Key Insight / Why It Works

这篇最重要的 insight 是：action chunk 的问题不是 chunk 本身，而是 chunk 的闭环边界放错了。固定 horizon 等于默认所有时间步的信息价值相同；Spatial Attention 则显式估计“新观测在此刻能改变动作分布多少”，从而把 inference budget 从低价值阶段转移到高价值阶段。

方法有效最可能来自更好的 test-time compute allocation，而不是 diffusion model 变强。policy 参数不变，平均 horizon 受控，提升主要来自把相同次数的采样放到更关键的相位。这属于 better inductive bias + test-time scheduling，而不是 scaling、retrieval 或更大数据。

最核心贡献是从 likelihood-drop 推到 attention-dependent resampling rate 的建模方式。Bayes decomposition、NCSN、VAE latent、transformer forecaster都是辅助工程；它们让指标可用，但不构成主要科学贡献。尤其 forecaster 可能隐式学到 task phase / skill boundary，这会让方法看起来像“attention-based optimal control”，但实际增益的一部分可能来自 phase segmentation。文中没有充分消融说明：若直接用接触阶段分类器、动作速度/曲率、ensemble disagreement 或 demonstration time index，能否达到类似效果。

需要警惕的是，Spatial Attention 并不等价于真实控制价值。高 ∇_o log π(a|o) 说明 policy distribution 对观测敏感，但不保证重新采样会产生更成功的闭环行为；如果 policy 在该区域本身 poorly calibrated 或 demonstration coverage 稀疏，高 attention 也可能只是模型不稳定。低 attention 也可能来自 policy 过度平滑，而不是真的安全。论文把 sensitivity 当成 responsiveness demand，这个假设合理但不是无条件成立。

Relation To Prior Work

它最接近三条线：Diffusion Policy/Consistency Policy 这类 action chunking generative policy；adaptive replanning with diffusion models；以及 minimum-attention control / Fisher sensitivity 风格的控制敏感性度量。

相对于 Diffusion Policy，本文不改变动作生成模型，而是改变 chunk 执行策略。相对于 RDM 和 uncertainty-based replanning，它避免生成未来观测/状态，因此更贴近高维视觉实时控制。相对于 Mixture of Horizons，它的区别不是“也能变 horizon”，而是 horizon 判据来自 likelihood-drop optimization，而非多 horizon 预测分歧 heuristic。

看似新的部分中，score-based estimation 和 classifier-guidance 式 Bayes 分解是已有思想重组；真正新增的信息是把 observation sensitivity 与 execution horizon 的最优分配联系起来，并将其作为 policy-agnostic 插件用于 chunking policy。它属于“test-time adaptive inference / compute allocation for robot policies”的谱系，而不是新的生成建模方法。

Dataset / Evaluation

实验设计基本对准了 claim。Robomimic 标准任务验证在常规 imitation benchmark 上不会因为动态 horizon 损害性能，且在 Square、Tool Hang 等精细操作中收益更明显；扰动版 Lift/Can 更直接测试 open-loop chunk 对外部变化的脆弱性；真机动态 cube grasping 是最强证据，因为它把“目标移动时需要更频繁重新观测”这个假设放到真实延迟和感知噪声下检验。

不过 evaluation 的覆盖仍偏窄。任务主要是单臂 manipulation，扰动形式相对简单，真机任务也基本是动态目标抓取，不足以证明在复杂 contact-rich、多物体、多阶段长程任务中的泛化。标准 Robomimic 的提升不能强证明鲁棒性，因为 benchmark 本身没有强 distribution shift。扰动实验和真机更支持核心 claim，但仍不能分清 Spatial Attention 指标本身与 forecaster 学到 task phase prior 的贡献。

Limitation

方法的核心前提是局部 likelihood geometry 能预测闭环需求。这个前提在 demonstration coverage 良好、policy calibrated、扰动局部且连续时较合理；但在 OOD 观测、遮挡、接触突变、多模态动作选择或 distribution support 边界附近，∇_o log π(a|o) 可能变成不稳定信号。

另一个限制是可扩展性。为了得到 attention 标签，需要训练额外的 observation score / action-conditioned observation score 模型；视觉场景还要借 VAE latent。对于大规模 VLA，文中只是推测可以在 backbone latent 中计算，未充分说明如何训练、如何校准、如何跨任务归一化 attention。这个成本可能比论文表述的“policy-agnostic module”更重。

方法还要求离散 action chunk，因此对 streaming policy、continuous flow execution 或高频低延迟控制不直接适用。更深层的问题是：它没有形成长期状态建模或规划能力，只是在当前 chunk 内决定何时重新采样。所谓 adaptive 不是 long-horizon reasoning，而是局部闭环频率调度。

增益归因仍不完全清楚。可能核心能力主要来自数据覆盖下的 phase-aware resampling；如果 benchmark phase structure 稳定，forecaster 可能通过动作序列识别“快到抓取/插入了”，而不是精确估计 Fisher sensitivity。文中未充分说明与更简单 phase/uncertainty/kinematic heuristics 的强对比。

Takeaway

• 1. 对 action chunking policy 来说，未来的重要方向不是只缩短 sampler，而是学习如何分配有限的重新观测机会；execution horizon 本身应成为 test-time decision variable。
• 2. Observation sensitivity 是一个值得迁移的信号：它可以用于 adaptive control frequency、selective perception、active sensing、VLA query scheduling，甚至决定何时调用大模型 backbone。
• 3. 这篇真正推动的是把“响应性”从经验超参数变成和 policy density geometry 相关的量；即使具体 Spatial Attention 估计方式被替换，这个资源分配视角也很有价值。
• 4. 下一步真正值得做的是更强的归因实验：和 phase classifier、ensemble uncertainty、action curvature、contact predictor、learned value-of-information baseline 比较，并在大规模多任务/VLA latent 中验证 attention 是否仍有一致语义。

一句话总结

这篇论文把 diffusion/action-chunk policy 的固定执行 horizon 问题重构为基于观测敏感性的 test-time 采样预算分配问题，实质贡献是给“何时重新闭环”提供了一个比启发式分歧更有理论支撑的局部几何判据。