AI & ML
prNet: Data-Driven Phase Retrieval via Stochastic Refinement
arXiv eess.IV0 views
arXiv:2507.09608v3 Announce Type: replace Abstract: 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. The source code and trained models are available at https://github.com/METU-SPACE-Lab/prNet-for-Phase-Retrieval
Source:https://arxiv.org/abs/2507.09608PreviousA Cytology Dataset for Early Detection of Oral Squamous Cell CarcinomaNextStreamAZ: Line-by-Line SAR Focusing with State Space Models
Related posts
Hypergraph Generation via Structured Stochastic Diffusion
0arXiv stat.CO
Sparse Probabilistic Richardson Extrapolation
0arXiv stat.CO
Dynamic Bayesian Predictive Stacking via Markovian Spatiotemporal Propagation
0arXiv stat.CO
Coherent forecast combination for linearly constrained multiple time series
0arXiv stat.CO
New optimisations for exact, scalable 2D Gaussian process inference, with applications to astronomical time-series
0arXiv stat.CO
A tractable transition function for the Wright-Fisher diffusion with selection via gamma duality
0arXiv stat.CO