Low-rank matrix and tensor approximations for compression of machine-learning interatomic potentials

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Vorotnikov, Igor, Romashov, Fedor, Rybin, Nikita, Rakhuba, Maxim, Novikov, Ivan S.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917195500486656
author Vorotnikov, Igor
Romashov, Fedor
Rybin, Nikita
Rakhuba, Maxim
Novikov, Ivan S.
author_facet Vorotnikov, Igor
Romashov, Fedor
Rybin, Nikita
Rakhuba, Maxim
Novikov, Ivan S.
contents Machine-learning interatomic potentials (MLIPs) have become a mainstay in computationally-guided materials science, surpassing traditional force fields due to their flexible functional form and superior accuracy in reproducing physical properties of materials. This flexibility is achieved through mathematically-rigorous basis sets that describe interatomic interactions within a local atomic environment. The number of parameters in these basis sets influences both the size of the training dataset required and the computational speed of the MLIP. Consequently, compressing MLIPs by reducing the number of parameters is a promising route to more efficient simulations. In this work, we use low-rank matrix and tensor factorizations under fixed-rank constraints to achieve this compression. In addition, we demonstrate that an algorithm with automatic rank augmentation helps to find a deeper local minimum of the fitted potential. The methodology is mainly verified using the Moment Tensor Potential (MTP) model and benchmarked on multi-component systems: a Mo-Nb-Ta-W medium-entropy alloy, molten LiF-NaF-KF, and a glycine molecular crystal. The proposed approach achieves up to 50 % compression without any loss of MTP accuracy. We also demonstrate that the developed methodology is universal and can be applied to compress other MLIPs on the example of Atomic Cluster Expansion (ACE).
format Preprint
id arxiv_https___arxiv_org_abs_2509_04440
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Low-rank matrix and tensor approximations for compression of machine-learning interatomic potentials
Vorotnikov, Igor
Romashov, Fedor
Rybin, Nikita
Rakhuba, Maxim
Novikov, Ivan S.
Chemical Physics
Machine-learning interatomic potentials (MLIPs) have become a mainstay in computationally-guided materials science, surpassing traditional force fields due to their flexible functional form and superior accuracy in reproducing physical properties of materials. This flexibility is achieved through mathematically-rigorous basis sets that describe interatomic interactions within a local atomic environment. The number of parameters in these basis sets influences both the size of the training dataset required and the computational speed of the MLIP. Consequently, compressing MLIPs by reducing the number of parameters is a promising route to more efficient simulations. In this work, we use low-rank matrix and tensor factorizations under fixed-rank constraints to achieve this compression. In addition, we demonstrate that an algorithm with automatic rank augmentation helps to find a deeper local minimum of the fitted potential. The methodology is mainly verified using the Moment Tensor Potential (MTP) model and benchmarked on multi-component systems: a Mo-Nb-Ta-W medium-entropy alloy, molten LiF-NaF-KF, and a glycine molecular crystal. The proposed approach achieves up to 50 % compression without any loss of MTP accuracy. We also demonstrate that the developed methodology is universal and can be applied to compress other MLIPs on the example of Atomic Cluster Expansion (ACE).
title Low-rank matrix and tensor approximations for compression of machine-learning interatomic potentials
topic Chemical Physics
url https://arxiv.org/abs/2509.04440