中国物理B ›› 2026, Vol. 35 ›› Issue (7): 77507-077507.doi: 10.1088/1674-1056/ae5db1

• • 上一篇    下一篇

Prediction of material parameters of spin textures from LTEM images by machine learning models CNN and ViT

Bohan Li(李博涵)1, Xinyuan Zhang(张歆媛)2, Guanhua Chen(陈冠桦)2, and Lin Chen(陈琳)2,†   

  1. 1 School of Computer Science, Nanjing University of Posts and Telecommunications, Nanjing 210003, China;
    2 College of Electronic and Optical Engineering, Nanjing University of Posts and Telecommunications, Nanjing 210023, China
  • 收稿日期:2025-12-23 修回日期:2026-03-24 接受日期:2026-04-10 发布日期:2026-07-15
  • 通讯作者: Lin Chen E-mail:cl@njupt.edu.cn

Prediction of material parameters of spin textures from LTEM images by machine learning models CNN and ViT

Bohan Li(李博涵)1, Xinyuan Zhang(张歆媛)2, Guanhua Chen(陈冠桦)2, and Lin Chen(陈琳)2,†   

  1. 1 School of Computer Science, Nanjing University of Posts and Telecommunications, Nanjing 210003, China;
    2 College of Electronic and Optical Engineering, Nanjing University of Posts and Telecommunications, Nanjing 210023, China
  • Received:2025-12-23 Revised:2026-03-24 Accepted:2026-04-10 Published:2026-07-15
  • Contact: Lin Chen E-mail:cl@njupt.edu.cn

PDF (PC)

2

赞

补充材料

1

摘要： Machine learning (ML) models were trained to predict the Dzyaloshinskii-Moriya interaction constant $D$ and anisotropy constant $K$ of spin textures from Lorentz transmission electron microscopy (LTEM) images. Two ML models, convolutional neural network (CNN) and vision transformer (ViT), were trained, tested and employed to predict the values of $D$ and $K$. Firstly, training and testing datasets composed of 9300 topological spin textures were prepared using the micromagnetic simulation method with the values of $D$ and $K$ randomly chosen. Secondly, the performance of the CNN model for predicting $D$ and $K$ values was investigated by varying the number of training data, the pooling process, and the number of fully connected layers, which indicated high prediction accuracies. Thirdly, better performance of the ViT model for predicting $D$ and $K$ values was achieved with the coefficient of determination $R^{2}$ reaching as high as 0.9987 and 0.9991 for predicting $D$ and $K$ values, respectively. Particularly, the reliance on synthetic data and noise robustness was discussed. Finally, the prediction abilities of CNN and ViT models were evaluated and compared. The results of this research indicate that the ML models can achieve the material parameters directly and effectively from LTEM images, and this process may contribute to the design of advanced devices based on the topological spin textures.

关键词: spin texture, LTEM, machine learning, ViT model

Abstract: Machine learning (ML) models were trained to predict the Dzyaloshinskii-Moriya interaction constant $D$ and anisotropy constant $K$ of spin textures from Lorentz transmission electron microscopy (LTEM) images. Two ML models, convolutional neural network (CNN) and vision transformer (ViT), were trained, tested and employed to predict the values of $D$ and $K$. Firstly, training and testing datasets composed of 9300 topological spin textures were prepared using the micromagnetic simulation method with the values of $D$ and $K$ randomly chosen. Secondly, the performance of the CNN model for predicting $D$ and $K$ values was investigated by varying the number of training data, the pooling process, and the number of fully connected layers, which indicated high prediction accuracies. Thirdly, better performance of the ViT model for predicting $D$ and $K$ values was achieved with the coefficient of determination $R^{2}$ reaching as high as 0.9987 and 0.9991 for predicting $D$ and $K$ values, respectively. Particularly, the reliance on synthetic data and noise robustness was discussed. Finally, the prediction abilities of CNN and ViT models were evaluated and compared. The results of this research indicate that the ML models can achieve the material parameters directly and effectively from LTEM images, and this process may contribute to the design of advanced devices based on the topological spin textures.

Key words: spin texture, LTEM, machine learning, ViT model

中图分类号:  (Spin arrangements in magnetically ordered materials (including neutron And spin-polarized electron studies, synchrotron-source x-ray scattering, etc.))

  • 75.25.-j
75.78.Cd (Micromagnetic simulations ?) 87.19.lv (Learning and memory) 07.05.Tp (Computer modeling and simulation)