Abstract:The phase retrieval problem, where one aims to recover a complex-valued image from far-field intensity measurements, is a classic problem encountered in a range of imaging applications. Modern phase retrieval approaches usually rely on gradient descent methods in a nonlinear minimization framework. Calculating closed-form gradients for use in these methods is tedious work, and formulating second order derivatives is even more laborious. Additionally, second order techniques often require the storage and inversion of large matrices of partial derivatives, with memory requirements that can be prohibitive for data-rich imaging modalities. We use a reverse-mode automatic differentiation (AD) framework to implement an efficient matrix-free version of the Levenberg-Marquardt (LM) algorithm, a longstanding method that finds popular use in nonlinear least-square minimization problems but which has seen little use in phase retrieval. Furthermore, we extend the basic LM algorithm so that it can be applied for general constrained optimization problems beyond just the least-square applications. Since we use AD, we only need to specify the physics-based forward model for a specific imaging application; the derivative terms are calculated automatically through matrix-vector products, without explicitly forming any large Jacobian or Gauss-Newton matrices. We demonstrate that this algorithm can be used to solve both the unconstrained ptychographic object retrieval problem and the constrained "blind" ptychographic object and probe retrieval problems, under both the Gaussian and Poisson noise models, and that this method outperforms best-in-class first-order ptychographic reconstruction methods: it provides excellent convergence guarantees with (in many cases) a superlinear rate of convergence, all with a computational cost comparable to, or lower than, the tested first-order algorithms.

Solving Blind Ptychography Effectively Via Linearized Alternating Direction Method of Multipliers

Multi-frame Blind Deconvolution Using Sparse Priors

Fast and Robust Phase Retrieval for Masked Coherent Diffractive Imaging

Deep Distributed Optimization for Blind Diffuser-Modulation Ptychography.

Efficient and flexible approach to ptychography using an optimization framework based on automatic differentiation

An adaptive noise-blind-separation algorithm for ptychography

Convergence properties of gradient methods for blind ptychography

An alternating direction method of multipliers for inverse lithography problem

Iterative X-ray Spectroscopic Ptychography

Correction Algorithms of Pupil Aberration for Fourier Ptychographic Microscopy

A parallel alternating direction method with application to compound l1-regularized imaging inverse problems.

Parallel linearized ADMM with application to multichannel image restoration and reconstruction

A stochastic ADMM algorithm for large-scale ptychography with weighted difference of anisotropic and isotropic total variation

Wavefront Modulation with Spatial Light Modulator for Fourier Ptychographic Microscopy

Positive Absorption Constraint Based Ptychographical Algorithm with Fast Convergence Rate

Blind beam-hardening correction from Poisson measurements

A matrix-free Levenberg-Marquardt algorithm for efficient ptychographic phase retrieval

Precocious Gauge Symmetry Breaking in $SU(6) \times SU(2)_R$ Model

Blind Inversion using Latent Diffusion Priors

Ptychography-based High-Throughput Lensless On-Chip Microscopy Via Incremental Proximal Algorithms