A2I Transformer: Permutation-equivariant attention network for pairwise and many-body interactions with minimal featurization

10/27/2021
by   Ji Woong Yu, et al.
0

The combination of neural network potential (NNP) with molecular simulations plays an important role in an efficient and thorough understanding of a molecular system's potential energy surface (PES). However, grasping the interplay between input features and their local contribution to NNP is growingly evasive due to heavy featurization. In this work, we suggest an end-to-end model which directly predicts per-atom energy from the coordinates of particles, avoiding expert-guided featurization of the network input. Employing self-attention as the main workhorse, our model is intrinsically equivariant under the permutation operation, resulting in the invariance of the total potential energy. We tested our model against several challenges in molecular simulation problems, including periodic boundary condition (PBC), n-body interaction, and binary composition. Our model yielded stable predictions in all tested systems with errors significantly smaller than the potential energy fluctuation acquired from molecular dynamics simulations. Thus, our work provides a minimal baseline model that encodes complex interactions in a condensed phase system to facilitate the data-driven analysis of physicochemical systems.

READ FULL TEXT
research
12/11/2017

DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics

Recent developments in many-body potential energy representation via dee...
research
09/29/2017

Hierarchical modeling of molecular energies using a deep neural network

We introduce the Hierarchically Interacting Particle Neural Network (HIP...
research
10/19/2021

Permutation invariant graph-to-sequence model for template-free retrosynthesis and reaction prediction

Synthesis planning and reaction outcome prediction are two fundamental p...
research
11/28/2021

Towards Conditional Generation of Minimal Action Potential Pathways for Molecular Dynamics

In this paper, we utilized generative models, and reformulate it for pro...
research
05/09/2023

Error estimate of the u-series method for molecular dynamics simulations

This paper provides an error estimate for the u-series decomposition of ...
research
06/27/2019

Deep neural network for Wannier function centers

We introduce a deep neural network (DNN) model that assigns the position...
research
08/17/2022

DPA-1: Pretraining of Attention-based Deep Potential Model for Molecular Simulation

Machine learning assisted modeling of the inter-atomic potential energy ...

Please sign up or login with your details

Forgot password? Click here to reset