Arxiv:2007.14139V2 [Physics.Optics] 1 Apr 2021 Plane and the Detector Plane
Total Page:16
File Type:pdf, Size:1020Kb
Accelerating ptychographic reconstructions using spectral initializations Lorenzo Valzania,1, 2, ∗ Jonathan Dong,1, 3, ∗ and Sylvain Gigan1 1Laboratoire Kastler Brossel, Ecole´ Normale Sup´erieure - Paris Sciences et Lettres (PSL) Research University, Sorbonne Universit´e,Centre National de la Recherche Scientifique (CNRS) UMR 8552, Coll`egede France, 24 rue Lhomond, 75005 Paris, France 2Laboratory for Transport at Nanoscale Interfaces, Empa, Swiss Federal Laboratories for Materials Science and Technology, 129 Uberlandstrasse, Dubendorf 8600, Switzerland 3Laboratoire de Physique de l'Ecole´ Normale Sup´erieure - Paris Sciences et Lettres (PSL) Research University, Sorbonne Universit´e,CNRS, Universit´ede Paris, 24 rue Lhomond, 75005 Paris, France (Dated: April 2, 2021) Ptychography is a promising phase retrieval technique for label-free quantitative phase imaging. Recent advances in phase retrieval algorithms witnessed the development of spectral methods, in order to accelerate gradient descent algorithms. Using spectral initializations on experimental data, for the first time we report three times faster ptychographic reconstructions than with a standard gradient descent algorithm and improved resilience to noise. Coming at no additional computational cost compared to gradient-descent-based algorithms, spectral methods have the potential to be implemented in large-scale iterative ptychographic algorithms. I. INTRODUCTION and the acquisition parameters. The quest for improved reconstruction performance culminated in the derivation Ptychography is a computational imaging technique of optimal spectral methods for the random measure- that enables label-free, quantitative phase imaging [1]. ments setting [9, 10]. Due to these breakthroughs, spec- It is based on a simple principle: scan a probe across tral methods for phase retrieval are rather well under- a sample, collect the corresponding intensity diffraction stood theoretically. patterns (also known as 'ptychograms'), and reconstruct Although spectral initializations to solve ptychography an image of the object of interest. Because it does not have been showcased on simulated measurements [11], no require complex optical elements, it has been adapted to gain was ever reported when employed on experimental a variety of settings and spectral ranges, from electron data. Here, we provide the first experimental demon- microscopy [2], for which it was originally conceived in stration of ptychographic reconstructions improved and the late 60s [3], to X-rays [4] and extreme ultra-violet accelerated by spectral initializations. light [5], all the way down to the terahertz (THz) range [6]. The computational reconstruction in ptychography re- II. METHODS quires to solve a phase retrieval problem, where the phase of the diffracted electric field has to be recovered Let us begin by describing a ptychogram y(l)(x) with from intensity-only measurements. Such a problem is the following forward model tractable if a minimum overlap ratio (empirically esti- (l) (l) 2 mated around 60% [7]) is ensured between subsequent y (x) = jPdfa(x − x ) (x)gj ; (1) probe positions. However, as a non-linear and non- where x is a two-dimensional spatial coordinate, (x) is convex optimization problem, it is still not completely the complex transmission function of the object, scanned understood and convergence towards its global minimum with the probe a(x) at the positions x(l) for l = 1;:::;L, is not guaranteed. with L being the total number of images. Pd is a known One way to avoid local minima is to provide an ini- linear operator describing the transmission through the tial estimate already close to the solution, and further optical system of optical length d between the object refine it with iterative algorithms. To this end, spec- arXiv:2007.14139v2 [physics.optics] 1 Apr 2021 plane and the detector plane. We hereby point out that tral methods have recently been proposed for the general the model equally applies to Fourier ptychography [12], phase retrieval problem with independent and identically if one regards (x) and a(x) as the Fourier transform of distributed (i.i.d.) random measurements, solved with a the object transmission and the probe functions respec- gradient descent (GD) approach [8]. The initial estimate tively, and Pd is an inverse Fourier transform. In what is defined as the leading eigenvector of a covariance ma- follows the case of ptychography will be considered, with trix, constructed from the experimental measurements a setup sketched in the top row of Fig. 1. To recover the object, ptychography solves a phase re- trieval problem that can be formulated in the vectorized form ∗ These authors equally contributed to this work. Corresponding author: [email protected] y = jS j2 : (2) 2 the bottom row of Fig. 1. Probe Object Propagator Ptychograms In order to apply spectral methods to ptychography, the forward model in the continuous domain of Eq. (1) needs to be brought into the discrete vectorized form of Eq. (2). Let a(l) 2 Cm×n be the discrete and vector- (l) m×m ized form of a(x − x ), and Pd 2 C be the matrix computing the linear transform Pd. Now we can adopt the matrix factorization used in [11] and finally write y = jS j2, with S ≡ PA and 2 3 2 (1) 3 Pd ··· 0 a 6 . .. 7 Lm×Lm 6 . 7 Lm×n P ≡ 4 . 5 2 C ; A ≡ 4 . 5 2 C ; (L) 0 ··· Pd a (4) from which, using Eq. (3) and the unitarity of P (i.e., P y = P −1), one obtains Leading eigenvector of 풁 Iterative optimization (GD, PIE, etc.) Z = AyP −1 diagfT (y)gPA : (5) FIG. 1. Sketch of a ptychography setup and algorithmic Our choice of the preprocessing function, reading pipeline with our spectral method. Top: an aperture gen- 0 0 erates the probe a(x), that is scanned with shifts x(l) across T (y ) ≡ max(−β; 1 − 1=y ) ; (6) an object of interest (x), for l = 1;:::;L. For each probe 0 position, a camera records a ptychogram y(l)(x) after free where y collects the ptychograms, each normalized to space propagation by a distance d. Middle: the ptychograms its own mean intensity, and 0 < β < 1, was inspired 0 are preprocessed and used to construct a weighted covariance by the theoretical works in [9, 10, 14]. For y & 1, matrix Z. Bottom: the spectral estimate, obtained as the the function corresponds to the optimal preprocessings leading eigenvector of Z, is already close to the solution and of [9, 10]. The lower bound set at low intensity values is further refined using an iterative optimization (dark blue prevents large negative eigenvalues of Z, that will de- point). Initial estimates further away from the solution (light grade the performance of the power iteration algorithm. blue point) may be stuck in local minima. GD: gradient de- It is indeed known that power iterations return the eigen- scent; PIE: ptychographic iterative engine. vector corresponding to the largest eigenvalue in absolute value. In addition, it reminds of the truncated amplitude flow algorithm, whose resilience to noise has been proved Here, 2 n represents the vector of the unknown C [14]. After plugging Eq. (6) into Eq. (3), the power object transmission coefficients at each of its n pixels, method resembles a GD update, proving that no com- S 2 p×n is the sensing matrix, with p being the num- C putational complexity is added when replacing one GD ber of measured intensity values (i.e., p = Lm, where m iteration with one power iteration (see Appendix B). is the number of camera pixels), y 2 p is the vector R In the box Algorithm 1 we summarize the overall al- collecting the ptychograms, and j · j denotes the element- gorithm, consisting of the combination of a spectral ini- wise modulus operation. tialization and a GD optimization with amplitude loss When the entries of S are i.i.d. random variables, p function L( ) ≡ k y −jS jk2, where k·k indicates the is initially estimated as the leading eigenvector of the 2 2 L2-norm. The corresponding code is available at Ref. following n × n matrix (Fig. 1, middle row): [15], and more details on a memory-efficient implemen- Z ≡ Sy diagfT (y)gS ; (3) tation of power iterations are outlined in Appendix C. where Sy denotes the Hermitian conjugate of the ma- trix S, T is a preprocessing function acting element-wise III. EXPERIMENTS on the measurements, and diagfug denotes a diagonal matrix with a general vector u on its main diagonal. A Experiments were performed using the THz imaging widely used method to compute the leading eigenvector is setup at Empa, the Swiss Federal Laboratories for Ma- power iterations, that converge exponentially towards the terials Science and Technology, equipped with a far- eigenvector corresponding to the largest eigenvalue in ab- infrared gas laser (FIRL 100, Edinburgh Instruments, solute value [13]. This estimate is finally used as the first Livingston, Scotland) emitting several tens of mW of estimate of an iterative optimization algorithm like gra- continuous-wave power at the wavelength λ = 96.5 µm. dient descent or, in the case of ptychography, dedicated An uncooled microbolometer array detector featuring m routines like the ptychographic iterative engine (PIE) [1]. = 480 × 640 pixels on a pitch of 17 µm (Gobi-640-GigE, The reconstruction pipeline is graphically summarized in Xenics, Leuven, Belgium), was used as a THz camera 3 Algorithm 1: Solve ptychography with a of the power method, which can be explained by com- spectral initialization paring one iteration of the power method with one GD Input: Measurements y, sensing matrix S, iteration (see Appendix B). When such a decrease came to a halt, power iterations were stopped, and the ob- preprocessing function T , initial estimate 0, number of power iterations M, number of GD tained spectral estimate was used as the initial guess for iterations N, step size γ GD, run with an exponentially decreasing step size, com- y 1 Z = S diagfT (y)gS patible with a backtracking line search, to ensure stable /* Spectral initialization */ convergence.