arXiv:2605.03712stat.MLcs.LG2026-05

无需训练的扩散采样新方法,提升图像修复效率与质量

Tempered Guided Diffusion

论文配图:Tempered Guided Diffusion
图 1 · 摘自论文原文
  • 用渐进粒子法在无训练条件下高效生成条件样本
  • 相比独立多轨迹采样,计算更集中且结果更稳定
  • 适合计算资源有限但需高质量重建的逆问题场景

无需训练的条件扩散模型为任务特定模型训练提供了灵活替代方案,但现有采样器常导致计算浪费:独立引导轨迹的质量差异大,单条轨迹中后续函数评估难以纠正早期错误决策。本文提出温控引导扩散(TGD),一种基于无训练扩散先验的退火序列蒙特卡洛框架。TGD针对干净信号的温控后验分布,仅将噪声扩散状态作为辅助变量用于提议重构并传播粒子。粒子通过增量似然比重加权、重采样并在不同噪声水平间传播,使计算集中在同时满足先验和观测的合理轨迹上。在理想完全重建假设下,随着粒子数增加,完整TGD可一致逼近后验分布。对于高成本重建任务,加速版A-TGD保留早期探索能力,但在采样中途剪枝至单一高似然轨迹。在二维控制逆问题和图像逆问题上的实验表明,TGD相比独立多轨迹基线,在后验近似精度和实际耗时-质量权衡上均有显著提升。

原文摘要 · Abstract (English)

Training-free conditional diffusion provides a flexible alternative to task-specific conditional model training, but existing samplers often allocate computation inefficiently: independent guided trajectories can vary widely in quality, and additional function evaluations along a single trajectory may not recover from poor early decisions. We propose Tempered Guided Diffusion (TGD), an annealed sequential Monte Carlo framework for training-free conditional sampling with diffusion priors. TGD targets tempered posterior distributions over the clean signal, using noisy diffusion states only as auxiliary variables for proposing reconstructions and propagating particles. Particles are reweighted by incremental likelihood ratios, resampled, and propagated across noise levels, concentrating computation on trajectories plausible under both the prior and observation. Under idealized exact-reconstruction assumptions, full TGD yields a consistent particle approximation to the posterior as the number of particles grows. For expensive reconstruction tasks, Accelerated TGD (A-TGD) retains early particle exploration but prunes to a single high-likelihood trajectory partway through sampling. Experiments on a controlled two-dimensional inverse problem and image inverse problems show improved posterior approximation and favorable wall-clock speed-quality tradeoffs over independent multi-trajectory baselines.

扩散模型逆问题采样优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。