A Lightweight Universal Machine-Learning Interatomic Potential via Knowledge Distillation for Scalable Atomistic Simulations

Abstract

We introduce a lightweight universal machine-learning interatomic potential (uMLIP), SevenNet-Nano, based on the graph neural network architecture SevenNet and enabled by a knowledge-distillation framework. The model inherits the broad generalization capability of a large multi-task foundation model, SevenNet-Omni, trained on diverse materials datasets across chemical, configurational, and computational spaces. By learning chemical representations from high-quality inference data generated by the teacher model within a unified computational framework, SevenNet-Nano achieves high accuracy and strong transferability despite its compact architecture. The model also accurately captures a wide range of interatomic interactions, enabling reliable simulations under both equilibrium and extreme conditions, including plasma etching of SiO$_2$. Comprehensive benchmarks on static and dynamical properties--such as Li-ion diffusion and liquid densities--demonstrate its broad applicability with minimal fine-tuning. Importantly, SevenNet-Nano significantly reduces computational cost, achieving over an order-of-magnitude speedup and enabling large-scale atomistic simulations involving thousands of atoms.