Adaptive Scanning in Ptychography through Deep Reinforcement Learning
Adaptive Scanning in Ptychography through Deep Reinforcement Learning
复制标题
DOI:
10.1017/s1431927621003238
复制
发表时间:
2021-07
影响因子:
2.8
通讯作者:
M. Schloz;Johannes Müller;T. Pekin;W. Van den Broek;C. Koch
中科院分区:
文献类型:
--
作者:
M. Schloz;Johannes Müller;T. Pekin;W. Van den Broek;C. Koch
The development of new hardware and reconstruction algorithms has led to ptychography becoming a mature electron microscopy technique. The current research to further improve this technique is driven by the desire to investigate thick samples as well as to measure with a low electron dose. For the latter, researches have tried to relax the strict scanning constraint, which is required for obtaining information redundancy through consecutively overlapping probes. One approach is the optimization of the scanning pattern. A deviation from a raster grid towards a generally more beneficial pattern is supposed to improve the reconstruction quality especially at a reduced dose [1]. However, since a generally optimized scanning pattern is completely unrelated to the structure of the investigated specimen, there is obviously a limit of the achievable reconstruction quality improvement. Here, we present an adaptive scanning algorithm that optimizes the scanning pattern based on structural information it receives from diffraction data that is acquired by the microscope. Our algorithm is based on a combination of deep learning [2] and reinforcement learning (RL) [3], or deep RL, which means that we model the described prediction process with deep neural networks and train it on previously acquired ptychographic data with RL. A recurrent neural network (RNN) is used to predict sub-sequences of scanning positions after starting from an initial sub-sequence. Input information is given by a combination X p of processed 2D coordinates of a currently scanned sub-sequence R p and processed content information z p that is a compressed representation of the partial ptychographic reconstruction V p . In addition, all the information from the previous prediction steps, represented by the hidden state H p , is combined with the input information X p and mapped by a gated recurrent unit (GRU) cell [4] to the next hidden state