Uncertainty quantification for ptychography using normalizing flows

Abstract

Ptychography, as an essential tool for high-resolution and nondestructive material characterization, presents a challenging large-scale nonlinear and non-convex inverse problem; however, its intrinsic photon statistics create clear opportunities for statistical-based deep learning approaches to tackle these challenges, which has been underexplored. In this work, we explore normalizing flows to obtain a surrogate for the high-dimensional posterior, which also enables the characterization of the uncertainty associated with the reconstruction: an extremely desirable capability when judging the reconstruction quality in the absence of ground truth, spotting spurious artifacts and guiding future experiments using the returned uncertainty patterns. We demonstrate the performance of the proposed method on a synthetic sample with added noise and in various physical experimental settings.

arXiv commentAccepted at the Fourth Workshop on Machine Learning for Physical Sciences, NeurIPS 2021

References in corpus (16)

• NICE: Non-linear Independent Components Estimation
• Variational Inference with Normalizing Flows
• Normalizing Flows: An Introduction and Review of Current Methods
• Phase recovery and holographic image reconstruction using deep learning in neural networks
• Physics-Constrained Deep Learning for High-dimensional Surrogate Modeling and Uncertainty Quantification without Labeled Data
• Density estimation using Real NVP
• Guided Image Generation with Conditional Invertible Neural Networks
• Flow-based generative models for Markov chain Monte Carlo in lattice field theory
• Reliable deep-learning-based phase imaging with uncertainty quantification
• Likelihood-informed dimension reduction for nonlinear inverse problems
• Analyzing Inverse Problems with Invertible Neural Networks
• Nested Sampling with Normalising Flows for Gravitational-Wave Inference
• Solving inverse problems using conditional invertible neural networks
• Latent Normalizing Flows for Discrete Sequences
• The CRLB and Maximum likelihood in ptychography with Poisson noise
• Deep Probabilistic Imaging: Uncertainty Quantification and Multi-modal Solution Characterization for Computational Imaging