Unrolled Wirtinger Flow with Deep Decoding Priors for Phaseless Imaging

Abstract

We introduce a deep learning (DL) based network and an associated exact recovery theory for imaging from intensity-only measurements. The network architecture uses a recurrent structure that unrolls the Wirtinger Flow (WF) algorithm with a deep decoding prior that enables performing the algorithm updates in a lower dimensional encoded image space. We use a separate deep network (DN), referred to as the encoding network, for transforming the spectral initialization used in the WF algorithm to an appropriate initial value for the encoded domain. The unrolling scheme models a fixed number of iterations of the underlying optimization algorithm into a recurrent neural network (RNN). Furthermore, it facilitates simultaneous learning of the parameters of the decoding and encoding networks and the RNN. We establish a sufficient condition to guarantee exact recovery under deterministic forward models. Additionally, we demonstrate the relation between the Lipschitz constants of the trained decoding prior and encoding networks to the convergence rate of the WF algorithm. We show the practical applicability of our method in synthetic aperture imaging using high fidelity simulation data from the PCSWAT software. Our numerical study shows that the decoding prior and the encoding network facilitate improvements in sample complexity.

References in corpus (10)

• Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising
• Phase Retrieval via Wirtinger Flow: Theory and Algorithms
• Relaxed Averaged Alternating Reflections for Diffraction Imaging
• Spectral Norm Regularization for Improving the Generalizability of Deep Learning
• Optimal Spectral Initialization for Signal Recovery with Applications to Phase Retrieval
• Deep Learning for Passive Synthetic Aperture Radar
• Algorithmic Guarantees for Inverse Imaging with Untrained Network Priors
• A Deterministic Theory for Exact Non-Convex Phase Retrieval
• A Generalization of Wirtinger Flow for Exact Interferometric Inversion
• Deep Learning for Waveform Estimation and Imaging in Passive Radar

Cited by in corpus (1)

• PtychoDV: Vision Transformer-Based Deep Unrolling Network for Ptychographic Image Reconstruction