arXiv:2605.26470cs.CV2026-05

优化扩散模型采样三要素调度,提升图像逆问题重建质量

Triadic Dynamics Aware Diffusion Posterior Sampling for Inverse Problems: Optimizing Guidance and Stochasticity Schedules

论文配图:Triadic Dynamics Aware Diffusion Posterior Sampling for Inverse Problems: Optimizing Guidance and Stochasticity Schedules
图 1 · 摘自论文原文
  • 将后验采样建模为时变控制问题,动态调整三要素调度策略
  • 在多个图像重建任务中,显著提升数据保真度与感知真实感
  • 适合需要高质量图像重建的科研与医疗影像应用

使用扩散模型进行生成式后验采样已成为解决成像逆问题的主要范式,通常包含三个核心组件:数据一致性(DC)引导、无分类器引导(CFG)和随机性。尽管已有研究聚焦于各组件的设计或联合优化,但对它们调度方式的关注不足,导致采用启发式固定或部分调整的次优调度。本文指出,三者调度间的相互作用对性能提升至关重要。分析表明,早期过强的CFG与DC引导冲突,而随机性有助于轨迹回归到高概率区域。基于此,提出三角动力感知后验采样(TriPS),将后验采样重构为时变控制问题,遵循DC与随机性尺度递减、CFG尺度递增的三元趋势。通过基于函数先验的模板搜索获得可靠基线调度,并利用组相对策略优化(GRPO)实现更灵活的时间曲线。实验表明,TriPS在数据保真度和感知真实感上均优于现有最优基线。

原文摘要 · Abstract (English)

Generative posterior sampling using diffusion models has emerged as a dominant paradigm for solving inverse problems in imaging, which usually consists of three main components: data consistency (DC) guidance, classifier-free guidance (CFG) and stochasticity. While prior arts have focused on how to develop each or all components, less attention has given to how to schedule them, leading to heuristically fixed or partially adjusted suboptimal schedules. In this work, we argue that the interactions among all three components in terms of scheduling are crucial for significantly improved performance in solving inverse problems in imaging. Our analysis shows that aggressive CFG early in sampling conflict with DC guidance, while stochasticity brings the trajectory back to higher-probability regions. Based on these findings, we propose Triadic Dynamics Aware Posterior Sampling (TriPS), which reformulates posterior sampling as a time-varying control problem and optimizes schedules following a triadic trend of decreasing DC and stochasticity scales alongside increasing CFG scale. TriPS achieves this through two strategies: template-based search over functional priors for reliable baseline schedules, and Group Relative Policy Optimization (GRPO)-based reinforcement learning for more flexible temporal curves. Experiments demonstrate TriPS outperforms state-of-the-art baselines in data fidelity and perceptual realism.

扩散模型图像重建逆问题调度优化

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