Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Molecules
read the original abstract
Many important tasks in chemistry revolve around molecules during reactions. This requires predictions far from the equilibrium, while most recent work in machine learning for molecules has been focused on equilibrium or near-equilibrium states. In this paper we aim to extend this scope in three ways. First, we propose the DimeNet++ model, which is 8x faster and 10% more accurate than the original DimeNet on the QM9 benchmark of equilibrium molecules. Second, we validate DimeNet++ on highly reactive molecules by developing the challenging COLL dataset, which contains distorted configurations of small molecules during collisions. Finally, we investigate ensembling and mean-variance estimation for uncertainty quantification with the goal of accelerating the exploration of the vast space of non-equilibrium structures. Our DimeNet++ implementation as well as the COLL dataset are available online.
This paper has not been read by Pith yet.
Forward citations
Cited by 16 Pith papers
-
Spin-Weighted Spherical Harmonics Enable Complete and Scalable $\mathrm{E}(3)$-Equivariant Networks
SpinGTP uses Spin-Weighted Spherical Harmonics to complete the Gaunt Tensor Product, achieving full E(3)-equivariance with GTP-like scaling and better handling of chiral and parity-odd cases.
-
Geometric Algebra Meets Cartesian Tensors: Higher-Order Equivariance for Interatomic Potentials
CliffordSTF couples Clifford multivectors to rank-2 and rank-3 symmetric-traceless tensor tracks through bilinear cross-track contractions, lifting force cosine similarity from 0.055 to 0.551 on rMD17 while outperform...
-
ConSolv: Solvent-Conditional Machine Learning Implicit Solvent Potential
ConSolv is a solvent-conditional attention-based MLP trained on experimental and ab initio solvation free energies that generalizes across 66 organic solvents and matches some experimental NMR data.
-
DPA4: Pushing the Accuracy-Cost Frontier of Interatomic Potentials with EMFA SO(2) Convolution
DPA4 is a new SE(3)-equivariant interatomic potential with EMFA SO(2) convolution that sets new accuracy-cost records on Matbench Discovery and SPICE benchmarks using fewer parameters than prior models.
-
Mixing Vector Model for Copolymer Inference via Mixed Integer Linear Programming
Extends the mol-infer MILP framework to copolymers via a mixing vector representation, trains ML predictors achieving R^2 >0.7 on 9/10 datasets, and shows tractable multi-monomer inverse design with external consisten...
-
h-MINT: Modeling Pocket-Ligand Binding with Hierarchical Molecular Interaction Network
h-MINT improves ligand-protein binding affinity prediction by 2-4% and virtual screening metrics by 1-3% via overlapping fragment tokenization and hierarchical modeling.
-
Transferable FB-GNN-MBE Framework for Potential Energy Surfaces: Data-Adaptive Transfer Learning in Deep Learned Many-Body Expansion Theory
FB-GNN-MBE integrates fragment-based graph neural networks into many-body expansion to predict two- and three-body energies for water, phenol, and mixture systems at chemical accuracy, with a teacher-student protocol ...
-
QUIVER: Quantum-Informed Views for Enhanced Representations in Large ML Models
QUIVER augments classical ML models with a quantum Fisher view from VQCs to improve performance on QM9 molecular properties and JetClass jet flavor prediction.
-
SurfDesign: Effective Protein Design on Molecular Surfaces
SurfDesign introduces surface-conditioned protein design via manifold modeling and equivariant message passing on surfaces integrated with pretrained language models, outperforming prior methods on binder and enzyme d...
-
Tensor Channel Equivariant Graph Neural Networks for Molecular Polarizability Prediction
A tensor-channel equivariant GNN based on PaiNN propagates symmetric rank-2 tensor features during message passing and achieves lower full-tensor and anisotropic error than readout-only and MACE baselines on QM7-X geometries.
-
TSAgent: An Agentic Workflow for Autonomous Transition State Search
TSAgent automates transition state searches at DFT accuracy via an agentic loop, reaching 83% success on 100 OC20NEB examples and 70% on 10 held-out cases versus 73% for human experts.
-
Benchmarking Compositional Generalisation for Machine Learning Interatomic Potentials
A new benchmark finds that state-of-the-art ML interatomic potentials struggle with compositional generalization, producing errors an order of magnitude higher on unseen molecular combinations than on training-like cases.
-
AIRA_2: Overcoming Bottlenecks in AI Research Agents
AIRA₂ improves AI research agents via asynchronous multi-GPU workers, hidden consistent evaluation, and interactive ReAct agents, reaching 81.5-83.1% percentile rank on MLE-bench-30 and exceeding human SOTA on 6 of 20...
-
Equivariant Graph Neural Networks Improve Optical Spectra Prediction for Materials Screening
Equivariant GNNs outperform prior models on optical spectra and static permittivity prediction using RPA datasets for materials screening.
-
Loss-Guided Adaptive Scale Refinement for Molecular Force Prediction
Loss-guided adaptive scale refinement on NaCl aqueous system reduces overall force MAE from 399.65 to 381.23 by discovering intermediate scales from initial anchors.
-
A Systematic Survey and Benchmark of Deep Learning for Molecular Property Prediction in the Foundation Model Era
A systematic survey and benchmark of four deep learning paradigms for molecular property prediction that organizes the field, critiques current data practices, and outlines three future directions.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.