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

Abstract

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).

This article may be downloaded for personal use only. Any other use requires prior permission of the author and AIP Publishing. This article appeared in The Journal of Chemical Physics and may be found at https://doi.org/10.1063/5.0300163

References in corpus (19)

• The density-matrix renormalization group in the age of matrix product states
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Moment Tensor Potentials: a class of systematically improvable interatomic potentials
• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• Active learning of linearly parametrized interatomic potentials
• Physically-informed artificial neural networks for atomistic modeling of materials
• Alternating minimal energy methods for linear systems in higher dimensions. Part I: SPD systems
• Alternating minimal energy methods for linear systems in higher dimensions. Part II: Faster algorithm and application to nonsymmetric systems
• Impact of lattice relaxations on phase transitions in a high-entropy alloy studied by machine-learning potentials
• Constrained evolutionary algorithm for structure prediction of molecular crystals: methodology and applications
• Machine learning of molecular properties: locality and active learning
• A complete description of thermodynamic stabilities of molecular crystals
• Tensor Numerical Methods in Quantum Chemistry: from Hartree-Fock Energy to Excited States
• Automated Calculation of Thermal Rate Coefficients using Ring Polymer Molecular Dynamics and Machine-Learning Interatomic Potentials with Active Learning
• Grid-based electronic structure calculations: the tensor decomposition approach
• Accurate and efficient machine learning interatomic potentials for finite temperature modeling of molecular crystals
• Equivariant Tensor Network Potentials