精读笔记

Problem Setting

Stochastic Optimal Control Sampling for Diffusion Inverse Problems（arXiv preprint / 2026-06-30）关注的是用预训练扩散模型解 inverse problems 时的 test-time posterior sampling。真正困难不在于定义 y=H(x)+n，而在于：测量约束越强，越容易破坏扩散 prior 的自然采样轨迹；约束越弱，又无法保证 data consistency。此前 DPS/DAPS/DDNM/DDRM 等方法大多在采样中加入 projection 或 likelihood gradient，本质是在经验上调节 prior 与 observation 的冲突；DOC 这类 SOC 方法虽然更原则化，但 trajectory-level 控制太贵。本文试图解决的核心矛盾是：如何用一个足够便宜的控制机制，在每一步都把样本往观测一致性推，同时显式限制偏离 prior trajectory 的能量。

Motivation

作者的核心观察是：无条件 diffusion prior 本身已经能产生高质量样本，inverse problem 的 guidance 不应该被设计成无限强的约束投影，否则会把轨迹拉离模型擅长的流形。已有 posterior sampling 方法通常默认某种 γ→∞ 的硬 data consistency 倾向，只是在实现上通过步长、噪声退火、projection 经验性地缓和。已有 SOC inverse problem 方法把整条 diffusion path 当成控制对象，理论自然但计算上不适合高维图像。缺口在于：需要一个局部、step-wise、闭式的控制律，既保留 SOC 的能量-终端代价 trade-off，又避免 iLQR/Hessian/whole-trajectory optimization。

Core Idea

SOCS 的本质是把 diffusion inverse sampling 重新组织成“endpoint correction + controlled pullback”。在每个采样时刻，先沿扩散模型自己的 probability-flow ODE 预测 clean endpoint；然后在 clean space 中施加 measurement consistency；最后用线性 SDE 下的 SOC 闭式解，把这个修正后的 endpoint 映射回当前噪声层，继续后续 denoising。这样观测信息不是直接粗暴作用在 noisy state 上，而是先作用于更语义化、更接近图像空间的 clean prediction，再通过控制律分配到中间轨迹。新的 inductive bias 是：measurement correction 应该服从最小控制能量原则，而不是仅最小化数据残差。这使得方法比纯 likelihood-gradient guidance 更像一个有预算的 trajectory steering。和 prior 的本质区别在于，它不是估计 conditional score，也不是全路径优化 controller，而是在每一步构造一个局部 LQG/SOC surrogate，并用 closed-form transport 实现低成本 guidance。

Method

关键机制有三层。第一，SOC formulation：目标函数同时包含 running control energy 和 terminal measurement penalty，解决的是 prior fidelity 与 measurement fidelity 的冲突；Girsanov/KL 解释给出控制能量约束为何等价于不远离 diffusion prior。第二，closed-form step-wise controller：在线性 SDE 与二次/局部二次 terminal cost 下，Pontryagin/certainty equivalence 给出解析控制律，避免 DOC 式 trajectory-level iLQR；这将 SOC 从昂贵优化变成 sampler 内的一步 transport。第三，clean endpoint correction：Linear 版本用 affine projection/pseudoinverse 处理线性退化，Nonlinear/γ 版本用 Jacobian-aware Gauss-Newton/Langevin refinement 处理非线性退化；这些 refinement 的作用不是理论核心，而是把难解的 H-consistency 子问题转成可执行的 test-time optimization。有限 γ 是方法中最重要的旋钮：它把硬投影变成 soft terminal penalty，避免过强 data term 破坏 prior。

Key Insight / Why It Works

最值得记住的 insight 是：很多 diffusion inverse 方法的失败不是因为没有 data gradient，而是因为 data gradient 没有“控制预算”。SOCS 的有效性来自把 measurement correction 放入一个能量正则化的控制问题中，使得每次修正都要为偏离 prior trajectory 付费。有限 γ 的意义很大：γ→∞ 对应硬约束/传统强 projection，容易产生过平滑或偏离流形；有限 γ 则相当于 tempered likelihood 或 effective noise precision，让模型在观测和先验之间保持可调平衡。最可能的核心贡献是 closed-form SOC pullback + finite-γ interpretation，而不是 Langevin refinement 本身；后者与 DAPS 等方法非常接近，更多是必要工程。增益来源中有一部分明显来自 test-time compute：few-step ODE prediction、inner Langevin、更多 denoising steps、low-noise scheduler 都会贡献质量。LDM/FM 扩展的效果也可能主要来自强 foundation prior 和 latent representation alignment，SOC 只是较干净的 guidance wrapper。文中 synthetic trajectory budget 对比支持 SOCS 比 DAPS 更低能量地达到 data consistency，但在真实图像上，SOC 项和 decoupled refinement 的归因仍不完全清楚。

