PtyGenography: using generative models for regularization of the phase retrieval problem
Selin Aslan, Tristan van Leeuwen, Allard Mosk, Palina Salanevich
TL;DR
The paper addresses phase retrieval from magnitude-only measurements and analyzes how generative priors can regularize the inverse problem at the cost of reconstruction bias. It derives error bounds for conventional and generative formulations and introduces a unified variational objective that blends data fidelity with a generative regularizer, enabling adaptive trade-offs between bias and variance. Numerical experiments on masked Fourier measurements with a PCA-trained generator show that the combined approach often outperforms the individual formulations across noise regimes, while pure generative priors can introduce bias on out-of-distribution signals. This work provides a practical framework for robust phase retrieval using data-driven priors and highlights avenues for adaptive regularization and theoretical tightening of bounds.
Abstract
In phase retrieval and similar inverse problems, the stability of solutions across different noise levels is crucial for applications. One approach to promote it is using signal priors in a form of a generative model as a regularization, at the expense of introducing a bias in the reconstruction. In this paper, we explore and compare the reconstruction properties of classical and generative inverse problem formulations. We propose a new unified reconstruction approach that mitigates overfitting to the generative model for varying noise levels.
