Aberration Corrector Tuning with Machine-Learning-Based Emittance Measurements and Bayesian Optimization
Chenyu Zhang, Zhaslan Baraissov, Cameron Duncan, Adi Hanuka, Auralee Edelen, Jared Maxson, David Muller
2021
Microscopy and Microanalysis
Aberration-corrected scanning transmission electron microscopes (STEM) use a series of multipole magnets to generate a sub-Ångstrom-sized electron beam for atomic resolution imaging and chemical composition mapping 1 . A new scheme for aberration corrector alignment is proposed, including both new ways of beam quality measurement and aberration corrector control. The new scheme targets fully automated corrector alignment to achieve microscope tuning with greater speed and less human bias. For
more »
... am quality measurements, we trained a convolutional neural network (CNN) to determine the beam emittance 2 from a single Ronchigram, as shown in Figure 1 (a). Beam emittance is a single variable that characterizes the volume occupied by the beam in the phase space. Emittance is convex against the aberration coefficients, and proportional to the root mean square of the phase error, as shown in Figure 1 (b). Both indicate that beam emittance is a single robust objective for aberration correction, and avoids the cusp-like instabilities of the individual aberration coefficients 3 . The emittance would be a conserved quantity without the introduction of aberrations, and aberration correction can be guided by minimizing the emittance growth. We trained a CNN with a VGG16 architecture using simulated Ronchigram-emittance pairs to predict emittance from a single Ronchigram. The Ronchigrams are simulated using a random phase plate following ref 4 , with emittance calculated from the probe profile. Validation results of the CNN shown in Figure 1 (c) suggest that it can predict the correct emittance value down to about 0.1 pm·rad except when the noise level is high as a result of too-short an exposure time (~0.1 ms). We applied Gaussian Processes (GP) 5 , a Bayesian approach, to search for optimal corrector parameters. A general particle tracer (GPT) 6 simulation model was used to simulate the electron probe at the sample for STEM column with a simple hexapole corrector 7 . The ray diagram of the GPT model with components marked is shown in Figure 2 (a). The beam emittance calculated from the phase space distribution of all the electrons in the GPT output has been used as GP's objective function to be minimized. Figure 2 (b) shows the emittance evolution over 100 iterations during the GP optimization of a partial alignment controlling the current of the two hexapole elements, for 10 random starting points. The beam quality improvement from uncorrected state to the corrected state guided by GP can be seen in the Ronchigram shown in Figure 2 (c) and (d), where the half-angle of the flat area increased from about 10 mrad to about 27 mrad. Overall, we have shown a new method to tune the aberration corrector using emittance as an alternate metric to measure beam quality and a combination of machine learning tools to monitor and optimize the beam. The emittance prediction by CNN was tested on simulated Ronchigrams and found to be robust over a wide range of experimental conditions. Corrector tuning with GP has been tested on a GPT simulation model and effectively tuned the corrector to eliminate spherical aberration in the system 8 . https://www.cambridge.org/core/terms. https://doi.
doi:10.1017/s1431927621003214
fatcat:qllnpmzhzvgsrbtictdiwybnpa