论文标题
STM图像中杂质的机器学习识别
Machine Learning Identification of Impurities in the STM Images
论文作者
论文摘要
在这项工作中,我们训练一个神经网络,以识别通过扫描隧道显微镜测量获得的实验图像中的杂质。首先对神经网络进行了大量模拟数据训练,然后应用训练有素的神经网络来确定一组在不同电压下拍摄的实验图像。我们使用卷积神经网络从图像中提取特征,并实施注意机制以捕获不同电压下拍摄的图像之间的相关性。我们注意到,模拟数据可以捕获通用的Friedel振荡,但无法正确描述附近杂质附近的非宇宙物理短距离物理学以及实验数据中的声音。我们强调,这种方法的关键是正确处理模拟数据和实验数据之间的这些差异。在这里,我们表明,即使在模拟数据中包括无关的白色噪声,神经网络在实验数据上的性能也可以显着改善。为了防止神经网络学习非物理短距离物理学,我们还开发了另一种方法来评估神经网络对实验数据的置信度,并将此置信度衡量添加到损失函数中。我们表明,添加这样的额外损失函数也可以提高实验数据的性能。我们的研究可以激发机器学习对实验数据分析的未来类似应用。
In this work we train a neural network to identify impurities in the experimental images obtained by the scanning tunneling microscope measurements. The neural network is first trained with large number of simulated data and then the trained neural network is applied to identify a set of experimental images taken at different voltages. We use the convolutional neural network to extract features from the images and also implement the attention mechanism to capture the correlations between images taken at different voltages. We note that the simulated data can capture the universal Friedel oscillation but cannot properly describe the non-universal physics short-range physics nearby an impurity, as well as noises in the experimental data. And we emphasize that the key of this approach is to properly deal these differences between simulated data and experimental data. Here we show that even by including uncorrelated white noises in the simulated data, the performance of neural network on experimental data can be significantly improved. To prevent the neural network from learning unphysical short-range physics, we also develop another method to evaluate the confidence of the neural network prediction on experimental data and to add this confidence measure into the loss function. We show that adding such an extra loss function can also improve the performance on experimental data. Our research can inspire future similar applications of machine learning on experimental data analysis.