PrNet: Efficient and Robust Phase Retrieval via Stochastic Refinement

2025-01-01
Kaya, Mehmet Onurcan
Öktem, Sevinç Figen
Phase retrieval is a fundamental inverse problem that arises in many scientific and engineering disciplines, where the goal is to reconstruct a signal from intensity-only measurements. In this work, we develop prNet, a novel phase retrieval framework that stochastically refines initial reconstructions with learned denoising and model-based updates. Our framework combines Langevin dynamics-based posterior sampling, adaptive noise schedule learning, warm-start initialization from classical solvers, and a progressive training strategy inspired by algorithm unrolling. By considering the perception-distortion tradeoff, our method also mitigates the over-smoothing effects commonly observed in prior approaches and enables reconstructions with fine details while minimizing artifacts. Extensive experiments demonstrate that our approach achieves state-of-the-art performance in both efficiency and reconstruction quality.
35th IEEE International Workshop on Machine Learning for Signal Processing, MLSP 2025
Citation Formats
M. O. Kaya and S. F. Öktem, “PrNet: Efficient and Robust Phase Retrieval via Stochastic Refinement,” presented at the 35th IEEE International Workshop on Machine Learning for Signal Processing, MLSP 2025, İstanbul, Türkiye, 2025, Accessed: 00, 2025. [Online]. Available: https://www.scopus.com/inward/record.uri?partnerID=HzOxMe3b&scp=105022071991&origin=inward.