Stochastic Multigrid Method for Blind Ptychographic Phase Retrieval
Borong Zhang, Junjing Deng, Yi Jiang, Zichao Wendy Di
TL;DR
This work tackles blind ptychographic phase retrieval, where both the object $\bm{z}$ and the probe $\bm{Q}$ must be recovered from phaseless diffraction data. It introduces eMAGPIE, a stochastic multigrid framework that minimizes a majorizing quadratic surrogate $\widetilde{\Phi}(\bm{Q},\bm{z};j)$ of the exit-wave misfit, with a guaranteed descent at each iteration. A joint object–probe update is derived by combining one-variable minimizers through a geometric-mean, phase-aligned rule, and the reconstruction is accelerated by a multilevel surrogate (MAGPIE) that couples fine- and coarse-grid problems. Across simulated and real datasets, the method achieves lower data misfit and phase error, produces smoother, artifact-reduced phase reconstructions, and demonstrates robustness to lower overlaps and higher noise than prior rPIE-based approaches.
Abstract
We present eMAGPIE (extended Multilevel-Adaptive-Guided Ptychographic Iterative Engine), a stochastic multigrid method for blind ptychographic phase retrieval that jointly recovers the object and the probe. We recast the task as the iterative minimization of a quadratic surrogate that majorizes the exit-wave misfit. From this surrogate, we derive closed-form updates, combined in a geometric-mean, phase-aligned joint step, yielding a simultaneous update of the object and probe with guaranteed descent of the sampled surrogate. This formulation naturally admits a multigrid acceleration that speeds up convergence. In experiments, eMAGPIE attains lower data misfit and phase error at comparable compute budgets and produces smoother, artifact-reduced phase reconstructions.
