Papers › GotenNet: Rethinking Efficient 3D Equivariant Graph Neural Networks

GotenNet: Rethinking Efficient 3D Equivariant Graph Neural Networks

24 Apr 2025ICLR 2025 4archive 2025-07-28

Sarp Aykent, Tian Xia

Understanding complex three-dimensional (3D) structures of graphs is essential for accurately modeling various properties, yet many existing approaches struggle with fully capturing the intricate spatial relationships and symmetries inherent in such systems, especially in large-scale, dynamic molecular datasets. These methods often must balance trade-offs between expressiveness and computational efficiency, limiting their scalability. To address this gap, we propose a novel Geometric Tensor Network (GotenNet) that effectively models the geometric intricacies of 3D graphs while ensuring strict equivariance under the Euclidean group E(3). Our approach directly tackles the expressiveness-efficiency trade-off by leveraging effective geometric tensor representations without relying on irreducible representations or Clebsch-Gordan transforms, thereby reducing computational overhead. We introduce a unified structural embedding, incorporating geometry-aware tensor attention and hierarchical tensor refinement that iteratively updates edge representations through inner product operations on high-degree steerable features, allowing for flexible and efficient representations for various tasks. We evaluated models on QM9, rMD17, MD22, and Molecule3D datasets, where the proposed model consistently outperforms state-of-the-art methods in both scalar and high-degree property predictions, demonstrating exceptional robustness across diverse datasets, and establishes GotenNet as a versatile and scalable framework for 3D equivariant Graph Neural Networks.

PaperPDFCode

Code

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Atomic ForcesComputational EfficiencyGraph Property PredictionGraph RegressionMolecular Property Prediction

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Graph Property Prediction QM9 GotenNet (Large) Standardized MAE 0.52 #1 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Large) alpha (ma) 28 #1 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Large) gap (meV) 19.8 #1 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Large) logMAE -6.41 #1 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Base) Standardized MAE 0.56 #2 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Base) alpha (ma) 32 #2 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Base) gap (meV) 20.5 #2 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Base) logMAE -6.35 #2 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Small) Standardized MAE 0.60 #3 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Small) alpha (ma) 33 #3 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Small) gap (meV) 21.2 #3 of 10 Archive leaderboard report
Graph Property Prediction QM9 GotenNet (Small) logMAE -6.29 #3 of 10 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

AttentionSoftmax

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections