Transfer learning electronic structure: millielectron volt accuracy for sub-million-atom moir\'e semiconductor
Abstract
The integration of density functional theory (DFT) with machine learning enables efficient \textit{ab initio} electronic structure calculations for ultra-large systems. In this work, we develop a transfer learning framework tailored for long-wavelength moir\'e systems. To balance efficiency and accuracy, we adopt a two-step transfer learning strategy: (1) the model is pre-trained on a large dataset of computationally inexpensive non-twisted structures until convergence, and (2) the network is then fine-tuned using a small set of computationally expensive twisted structures. Applying this method to twisted MoTe, the neural network model generates the resulting Hamiltonian for a 1000-atom system in 200 seconds, achieving a mean absolute error below 0.1 meV. To demonstrate scalability, we model nanoribbon systems with up to 0.25 million atoms ( million orbitals), accurately capturing edge states consistent with predicted Chern numbers. This approach addresses the challenges of accuracy, efficiency, and scalability, offering a viable alternative to conventional DFT and enabling the exploration of electronic topology in large scale moir\'e systems towards simulating realistic device architectures.
Keywords
Cite
@article{arxiv.2501.12452,
title = {Transfer learning electronic structure: millielectron volt accuracy for sub-million-atom moir\'e semiconductor},
author = {Ting Bao and Ning Mao and Wenhui Duan and Yong Xu and Adrian Del Maestro and Yang Zhang},
journal= {arXiv preprint arXiv:2501.12452},
year = {2025}
}
Comments
5+14 pages, 4+ 11 figures