Equivariant Neural Diffusion for Molecule Generation
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
We introduce Equivariant Neural Diffusion (END), a novel diffusion model for molecule generation in 3D that is equivariant to Euclidean transformations. Compared to current state-of-the-art equivariant diffusion models, the key innovation in END lies in its learnable forward process for enhanced generative modelling. Rather than pre-specified, the forward process is parameterized through a time- and data-dependent transformation that is equivariant to rigid transformations. Through a series of experiments on standard molecule generation benchmarks, we demonstrate the competitive performance of END compared to several strong baselines for both unconditional and conditional generation.
Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 38 of its 56 references that are in the map. First posted on arXiv 12 Jun 2025.
Builds on
38 of 56 references.
Closest ancestors · 12
Same direction, newest first
- 2024Conditional Synthesis of 3D Molecules with Time Correction Sampler1 Nov
- 2023Equivariant Flow Matching with Hybrid Probability Transport12 Dec
- 2023Navigating the Design Space of Equivariant Diffusion-Based Generative Models for De Novo 3D Molecule Generation29 Sep
- 2023Practical and Asymptotically Exact Conditional Sampling in Diffusion Models30 Jun
- 2023Coarse-to-Fine: a Hierarchical Diffusion Model for Molecule Generation in 3D5 May
- 2023Geometric Latent Diffusion Models for 3D Molecule Generation2 May
- 2023EigenFold: Generative Protein Structure Prediction with Diffusion Models5 Apr
- 2023MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation17 Feb
- 2022Equivariant 3D-Conditional Diffusion Models for Molecular Linker Design11 Oct
- 2022Flow Matching for Generative Modeling6 Oct
- 2022Equivariant Diffusion for Molecule Generation in 3D31 Mar
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation6 Mar
Other references · 26
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2021Variational Diffusion Modelscited by 1,153
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applicationscited by 1,114
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 708
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 474
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 441
- 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 373
- 2020GEOM: Energy-annotated molecular conformations for property prediction and molecular generationcited by 243
- 2020Machine Learning Force Fieldscited by 207
14 more
- 2019Symmetry-adapted generation of 3d point sets for the targeted discovery of moleculescited by 172
- 2021E(n) Equivariant Normalizing Flowscited by 165
- 2022Diffusion-based Molecule Generation with Informative Prior Bridgescited by 106
- 2021Inverse design of 3d molecular structures with conditional generative neural networkscited by 93
- 2021Generating 3D Molecules Conditional on Receptor Binding Sites with Deep Generative Modelscited by 73
- 2022Equivariant Energy-Guided SDE for Inverse Molecular Designcited by 73
- 2023Timewarp: Transferable Acceleration of Molecular Dynamics by Learning Time-Coarsened Dynamicscited by 66
- 2023MolDiff: Addressing the Atom-Bond Inconsistency Problem in 3D Molecule Diffusion Generationcited by 65
- 2020Reinforcement Learning for Molecular Design Guided by Quantum Mechanicscited by 51
- 2024Neural Flow Diffusion Models: Learnable Forward Process for Improved Diffusion Modellingcited by 45
- 20233D molecule generation by denoising voxel gridscited by 40
- 2023Neural Diffusion Modelscited by 21
- 2023DiffEnc: Variational Diffusion with a Learned Encodercited by 17
- 2023Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for 3D Molecule Generationcited by 17
Built on by
Citing papers per year: 2025: 4; 2026: 9.
13 papers.
Sort:
By area:
- 2025Controlled Generation with Equivariant Variational Flow Matchingcited by 9
- 2025Kinetic Langevin Diffusion for Crystalline Materials Generationcited by 9
- 2025Guiding Diffusion Models with Reinforcement Learning for Stable Molecule Generationcited by 4
- 20263D Molecule Generation from Rigid Motifs via SE(3) Flowscited by 3
- 2026Elign: Equivariant Diffusion Model Alignment from Foundational Machine Learning Force Fieldscited by 3
- 2026GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion Modelscited by 1
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learningcited by 1
- 2025Challenges in Non-Polymeric Crystal Structure Prediction: Why a Geometric, Permutation-Invariant Loss is Neededcited by 0
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutionscited by 0
- 2026Generating Physically Consistent Molecules with Energy-Based Modelscited by 0
- 2026Uncertainty Estimation for Molecular Diffusion Modelscited by 0
- 2026Renormalization Group Flow Matching for Scalable Local Generative Modelingcited by 0
- 2026A Systematic Evaluation of Molecule Generation Models for De Novo Drug Design: From Benchmarks to Practical Insights9 Sep 2026
- 2026Renormalization Group Flow Matching for Scalable Local Generative Modeling24 Aug 2026
- 2026Uncertainty Estimation for Molecular Diffusion Models11 Jun 2026
- 2026AtomComposer: Discovering Chemical Space from First Principles with Reinforcement Learning27 May 2026
- 2026Generating Physically Consistent Molecules with Energy-Based Models18 May 2026
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutions10 Feb 2026
- 2026GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion Models30 Jan 2026
- 2026Elign: Equivariant Diffusion Model Alignment from Foundational Machine Learning Force Fields29 Jan 2026
- 20263D Molecule Generation from Rigid Motifs via SE(3) Flows23 Jan 2026
- 2025Challenges in Non-Polymeric Crystal Structure Prediction: Why a Geometric, Permutation-Invariant Loss is Needed31 Aug 2025
- 2025Guiding Diffusion Models with Reinforcement Learning for Stable Molecule Generation22 Aug 2025
- 2025Kinetic Langevin Diffusion for Crystalline Materials Generation4 Jul 2025
1 more
- 2025Controlled Generation with Equivariant Variational Flow Matchingcited by 9
- 2025Kinetic Langevin Diffusion for Crystalline Materials Generationcited by 9
- 2025Guiding Diffusion Models with Reinforcement Learning for Stable Molecule Generationcited by 4
- 2026Elign: Equivariant Diffusion Model Alignment from Foundational Machine Learning Force Fieldscited by 3
- 2026GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion Modelscited by 1
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutionscited by 0
- 2026Uncertainty Estimation for Molecular Diffusion Modelscited by 0
- 2026Renormalization Group Flow Matching for Scalable Local Generative Modelingcited by 0
- 2026Renormalization Group Flow Matching for Scalable Local Generative Modeling24 Aug 2026
- 2026Uncertainty Estimation for Molecular Diffusion Models11 Jun 2026
- 2026DEGMC: Denoising Diffusion Models Based on Riemannian Equivariant Group Morphological Convolutions10 Feb 2026
- 2026GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion Models30 Jan 2026
- 2026Elign: Equivariant Diffusion Model Alignment from Foundational Machine Learning Force Fields29 Jan 2026
- 2025Guiding Diffusion Models with Reinforcement Learning for Stable Molecule Generation22 Aug 2025
- 2025Kinetic Langevin Diffusion for Crystalline Materials Generation4 Jul 2025
- 2025Controlled Generation with Equivariant Variational Flow Matching23 Jun 2025
- 20263D Molecule Generation from Rigid Motifs via SE(3) Flowscited by 3
- 2025Challenges in Non-Polymeric Crystal Structure Prediction: Why a Geometric, Permutation-Invariant Loss is Neededcited by 0
- 2026Generating Physically Consistent Molecules with Energy-Based Modelscited by 0
- 2026A Systematic Evaluation of Molecule Generation Models for De Novo Drug Design: From Benchmarks to Practical Insightscited by 0
- 2026A Systematic Evaluation of Molecule Generation Models for De Novo Drug Design: From Benchmarks to Practical Insights9 Sep 2026
- 2026Generating Physically Consistent Molecules with Energy-Based Models18 May 2026
- 20263D Molecule Generation from Rigid Motifs via SE(3) Flows23 Jan 2026
- 2025Challenges in Non-Polymeric Crystal Structure Prediction: Why a Geometric, Permutation-Invariant Loss is Needed31 Aug 2025
Similar, not cited
Close in wording, no citation either way
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment
- 2025Frame-based Equivariant Diffusion Models for 3D Molecular Generation
- 2024SymDiff: Equivariant Diffusion via Stochastic Symmetrisation
- 2025D3MES: Diffusion Transformer with multihead equivariant self-attention for 3D molecule generation
- 2026Quotient-Space Diffusion Models
- 2026Autoregressive latent diffusion for 3D molecule generation
- 2022MDM: Molecular Diffusion Model for 3D Molecule Generation
- 2024Equivariant Blurring Diffusion for Hierarchical Molecular Conformer Generation