Enhancing Molecular Property Predictions by Learning from Bond Modelling and Interactions
Abstract
Molecule representation learning is crucial for understanding and predicting molecular properties. However, conventional atom-centric models, which treat chemical bonds merely as pairwise interactions, often overlook complex bond-level phenomena like resonance and stereoselectivity. This oversight limits their predictive accuracy for nuanced chemical behaviors. To address this limitation, we introduce DeMol, a dual-graph framework whose architecture is motivated by a rigorous information-theoretic analysis demonstrating the information gain from a bond-centric perspective. DeMol explicitly models molecules through parallel atom-centric and bond-centric channels. These are synergistically fused by multi-scale Double-Helix Blocks designed to learn intricate atom-atom, atom-bond, and bond-bond interactions. The framework's geometric consistency is further enhanced by a regularization term based on covalent radii to enforce chemically plausible structures. Comprehensive evaluations on diverse benchmarks, including PCQM4Mv2, OC20 IS2RE, QM9, and MoleculeNet, show that DeMol establishes a new state-of-the-art, outperforming existing methods. These results confirm the superiority of explicitly modelling bond information and interactions, paving the way for more robust and accurate molecular machine learning.
Benchmark Results
Tested on 1 dataset 4 baseline values reported too low
Tables show presented results in global context (best 3 papers + GCN + MLP).
OGB
| # | Model | MAE ↓ | Paper |
|---|---|---|---|
| 1 | DeMol | 0.0603 | This paper |
| 2 | TGT-At | 0.0671 | This paper |
| 3 | TGT-At (+RDKit) | 0.0683 | |
| 4 | UniMol | 0.0693 | This paper |
| 6 | GPS | 0.0720 best reported | |
| 7 | Transformer-M | 0.0772 | |
| 9 | GEM-2 | 0.0793 | |
| 21 | GraphGPS_DEEP | 0.0852 | This paper |
| 25 | GraphGPS_BASE | 0.0858 | This paper |
| 29 | EGT | 0.0862 best reported | |
| 31 | Graphormer | 0.0864 | This paper |
| 33 | GRPE_LARGE | 0.0867 | This paper |
| 40 | GRPE_BASE | 0.0890 | This paper |
| 44 | TokenGT | 0.0910 | This paper |
| 52 | GraphGPS_SMALL | 0.0938 | This paper |
| 55 | DeeperGCN-VN | 0.1021 | This paper |
| 58 | GIN-VN | 0.1083 | This paper |
| 60 | GCN-VN | 0.1152 best reported | best at pub: GRPE: Relative Positional Encoding for Graph Transformer (2022) |
| 62 | GINE-VN | 0.1167 | This paper |
| 65 | GCN | 0.1195 best reported | best at pub: Enhancing Graph Neural Networks with Quantum Computed Encodings (2023) |
| 64 | GIN | 0.1195 | This paper |
| 82 | MLP-Fingerprint | 0.1735 | prev best: One Transformer Can Understand Both 2D & 3D Molecular Data (ICLR 2022) |
