Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Molecules
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 254 papers in the map (data through 7 Oct 2026). First posted on arXiv 28 Nov 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 27; 2022: 47; 2023: 61; 2024: 39; 2025: 42; 2026: 38.
254 papers.
Sort:
By area:
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 467
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 438
- 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
- 2021OGB-LSC: A Large-Scale Challenge for Machine Learning on Graphscited by 328
- 2021Crystal Diffusion Variational Autoencoder for Periodic Material Generationcited by 268
- 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 258
- 2022Learning Local Equivariant Representations for Large-Scale Atomistic Dynamicscited by 254
- 2021Geometric and Physical Quantities Improve E(3) Equivariant Message Passingcited by 225
- 2023EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representationscited by 201
- 2022Graph neural networks for materials science and chemistrycited by 181
- 2022TorchMD-NET: Equivariant Transformers for Neural Network based Molecular Potentialscited by 179
38 more
- 2021Spherical Message Passing for 3D Graph Networkscited by 166
- 20213D Infomax improves GNNs for Molecular Property Predictioncited by 146
- 2023Crystal Structure Prediction by Joint Equivariant Diffusioncited by 142
- 2022Pre-training via Denoising for Molecular Property Predictioncited by 124
- 2021Geometric Deep Learning on Molecular Representationscited by 113
- 2023Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNscited by 113
- 2022The Open Catalyst 2022 (OC22) Dataset and Challenges for Oxide Electrocatalystscited by 95
- 2022Periodic Graph Transformers for Crystal Material Property Predictioncited by 87
- 2021DIG: A Turnkey Library for Diving into Graph Deep Learning Researchcited by 76
- 2022Equivariant Energy-Guided SDE for Inverse Molecular Designcited by 72
- 2023FAENet: Frame Averaging Equivariant GNN for Materials Modelingcited by 72
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2022Molecular Geometry Pretraining with SE(3)-Invariant Denoising Distance Matchingcited by 67
- 2022MolGenSurvey: A Systematic Survey in Machine Learning Models for Molecule Designcited by 66
- 2022GemNet-OC: Developing Graph Neural Networks for Large and Diverse Molecular Simulation Datasetscited by 61
- 2021ForceNet: A Graph Neural Network for Large-Scale Quantum Calculationscited by 58
- 2023TensorNet: Cartesian Tensor Representations for Efficient Learning of Molecular Potentialscited by 53
- 2021Rotation Invariant Graph Neural Networks using Spin Convolutionscited by 52
- 2025SymmCD: Symmetry-Preserving Crystal Generation with Diffusion Modelscited by 48
- 2022Spherical Channels for Modeling Atomic Interactionscited by 47
- 2023Ewald-based Long-Range Message Passing for Molecular Graphscited by 46
- 2021Simple GNN Regularisation for 3D Molecular Property Prediction & Beyondcited by 44
- 2021Molformer: Motif-based Transformer on 3D Heterogeneous Molecular Graphscited by 44
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 40
- 2024The Importance of Being Scalable: Improving the Speed and Accuracy of Neural Network Interatomic Potentials Across Chemical Domainscited by 40
- 2023Smooth, exact rotational symmetrization for deep learning on point cloudscited by 39
- 2023Efficient and Equivariant Graph Networks for Predicting Quantum Hamiltoniancited by 38
- 2023Efficient Approximations of Complete Interatomic Potentials for Crystal Property Predictioncited by 37
- 2023Continual Learning on Dynamic Graphs via Parameter Isolationcited by 35
- 2024Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Productscited by 35
- 2022Crystal Twins: Self-supervised Learning for Crystalline Material Property Predictioncited by 34
- 2021Graph Neural Networks Accelerated Molecular Dynamicscited by 33
- 2023Scaling the leading accuracy of deep equivariant models to biomolecular simulations of realistic sizecited by 33
- 2023Fractional Denoising for 3D Molecular Pre-trainingcited by 33
- 2021Learning 3D Representations of Molecular Chirality with Invariance to Bond Rotationscited by 29
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2021Informing Geometric Deep Learning with Electronic Interactions to Accelerate Quantum Chemistrycited by 27
- 2021Learning neural network potentials from experimental data via Differentiable Trajectory Reweightingcited by 26
50 most cited of 254 All 254 (JSON)
- 2026Measuring trainable degrees of freedom in materials graph neural networks: a random-subspace intrinsic dimension analysis28 Sep 2026
- 2026GSF-chi: Global Stereochemical Fields for Chiral Graph Transformers11 Sep 2026
- 2026Hessian-based molecular conformation augmentation for a scalable and efficient strategy of machine learning interatomic potentials4 Sep 2026
- 2026Learning Materials Properties from Scarce Labels and Unlabeled Crystals31 Aug 2026
- 2026Temperature-Driven Sequential Modeling for the Prediction of Annual Power Conversion Efficiency Profiles of Organic Photovoltaic Materials: Douala Case Study9 Aug 2026
- 2026Chem World: A Large-Scale Benchmark and Physics-Informed Framework for Trustworthy Chemical Property Prediction30 Jul 2026
- 2026Dual-Level Atomic and Coordination Geometry Learning for Crystal Property Prediction Using Graph Neural Networks17 Jul 2026
- 2026Spin-Weighted Spherical Harmonics Enable Complete and Scalable E(3)-Equivariant Networks1 Jul 2026
- 2026Optimizing Expert-Designed Crystal Graph Networks for Band-Gap Prediction with an Autonomous LLM Research Loop29 Jun 2026
- 2026Geometric Algebra Meets Cartesian Tensors: Higher-Order Equivariance for Interatomic Potentials28 Jun 2026
- 2026Equivariant Graph Neural Networks Improve Optical Spectra Prediction for Materials Screening17 Jun 2026
- 2026Autonomous heterogeneous catalyst discovery with a self-evolving multi-agent digital twin3 Jun 2026
38 more
- 2026QUIVER: Quantum-Informed Views for Enhanced Representations in Large ML Models1 Jun 2026
- 2026MLIPilot: LLM-Driven Auto-Research for Machine-Learned Interatomic Potentials29 May 2026
- 2026Mixing Vector Model for Copolymer Inference via Mixed Integer Linear Programming28 May 2026
- 2026SurfDesign: Effective Protein Design on Molecular Surfaces25 May 2026
- 2026Machine Learning Multiscale Interactions25 May 2026
- 2026TriForces: Augmenting Atomistic GNNs for Transferable Representations20 May 2026
- 2026CatalyticMLLM: A Graph-Text Multimodal Large Language Model for Catalytic Materials17 May 2026
- 2026Tensor Channel Equivariant Graph Neural Networks for Molecular Polarizability Prediction16 May 2026
- 2026Benchmarking Compositional Generalisation for Machine Learning Interatomic Potentials9 May 2026
- 2026h-MINT: Modeling Pocket-Ligand Binding with Hierarchical Molecular Interaction Network25 Apr 2026
- 2026A Systematic Survey and Benchmark of Deep Learning for Molecular Property Prediction in the Foundation Model Era17 Apr 2026
- 2026AIRA_2: Overcoming Bottlenecks in AI Research Agents27 Mar 2026
- 2026Universal and efficient graph neural networks with dynamic attention for machine learning interatomic potentials24 Mar 2026
- 2026Proof-Carrying Materials: Falsifiable Safety Certificates for Machine-Learned Interatomic Potentials12 Mar 2026
- 2026Flexible Cutoff Learning: Optimizing Machine Learning Potentials After Training10 Mar 2026
- 2026Information Routing in Atomistic Foundation Models: How Task Alignment and Equivariance Shape Linear Disentanglement3 Mar 2026
- 2026PolyMon: A Unified Framework for Polymer Property Prediction2 Mar 2026
- 2026MatRIS: Toward Reliable and Efficient Pretrained Machine Learning Interatomic Potentials2 Mar 2026
- 2026Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials24 Feb 2026
- 2026Learning Molecular Chirality via Chiral Determinant Kernels7 Feb 2026
- 2026Electron-Informed Coarse-Graining Molecular Representation Learning for Real-World Molecular Physics6 Feb 2026
- 2026Molecular Representations in Implicit Functional Space via Hyper-Networks29 Jan 2026
- 2026Topology-Aware Multiscale Mixture of Experts for Efficient Molecular Property Prediction19 Jan 2026
- 2026M^4olGen: Multi-Agent, Multi-Stage Molecular Generation under Precise Multi-Property Constraints15 Jan 2026
- 2026Surface-based Molecular Design with Multi-modal Flow Matching8 Jan 2026
- 2026Quantized SO(3)-Equivariant Graph Neural Networks for Efficient Molecular Property Prediction5 Jan 2026
- 2025QE-Catalytic: A Graph-Language Multimodal Base Model for Relaxed-Energy Prediction in Catalytic Adsorption23 Dec 2025
- 2025GoMS: Graph of Molecule Substructure Network for Molecule Property Prediction13 Dec 2025
- 2025Generalization of Long-Range Machine Learning Potentials in Complex Chemical Spaces8 Dec 2025
- 2025Multi-Scale Protein Structure Modelling with Geometric Graph U-Nets7 Dec 2025
- 2025QuantumCanvas: A Multimodal Benchmark for Visual Learning of Atomic Interactions1 Dec 2025
- 2025Toward bilipshiz geometric models13 Nov 2025
- 2025Benchmarking GNNs for OOD Materials Property Prediction with Uncertainty Quantification12 Nov 2025
- 2025Direct Molecular Polarizability Prediction with SO(3) Equivariant Local Frame GNNs10 Nov 2025
- 2025Learning 3D Anisotropic Noise Distributions Improves Molecular Force Field Modeling25 Oct 2025
- 2025Layer-to-Layer Knowledge Mixing in Graph Neural Network for Chemical Property Prediction23 Oct 2025
- 2025Mixing Configurations for Downstream Prediction22 Oct 2025
- 20253D-GSRD: 3D Molecular Graph Auto-Encoder with Selective Re-mask Decoding19 Oct 2025
50 newest of 254 All 254 (JSON)
- 2026AIRA_2: Overcoming Bottlenecks in AI Research Agentscited by 10
- 2024Adsorb-Agent: Autonomous Identification of Stable Adsorption Configurations via Large Language Model Agentcited by 8
- 2026M^4olGen: Multi-Agent, Multi-Stage Molecular Generation under Precise Multi-Property Constraintscited by 3
- 2026MLIPilot: LLM-Driven Auto-Research for Machine-Learned Interatomic Potentialscited by 3
- 2026MLIPilot: LLM-Driven Auto-Research for Machine-Learned Interatomic Potentials29 May 2026
- 2026AIRA_2: Overcoming Bottlenecks in AI Research Agents27 Mar 2026
- 2026M^4olGen: Multi-Agent, Multi-Stage Molecular Generation under Precise Multi-Property Constraints15 Jan 2026
- 2024Adsorb-Agent: Autonomous Identification of Stable Adsorption Configurations via Large Language Model Agent22 Oct 2024
- 2023Crystal Structure Prediction by Joint Equivariant Diffusioncited by 142
- 2023Diffusion probabilistic models enhance variational autoencoder for crystal structure generative modelingcited by 17
- 2025Controlled Generation with Equivariant Variational Flow Matchingcited by 9
- 2024Chemistry-Inspired Diffusion with Non-Differentiable Guidancecited by 6
- 2025Controlled Generation with Equivariant Variational Flow Matching23 Jun 2025
- 2024Chemistry-Inspired Diffusion with Non-Differentiable Guidance9 Oct 2024
- 2023Diffusion probabilistic models enhance variational autoencoder for crystal structure generative modeling4 Aug 2023
- 2023Crystal Structure Prediction by Joint Equivariant Diffusion30 Jul 2023
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 467
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 438
- 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
- 2021OGB-LSC: A Large-Scale Challenge for Machine Learning on Graphscited by 328
- 2021Crystal Diffusion Variational Autoencoder for Periodic Material Generationcited by 268
- 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 258
- 2022Learning Local Equivariant Representations for Large-Scale Atomistic Dynamicscited by 254
- 2021Geometric and Physical Quantities Improve E(3) Equivariant Message Passingcited by 225
- 2023EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representationscited by 201
- 2022Graph neural networks for materials science and chemistrycited by 181
- 2022TorchMD-NET: Equivariant Transformers for Neural Network based Molecular Potentialscited by 179
12 most cited of 245 All 245 (JSON)
- 2026Measuring trainable degrees of freedom in materials graph neural networks: a random-subspace intrinsic dimension analysis28 Sep 2026
- 2026GSF-chi: Global Stereochemical Fields for Chiral Graph Transformers11 Sep 2026
- 2026Hessian-based molecular conformation augmentation for a scalable and efficient strategy of machine learning interatomic potentials4 Sep 2026
- 2026Learning Materials Properties from Scarce Labels and Unlabeled Crystals31 Aug 2026
- 2026Temperature-Driven Sequential Modeling for the Prediction of Annual Power Conversion Efficiency Profiles of Organic Photovoltaic Materials: Douala Case Study9 Aug 2026
- 2026Chem World: A Large-Scale Benchmark and Physics-Informed Framework for Trustworthy Chemical Property Prediction30 Jul 2026
- 2026Dual-Level Atomic and Coordination Geometry Learning for Crystal Property Prediction Using Graph Neural Networks17 Jul 2026
- 2026Spin-Weighted Spherical Harmonics Enable Complete and Scalable E(3)-Equivariant Networks1 Jul 2026
- 2026Optimizing Expert-Designed Crystal Graph Networks for Band-Gap Prediction with an Autonomous LLM Research Loop29 Jun 2026
- 2026Geometric Algebra Meets Cartesian Tensors: Higher-Order Equivariance for Interatomic Potentials28 Jun 2026
- 2026Equivariant Graph Neural Networks Improve Optical Spectra Prediction for Materials Screening17 Jun 2026
- 2026Autonomous heterogeneous catalyst discovery with a self-evolving multi-agent digital twin3 Jun 2026
12 newest of 245 All 245 (JSON)