ænet-PyTorch: a GPU-supported implementation for machine learning atomic potentials training

Abstract

In this work, we present ænet-PyTorch, a PyTorch-based implementation for training artificial neural network-based machine learning interatomic potentials. Developed as an extension of the atomic energy network (ænet), ænet-PyTorch provides access to all the tools included in ænet for the application and usage of the potentials. The package has been designed as an alternative to the internal training capabilities of ænet, leveraging the power of graphic processing units to facilitate direct training on forces in addition to energies. This leads to a substantial reduction of the training time by one to two orders of magnitude compared to the CPU implementation, enabling direct training on forces for systems beyond small molecules. Here we demonstrate the main features of ænet-PyTorch and show its performance on open databases. Our results show that training on all the force information within a data set is not necessary, and including between 10% to 20% of the force information is sufficient to achieve optimally accurate interatomic potentials with the least computational resources.

arXiv comment9 pages, 7 figures and 1 table

References in corpus (31)

• Adam: A Method for Stochastic Optimization
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Decoupled Weight Decay Regularization
• ADADELTA: An Adaptive Learning Rate Method
• Neural Message Passing for Quantum Chemistry
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• On representing chemical environments
• DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics
• Perspective on density functional theory
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• Comparing molecules and solids across structural and alchemical space
• Machine Learning Unifies the Modelling of Materials and Molecules
• On-the-fly machine learning force field generation: Application to melting points
• Physics-inspired structural representations for molecules and materials
• Machine learning a general purpose interatomic potential for silicon
• Accelerating crystal structure prediction by machine-learning interatomic potentials with active learning
• Ab initio thermodynamics of liquid and solid water
• Efficient and Accurate Machine-Learning Interpolation of Atomic Energies in Compositions with Many Species
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• Extending the Accuracy of the SNAP Interatomic Potential Form
• A Machine Learning Potential for Graphene
• A learning scheme to predict atomic forces and accelerate materials simulations
• Machine learning of molecular properties: locality and active learning
• Constructing first-principles phase diagrams of amorphous LixSi using machine-learning-assisted sampling with an evolutionary algorithm
• A Deep Dive into Machine Learning Density Functional Theory for Materials Science and Chemistry
• Strategies for the Construction of Machine-Learning Potentials for Accurate and Efficient Atomic-Scale Simulations
• Efficient Training of ANN Potentials by Including Atomic Forces via Taylor Expansion and Application to Water and a Transition-Metal Oxide
• Self-learning Hybrid Monte Carlo: A First-principles Approach
• AENET-LAMMPS and AENET-TINKER: Interfaces for Accurate and Efficient Molecular Dynamics Simulations with Machine Learning Potentials
• Machine Learning for compositional disorder: A Comparison Between Different Descriptors and Machine Learning Frameworks

Cited by in corpus (5)

• MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflows
• MGNN: Moment Graph Neural Network for Universal Molecular Potentials
• Guest Editorial: Special Topic on Software for Atomistic Machine Learning
• Efficient GPU-Accelerated Training of a Neuroevolution Potential with Analytical Gradients
• Scalable Training of Neural Network Potentials for Complex Interfaces Through Data Augmentation