prNet: Data-Driven Phase Retrieval via Stochastic Refinement
IEEE Transactions on Computational Imaging, 2026 (SCI-Expanded, Scopus)
- Yayın Türü: Makale / Tam Makale
- Basım Tarihi: 2026
- Doi Numarası: 10.1109/tci.2026.3719489
- Dergi Adı: IEEE Transactions on Computational Imaging
- Derginin Tarandığı İndeksler: Science Citation Index Expanded (SCI-EXPANDED), Scopus, Aerospace Database, Compendex, INSPEC, MathSciNet, Technology Collection (ProQuest)
- Anahtar Kelimeler: deep learning, diffusion models, nonlinear inverse problems, Phase retrieval
- Orta Doğu Teknik Üniversitesi Adresli: Evet
Özet
Phase retrieval is an ill-posed inverse problem in which existing methods often struggle to balance reconstruction fidelity and perceptual quality. We propose a novel Langevin-based framework for phase retrieval that bridges model-driven and data-driven approaches by combining stochastic posterior sampling, learned denoising, and model-based updates to better navigate the perception-distortion tradeoff. The framework consists of three progressively more advanced variants. The first, prNet-Small, provides a theoretically grounded and lightweight pipeline that integrates Langevin dynamics with learned noise and denoising processes and alternating projection-based updates. Building upon this foundation, prNet-Large exploits parallel sampling of diverse reconstructions and aggregates them to approximate the MMSE solution, significantly improving distortion metrics. Finally, prNet-Large-Adversarial enhances perceptual quality by replacing simple averaging with an adversarially trained aggregation network while preserving reconstruction fidelity. In addition, we incorporate test time augmentation to further improve the performance. Extensive experiments demonstrate that the proposed framework consistently outperforms classical, deep learning-based, and diffusion-based baselines across multiple benchmarks in both reconstruction fidelity and perceptual quality.