arXiv:2512.07150cs.LGcs.AI2025-12被引 5

提出FlowLPS,用采样与优化结合提升图像逆问题重建质量

FlowLPS: Langevin-Proximal Sampling for Flow-based Inverse Problem Solvers

  • 结合朗之万采样与近端优化,分步提升重建精度
  • 在五个线性逆问题上实现测量一致性与视觉质量的平衡
  • 适用于无需训练的生成模型逆问题求解,尤其适合高保真重建

深度生成模型是成像逆问题的强大先验,但无训练的潜在空间流模型求解器面临有限步数的权衡。依赖优化的方法虽能快速提高测量一致性,但在高度非线性的潜在空间中,结果对局部优化初始点敏感,常导致感知真实感下降。相反,随机采样方法更利于后验探索,但需大量迭代才能获得清晰且符合测量的重构结果。为此,我们提出FlowLPS,一种基于朗之万-近端采样的无训练潜在流逆问题求解器。每一步反向过程中,FlowLPS利用若干朗之万更新,沿后验方向扰动模型预测的干净估计,为局部优化提供随机初始化;随后应用类似最大后验的近端精修,从朗之万更新后的估计出发快速提升测量一致性。此外,通过受控的pCN式去噪稳定反向轨迹,同时保持轨迹连贯性。在FFHQ和DIV2K数据集上针对五种线性逆问题的实验表明,FlowLPS在测量保真度与感知质量之间实现了良好平衡,并在像素空间逆问题与相位恢复任务中进行了额外验证。

原文摘要 · Abstract (English)

Deep generative models are powerful priors for imaging inverse problems, but training-free solvers for latent flow models face a practical finite-step trade-off. Optimization-heavy methods quickly improve measurement consistency, but in highly nonlinear latent spaces, their results can depend strongly on where local refinement is initialized, often degrading perceptual realism. In contrast, stochastic sampling methods better preserve posterior exploration, but often require many iterations to obtain sharp, measurement-consistent reconstructions. To address this trade-off, we propose FlowLPS, a training-free latent flow inverse solver based on Langevin-Proximal Sampling. At each reverse step, FlowLPS uses a few Langevin updates to perturb the model-predicted clean estimate in posterior-oriented directions, providing stochastic initializations for local refinement. It then applies local MAP-style proximal refinement to rapidly improve measurement consistency from the Langevin-updated estimate. We additionally use controlled pCN-style re-noising to stabilize the reverse trajectory while retaining trajectory coherence. Experiments on FFHQ and DIV2K across five linear inverse problems show that FlowLPS achieves a strong balance between measurement fidelity and perceptual quality, with additional experiments on pixel-space inverse problems and phase retrieval.

图像重建生成模型逆问题采样优化

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