REANN: A PyTorch-based End-to-End Multi-functional Deep Neural Network Package for Molecular, Reactive and Periodic Systems

Abstract

In this work, we present a general purpose deep neural network package for representing energies, forces, dipole moments, and polarizabilities of atomistic systems. This so-called recursively embedded atom neural network model takes both advantages of the physically inspired atomic descriptor based neural networks and the message-passing based neural networks. Implemented in the PyTorch framework, the training process is parallelized on both CPU and GPU with high efficiency and low memory, in which all hyperparameters can be optimized automatically. We demonstrate the state-of-the-art accuracy, high efficiency, scalability, and universality of this package by learning not only energies (with or without forces), but also dipole moment vectors and polarizability tensors, in various molecular, reactive, and periodic systems. An interface between a trained model and LAMMPs is provided for large scale molecular dynamics simulations. We hope that this open-source toolbox will allow future method development and applications of machine learned potential energy surfaces and quantum-chemical properties of molecules, reactions, and materials.

References in corpus (13)

• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• Efficient and Accurate Machine-Learning Interpolation of Atomic Energies in Compositions with Many Species
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Raman Spectrum and Polarizability of Liquid Water from Deep Neural Networks
• Gaussian Moments as Physically Inspired Molecular Descriptors for Accurate and Scalable Machine Learning Potentials
• Physically Motivated Recursively Embedded Atom Neural Networks: Incorporating Local Completeness and Nonlocality
• Recursive evaluation and iterative contraction of $N$-body equivariant features
• Symmetry-Adapted High Dimensional Neural Network Representation of Electronic Friction Tensor of Adsorbates on Metals
• Conceptual and practical bases for the high accuracy of machine learning interatomic potential
• A Novel Approach to Describe Chemical Environments in High Dimensional Neural Network Potentials
• Mechanical Vibrational Relaxation of NO Scattering from Metal and Insulator Surfaces: When and Why Are They Different?
• An Extendible, Graph-Neural-Network-Based Approach for Accurate Force Field Development of Large Flexible Organic Molecules