arXiv:2507.09609eess.IVcs.CV2025-07被引 2

用扩散模型迭代优化相位重建,提升精度与鲁棒性。

I2I-PR: Deep Iterative Refinement for Phase Retrieval using Image-to-Image Diffusion Models

  • 以物理合理初值为基础,通过图像到图像扩散模型迭代精修。
  • 在多个数据集上显著优于经典与最新方法,重建质量提升明显。
  • 适合光学成像、显微、晶体学等需要高精度相位恢复的场景。

相位恢复旨在仅从强度测量中恢复信号,是成像、全息、光学计算、晶体学和显微等领域中的基础问题。尽管已有多种经典相位恢复算法(如基于交替投影的方法),其重建性能仍对初始值和测量噪声敏感。近期,扩散模型在图像重建任务中表现出色,带来理论与实践进展。本文提出一种深度迭代精修框架,重新定义扩散模型在相位恢复中的角色:不从随机噪声生成图像,而是从多个物理一致的初始估计出发,通过学习的图像到图像扩散过程进行迭代优化。该方法结合经典求解器优势,克服其缺陷,实现可解释且鲁棒的数据驱动相位恢复。此外,提出一种增强型初始化策略,融合经典算法与新型加速机制以获得可靠初值;推理阶段采用基于输入翻转的几何自集成策略,并结合输出聚合进一步提升重建质量。全面实验表明,本方法在训练效率与重建质量上均取得显著提升,持续优于经典及当前最先进方法。结果表明,扩散驱动精修是一种有效且通用的鲁棒相位恢复框架,适用于多样化应用场景。源代码与训练模型见 https://github.com/METU-SPACE-Lab/I2I-PR-for-Phase-Retrieval。

原文摘要 · Abstract (English)

Phase retrieval aims to recover a signal from intensity-only measurements, a fundamental problem in many fields such as imaging, holography, optical computing, crystallography, and microscopy. Although there are several well-known phase retrieval algorithms, including classical alternating projection-based solvers, the reconstruction performance often remains sensitive to initialization and measurement noise. Recently, diffusion models have gained traction in various image reconstruction tasks, yielding significant theoretical insights and practical advances. In this work, we introduce a deep iterative refinement framework that redefines the role of diffusion models in phase retrieval. Instead of generating images from random noise, our method starts with multiple physically consistent initial estimates and iteratively refines them through a learned image-to-image diffusion process. This enables data-driven phase retrieval that is both interpretable and robust, leveraging the strengths of classical solvers while mitigating their weaknesses. Furthermore, we propose an enhanced initialization strategy that integrates classical algorithms with a novel acceleration mechanism to obtain reliable initial estimates. During inference, we adopt a geometric self-ensemble strategy based on input flipping, together with output aggregation to further improve the final reconstruction quality. Comprehensive experiments demonstrate that our approach achieves substantial gains in both training efficiency and reconstruction quality, consistently outperforming classical and recent state-of-the-art methods. These results highlight the potential of diffusion-driven refinement as an effective and general framework for robust phase retrieval across diverse applications. The source code and trained models are available at https://github.com/METU-SPACE-Lab/I2I-PR-for-Phase-Retrieval

相位恢复扩散模型图像重建光学成像

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