Rotation Invariant Graph Neural Networks using Spin Convolutions

06/17/2021
by   Muhammed Shuaibi, et al.
46

Progress towards the energy breakthroughs needed to combat climate change can be significantly accelerated through the efficient simulation of atomic systems. Simulation techniques based on first principles, such as Density Functional Theory (DFT), are limited in their practical use due to their high computational expense. Machine learning approaches have the potential to approximate DFT in a computationally efficient manner, which could dramatically increase the impact of computational simulations on real-world problems. Approximating DFT poses several challenges. These include accurately modeling the subtle changes in the relative positions and angles between atoms, and enforcing constraints such as rotation invariance or energy conservation. We introduce a novel approach to modeling angular information between sets of neighboring atoms in a graph neural network. Rotation invariance is achieved for the network's edge messages through the use of a per-edge local coordinate frame and a novel spin convolution over the remaining degree of freedom. Two model variants are proposed for the applications of structure relaxation and molecular dynamics. State-of-the-art results are demonstrated on the large-scale Open Catalyst 2020 dataset. Comparisons are also performed on the MD17 and QM9 datasets.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
06/29/2022

Spherical Channels for Modeling Atomic Interactions

Modeling the energy and forces of atomic systems is a fundamental proble...
research
03/18/2022

Towards Training Billion Parameter Graph Neural Networks for Atomic Simulations

Recent progress in Graph Neural Networks (GNNs) for modeling atomic simu...
research
05/02/2022

FINETUNA: Fine-tuning Accelerated Molecular Simulations

Machine learning approaches have the potential to approximate Density Fu...
research
02/28/2023

CHGNet: Pretrained universal neural network potential for charge-informed atomistic modeling

The simulation of large-scale systems with complex electron interactions...
research
08/01/2022

Graph Neural Network with Local Frame for Molecular Potential Energy Surface

Modeling molecular potential energy surface is of pivotal importance in ...
research
11/01/2021

Decoupled coordinates for machine learning-based molecular fragment linking

Recent developments in machine-learning based molecular fragment linking...
research
10/29/2020

Graph Neural Network for Metal Organic Framework Potential Energy Approximation

Metal-organic frameworks (MOFs) are nanoporous compounds composed of met...

Please sign up or login with your details

Forgot password? Click here to reset