Relation To Prior Work

这篇最接近三条线：DPS/DAPS 类 posterior sampling、DDNM/DDRM 类 projection/pseudoinverse correction、DOC 类 diffusion optimal control。相对 DPS，它不是直接近似 conditional score，而是把数据一致性梯度解释为 SOC terminal penalty 的结果；相对 DAPS，它共享“clean prediction—inner refinement—返回轨迹”的 decoupled 思路，但多了闭式 SOC pullback 和有限 γ 控制预算；相对 DDNM/DDRM，它把线性 projection 推广到更一般的 SOC transport 与 nonlinear refinement；相对 DOC，它放弃全路径 iLQR，改成 step-wise closed-form controller，计算上更可扩展。看似新的部分中，Langevin refinement、projection、ODE clean estimate 都是已有思想重组；实质创新是用 SOC 推导把这些 correction 统一到控制能量—终端代价框架，并把 γ 从隐含无限惩罚变成显式可调变量。

Dataset / Evaluation

评估覆盖面较广：FFHQ/ImageNet 上多种线性和非线性退化，VE/VP SDE，SD v1.5 与 SD3 latent/flow-matching backbone。这个覆盖能支持“方法不是绑定单一 SDE 或单一退化”的 claim。与 DOC 的效率对比也支持其 scalability claim：同 NFE 下避免 Hessian/trajectory optimization 确实有优势。但 evaluation 仍主要是 synthetic/standard benchmark degradation，不是真实相机退化、真实噪声或物理系统；因此不能证明真实 deployment robustness。FID 用 100 validation images 虽然作者补充稳定性实验，但仍不是很强证据。表格中 SOCS 与 DAPS 在若干任务上差距并不总是压倒性，说明很多性能可能来自 decoupled posterior refinement 和 compute budget，而不是 SOC 理论本身。

Limitation

理论上限很明确：闭式解依赖线性 SDE、二次 terminal cost、可逆/良条件 normal matrix、certainty equivalence 和局部 Gauss-Newton 线性化；真实 diffusion sampler 是强非线性神经 ODE/SDE，H 也可能强非线性或不可微，因此 SOCS 实际是 SOC-inspired heuristic，而不是原问题精确最优控制。Nonlinear variant 把难点转移到 Jacobian/Langevin refinement 和 H 的可优化性上；若 H 不可微、含复杂物理 simulator、噪声模型错误或 measurement likelihood 非高斯，方法需要重新设计。有限 γ 的推荐范围很宽但解释不够闭合，因为实际还有 rt、preconditioning、step size、scheduler 共同改变 effective precision。LDM/FM 扩展中的 linear/straight-line 假设较粗，增益可能主要来自 SD prior 的数据覆盖与 latent structure，而非 SOC 本身。文中未充分说明在 out-of-distribution degradation、真实测量噪声、large-scale batch inference 下的稳定性和成本。

Takeaway

• 第一，diffusion inverse problem 中值得显式建模的不是 data consistency 本身，而是 data consistency 对 prior trajectory 的扰动成本。
• 第二，finite terminal penalty 是一个有迁移价值的设计：不要默认 γ→∞，soft constraint 往往比硬投影更符合生成先验。
• 第三，clean-space correction 再 pullback 到 noisy manifold 是一个可复用范式，适合迁移到其他 test-time conditioning、control 或 planning-style generation。
• 第四，未来真正值得做的是把 SOCS 从局部线性/Gauss-Newton heuristic 推到更真实的非线性、非高斯、物理退化场景，并做更干净的增益归因。

一句话总结

SOCS 是把 diffusion inverse posterior sampling 从经验性 data-gradient guidance 推向“带控制能量预算的 step-wise closed-form trajectory steering”的方法演化，核心贡献在于用有限 γ 的 SOC pullback 统一并约束测量一致性注入。