Abstract:A rotationally invariant graph neural network (RotNet) is proposed for accurate and accelerated quantum mechanical calculations. Benchmark experiments demonstrate that RotNet outperforms prior state‐of‐the‐art methods and generalizes well to varying rotations. The high accuracy and efficiency suggest that RotNet has tremendous potential to significantly facilitate studies of protein dynamics simulation and materials engineering while maintaining physical plausibility. Deep learning has proven promising in biological and chemical applications, aiding in accurate predictions of properties such as atomic forces, energies, and material band gaps. Traditional methods with rotational invariance, one of the most crucial physical laws for predictions made by machine learning, have relied on Fourier transforms or specialized convolution filters, leading to complex model design and reduced accuracy and efficiency. However, models without rotational invariance exhibit poor generalization ability across datasets. Addressing this contradiction, this work proposes a rotationally invariant graph neural network, named RotNet, for accurate and accelerated quantum mechanical calculations that can overcome the generalization deficiency caused by rotations of molecules. RotNet ensures rotational invariance through an effective transformation and learns distance and angular information from atomic coordinates. Benchmark experiments on three datasets (protein fragments, electronic materials, and QM9) demonstrate that the proposed RotNet framework outperforms popular baselines and generalizes well to spatial data with varying rotations. The high accuracy, efficiency, and fast convergence of RotNet suggest that it has tremendous potential to significantly facilitate studies of protein dynamics simulation and materials engineering while maintaining physical plausibility.

RotEqNet: Rotation-equivariant network for fluid systems with symmetric high-order tensors

RotNet: A Rotationally Invariant Graph Neural Network for Quantum Mechanical Calculations

Rotation equivariant vector field networks

Importance of equivariant and invariant symmetries for fluid flow modeling

Predicting fluid-structure interaction with graph neural networks

Predicting fluid–structure interaction with graph neural networks

Deep Rotation Equivariant Network

Multi-scale rotation-equivariant graph neural networks for unsteady Eulerian fluid dynamics

Rethinking Local-to-global Representation Learning for Rotation-Invariant Point Cloud Analysis

Equivariant Vector Field Network for Many-Body System Modeling.

REMuS-GNN: A Rotation-Equivariant Model for Simulating Continuum Dynamics

Rotation Equivariant Proximal Operator for Deep Unfolding Methods in Image Restoration

Discrete Rotation Equivariance for Point Cloud Recognition

E($3$) Equivariant Graph Neural Networks for Particle-Based Fluid Mechanics

Rotation invariant point cloud analysis: Where local geometry meets global topology

Rotation-Invariant Deep Embedding for Remote Sensing Images

An equivariant neural operator for developing nonlocal tensorial constitutive models

Unifying O(3) Equivariant Neural Networks Design with Tensor-Network Formalism

Simulation and Prediction of Three-Dimensional Rotating Flows Based on Convolutional Neural Networks

A finite element-inspired hypergraph neural network: Application to fluid dynamics simulations

Equivariance-bridged SO(2)-Invariant Representation Learning using Graph Convolutional Network