Kolmogorov--Arnold networks in molecular dynamics

Yuki Nagai,Masahiko Okumura
2024-07-25
Abstract:We explore the integration of Kolmogorov Networks (KANs) into molecular dynamics (MD) simulations to improve interatomic potentials. We propose that widely used potentials, such as the Lennard-Jones (LJ) potential, the embedded atom model (EAM), and artificial neural network (ANN) potentials, can be interpreted within the KAN framework. Specifically, we demonstrate that the descriptors for ANN potentials, typically constructed using polynomials, can be redefined using KAN's non-linear functions. By employing linear or cubic spline interpolations for these KAN functions, we show that the computational cost of evaluating ANN potentials and their derivatives is reduced.
Materials Science
What problem does this paper attempt to address?
The paper attempts to address the problem of improving interatomic potentials in molecular dynamics (MD) simulations to enhance the accuracy and efficiency of the simulations. Specifically, the authors explore the application of Kolmogorov-Arnold Networks (KANs) in molecular dynamics and propose reinterpreting existing potential functions (such as Lennard-Jones potential, Embedded Atom Model (EAM), and Artificial Neural Network (ANN) potentials) in the form of the KAN framework. By using linear or cubic spline interpolation to approximate KAN descriptors, the computational cost of ANN potentials and their derivatives can be reduced, thereby increasing simulation speed. Summary: - **Objective**: Improve the accuracy and efficiency of molecular dynamics simulations. - **Method**: Redefine existing potential functions using Kolmogorov-Arnold Networks (KANs). - **Advantage**: Reduce computational cost through linear or cubic spline interpolation.