Overlapping domain decomposition methods for ptychographic imaging

Huibin Chang*, Roland Glowinski, Stefano Marchesini, Xue Cheng Tai, Yang Wang, Tieyong Zeng

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

In ptychography experiments, redundant scanning is usually required to guarantee the stable recovery, such that a huge number of frames are generated, and thus it poses a great demand on parallel computing to solve this large-scale inverse problem. In this paper, we propose the overlapping domain decomposition methods to solve the nonconvex optimization problem in ptychographic imaging. They decouple the problem defined on the whole domain into subproblems only defined on the subdomains with synchronizing information in the overlapping regions of these subdomains, thus leading to highly parallel algorithms with good load balance. More specifically, for the nonblind recovery (with known probe in advance), by enforcing the continuity of the overlapping regions for the image (sample), the nonlinear optimization model is established based on a novel smooth-truncated amplitude-Gaussian metric (ST-AGM). Such a metric allows for fast calculation of the proximal mapping with closed form, and meanwhile provides the possibility for the convergence guarantee of the first-order nonconvex optimization algorithm due to its Lipschitz smoothness. Then the alternating direction method of multipliers is utilized to generate an efficient overlapping domain decomposition based ptychography algorithm (OD2P) for the two-subdomain domain decomposition (DD), where all subproblems can be computed with closed-form solutions. Due to the Lipschitz continuity for the gradient of the objective function with ST-AGM, the convergence of the proposed OD2P is derived under mild conditions. Moreover, it is extended to more general cases including multiple-subdomain DD and blind recovery. Numerical experiments are further conducted to show the performance of proposed algorithms, demonstrating good convergence speed and robustness to the noise. Especially, we report the virtual wall-clock time of proposed algorithm up to 10 processors, which shows potential for upcoming massively parallel computations.

Original languageEnglish
Pages (from-to)B570-B597
JournalSIAM Journal on Scientific Computing
Volume43
Issue number3
DOIs
Publication statusPublished - May 2021

Scopus Subject Areas

  • Computational Mathematics
  • Applied Mathematics

User-Defined Keywords

  • Blind recovery
  • Overlapping domain decomposition method
  • Parallel computing
  • Phase retrieval
  • Ptychography
  • Smooth truncated amplitude-Gaussian metric

Fingerprint

Dive into the research topics of 'Overlapping domain decomposition methods for ptychographic imaging'. Together they form a unique fingerprint.

Cite this