Training Algorithm Matters for the Performance of Neural Network Potential: A Case Study of Adam and the Kalman Filter Optimizers

Abstract

One hidden yet important issue for developing neural network potentials (NNPs) is the choice of training algorithm. Here we compare the performance of two popular training algorithms, the adaptive moment estimation algorithm (Adam) and the Extended Kalman Filter algorithm (EKF), using the Behler-Parrinello neural network (BPNN) and two publicly accessible datasets of liquid water [Proc. Natl. Acad. Sci. U.S.A. 2016, 113, 8368-8373 and Proc. Natl. Acad. Sci. U.S.A. 2019, 116, 1110-1115]. This is achieved by implementing EKF in TensorFlow. It is found that NNPs trained with EKF are more transferable and less sensitive to the value of the learning rate, as compared to Adam. In both cases, error metrics of the validation set do not always serve as a good indicator for the actual performance of NNPs. Instead, we show that their performance correlates well with a Fisher information based similarity measure.

References in corpus (20)

• Adam: A Method for Stochastic Optimization
• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• CP2K: An Electronic Structure and Molecular Dynamics Software Package -- Quickstep: Efficient and Accurate Electronic Structure Calculations
• On the difficulty of training Recurrent Neural Networks
• SchNet - a deep learning architecture for molecules and materials
• Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals
• Machine Learning Force Fields
• DP-GEN: A concurrent learning platform for the generation of reliable deep learning based potential energy models
• Less is more: sampling chemical space with active learning
• Machine Learning Molecular Dynamics for the Simulation of Infrared Spectra
• How van der Waals interactions determine the unique properties of water
• Optimizing Neural Networks with Kronecker-factored Approximate Curvature
• Ab initio thermodynamics of liquid and solid water
• Automated Fitting of Neural Network Potentials at Coupled Cluster Accuracy: Protonated Water Clusters as Testing Ground
• Fisher-Rao Metric, Geometry, and Complexity of Neural Networks
• PiNN: A Python Library for Building Atomic Neural Networks of Molecules and Materials
• Temperature effects on the ionic conductivity in concentrated alkaline electrolyte solutions
• Generalization in Deep Networks: The Role of Distance from Initialization
• Pushing the limit of molecular dynamics with ab initio accuracy to 100 million atoms with machine learning