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 (45)

• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• On representing chemical environments
• Deep Potential Molecular Dynamics: a scalable model with the accuracy of quantum mechanics
• SchNet - a deep learning architecture for molecules and materials
• DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Moment Tensor Potentials: a class of systematically improvable interatomic potentials
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• PhysNet: A Neural Network for Predicting Energies, Forces, Dipole Moments and Partial Charges
• Towards Exact Molecular Dynamics Simulations with Machine-Learned Force Fields
• Physics-inspired structural representations for molecules and materials
• Machine Learning Molecular Dynamics for the Simulation of Infrared Spectra
• SchNetPack: A Deep Learning Toolbox For Atomistic Systems
• Machine learning for electronically excited states of molecules
• Ab initio thermodynamics of liquid and solid water
• SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effects
• Efficient and Accurate Machine-Learning Interpolation of Atomic Energies in Compositions with Many Species
• FCHL revisited: faster and more accurate quantum machine learning
• Alchemical and structural distribution based representation for improved QML
• Symmetry-Adapted Machine-Learning for Tensorial Properties of Atomistic Systems
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• WACSF - Weighted Atom-Centered Symmetry Functions as Descriptors in Machine Learning Potentials
• Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation
• Combining SchNet and SHARC: The SchNarc machine learning approach for excited-state dynamics
• On the Completeness of Atomic Structure Representations
• Inclusion of machine learning kernel ridge regression potential energy surfaces in on-the-fly nonadiabatic molecular dynamics simulation
• Atom-Density Representations for Machine Learning
• Structure-based Sampling and Self-correcting Machine Learning for Accurate Calculations of Potential Energy Surfaces and Vibrational Levels
• Raman Spectrum and Polarizability of Liquid Water from Deep Neural Networks
• Efficient non-parametric fitting of potential energy surfaces for polyatomic molecules with Gaussian processes
• 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
• Efficient and Accurate Simulations of Vibrational and Electronic Spectra with Symmetry-Preserving Neural Network Models for Tensorial Properties
• Sparse Gaussian Process Potentials: Application to Lithium Diffusivity in Superionic Conducting Solid Electrolytes
• 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
• PANNA: Properties from Artificial Neural Network Architectures
• Conceptual and practical bases for the high accuracy of machine learning interatomic potential
• Atomic Permutationally Invariant Polynomials for Fitting Molecular Force Fields
• A Novel Approach to Describe Chemical Environments in High Dimensional Neural Network Potentials
• Accelerating Atomistic Simulations with Piecewise Machine Learned Ab Initio Potentials at Classical Force Field-like Cost
• 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
• Ready, set and no action: A static perspective on potential energy surfaces commonly used in gas-surface dynamics

Cited by in corpus (2)

• GPUMD: A package for constructing accurate machine-learned potentials and performing highly efficient atomistic simulations
• Neural network with optimal neuron activation functions based on additive Gaussian process regression