Transfer learning electronic structure: millielectron volt accuracy for sub-million-atom moiré semiconductor

Ting Bao, Ning Mao, Wenhui Duan, Yong Xu, Adrian Del Maestro, Yang Zhang

arXiv:2501.12452·cond-mat.mtrl-sci·Published 2025-01-21

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é 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$_2$, 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 $O(N)$ scalability, we model nanoribbon systems with up to 0.25 million atoms ($\sim9$ 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é systems towards simulating realistic device architectures.

TopicsMaterials Science & Condensed Matter

Tagselectronic-structure

arXiv categoriescond-mat.mtrl-sci, cond-mat.str-el

arXiv abstract pagePDF