Molecule Graph Networks with Many-body Equivariant Interactions
Abstract
Message passing neural networks have demonstrated significant efficacy in predicting molecular interactions. Introducing equivariant vectorial representations augments expressivity by capturing geometric data symmetries, thereby improving model accuracy. However, two-body bond vectors in opposition may cancel each other out during message passing, leading to the loss of directional information on their shared node. In this study, we develop Equivariant N-body Interaction Networks (ENINet) that explicitly integrates $l = 1$ equivariant many-body interactions to enhance directional symmetric information in the message passing scheme. We provided a mathematical analysis demonstrating the necessity of incorporating many-body equivariant interactions and generalized the formulation to $N$-body interactions. Experiments indicate that integrating many-body equivariant representations enhances prediction accuracy across diverse scalar and tensorial quantum chemical properties.
Benchmark Results
Tested on 1 dataset 3 baseline values reported too low
Tables show presented results in global context (best 3 papers + GCN + MLP).
Quantum Chemistry
| # | Model | MAE ↓ | Paper |
|---|---|---|---|
| 1 | QHNet | 0.0000 | |
| 2 | QHNetV2 | 0.0000 | |
| 3 | PhiSNet | 0.0000 | |
| 22 | PaiNN | 0.0100 best reported | best at pub: Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materials (NeurIPS 2023) |
| 41 | ENINet | 0.0430 | This paper |
| 46 | SchNet | 0.0500 best reported | |
| 54 | ET | 0.0520 | This paper |
| 63 | NewtonNet | 0.0610 best reported | |
| 64 | DimeNet++ | 0.0640 | This paper |
| 74 | NequIP | 0.0970 | This paper |
