Geometric and Physical Quantities improve E(3) Equivariant Message Passing

10/06/2021
by   Johannes Brandstetter, et al.
0

Including covariant information, such as position, force, velocity or spin is important in many tasks in computational physics and chemistry. We introduce Steerable E(3) Equivariant Graph Neural Networks (SEGNNs) that generalise equivariant graph networks, such that node and edge attributes are not restricted to invariant scalars, but can contain covariant information, such as vectors or tensors. This model, composed of steerable MLPs, is able to incorporate geometric and physical information in both the message and update functions. Through the definition of steerable node attributes, the MLPs provide a new class of activation functions for general use with steerable feature fields. We discuss ours and related work through the lens of equivariant non-linear convolutions, which further allows us to pin-point the successful components of SEGNNs: non-linear message aggregation improves upon classic linear (steerable) point convolutions; steerable messages improve upon recent equivariant graph networks that send invariant messages. We demonstrate the effectiveness of our method on several tasks in computational physics and chemistry and provide extensive ablation studies.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
05/03/2023

An Exploration of Conditioning Methods in Graph Neural Networks

The flexibility and effectiveness of message passing based graph neural ...
research
01/04/2022

Graph Decipher: A transparent dual-attention graph neural network to understand the message-passing mechanism for the node classification

Graph neural networks can be effectively applied to find solutions for m...
research
05/11/2023

E(n) Equivariant Message Passing Simplicial Networks

This paper presents E(n) Equivariant Message Passing Simplicial Networks...
research
06/15/2022

MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fields

Creating fast and accurate force fields is a long-standing challenge in ...
research
06/09/2021

Breaking the Limits of Message Passing Graph Neural Networks

Since the Message Passing (Graph) Neural Networks (MPNNs) have a linear ...
research
10/24/2020

Pathfinder Discovery Networks for Neural Message Passing

In this work we propose Pathfinder Discovery Networks (PDNs), a method f...
research
03/03/2022

Thermodynamics-informed graph neural networks

In this paper we present a deep learning method to predict the temporal ...

Please sign up or login with your details

Forgot password? Click here to reset