Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Are High-Degree Representations Really Unnecessary in Equivariant Graph Neural Networks?

About

Equivariant Graph Neural Networks (GNNs) that incorporate E(3) symmetry have achieved significant success in various scientific applications. As one of the most successful models, EGNN leverages a simple scalarization technique to perform equivariant message passing over only Cartesian vectors (i.e., 1st-degree steerable vectors), enjoying greater efficiency and efficacy compared to equivariant GNNs using higher-degree steerable vectors. This success suggests that higher-degree representations might be unnecessary. In this paper, we disprove this hypothesis by exploring the expressivity of equivariant GNNs on symmetric structures, including $k$-fold rotations and regular polyhedra. We theoretically demonstrate that equivariant GNNs will always degenerate to a zero function if the degree of the output representations is fixed to 1 or other specific values. Based on this theoretical insight, we propose HEGNN, a high-degree version of EGNN to increase the expressivity by incorporating high-degree steerable vectors while maintaining EGNN's efficiency through the scalarization trick. Our extensive experiments demonstrate that HEGNN not only aligns with our theoretical analyses on toy datasets consisting of symmetric structures, but also shows substantial improvements on more complicated datasets such as $N$-body and MD17. Our theoretical findings and empirical results potentially open up new possibilities for the research of equivariant GNNs.

Jiacheng Cen, Anyi Li, Ning Lin, Yuxiang Ren, Zihe Wang, Wenbing Huang• 2024

Related benchmarks

TaskDatasetResultRank
Force PredictionMD17 (test)
Aspirin Force Error9.38
39
State-to-state predictionMD17 (test)
RMSD (Aspirin)0.0306
9
S2S (T=1) 3D Joint Trajectory PredictionCMU Motion Capture Walk, Subject #35 Graphics Lab 2003 (test)
F-MSE16.8
9
S2S (T=1) 3D Joint Trajectory PredictionCMU Motion Capture Run, Subject #9 2003 (test)
F-MSE5.48
9
State-to-trajectory predictionMD17 (test)
Aspirin Error3.09
8
3D Dynamics PredictionMD22 Stachyose (test)
F-MSE (S2S)33.4
5
3D Dynamics PredictionMD22 Ac-Ala3-NHMe (test)
F-MSE (S2S)124.1
5
Molecular Dynamics trajectory predictionAdK protein MD (test)--
5
3D Dynamics PredictionMotion Capture Walk (test)
F-MSE (T=1)0.83
4
3D Dynamics PredictionMotion Capture Run (test)
F-MSE (T=1)53.3
4
Showing 10 of 12 rows

Other info

Follow for update