MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
This work introduces MiDi, a novel diffusion model for jointly generating molecular graphs and their corresponding 3D arrangement of atoms. Unlike existing methods that rely on predefined rules to determine molecular bonds based on the 3D conformation, MiDi offers an end-to-end differentiable approach that streamlines the molecule generation process. Our experimental results demonstrate the effectiveness of this approach. On the challenging GEOM-DRUGS dataset, MiDi generates 92% of stable molecules, against 6% for the previous EDM model that uses interatomic distances for bond prediction, and 40% using EDM followed by an algorithm that directly optimize bond orders for validity. Our code is available at github.com/cvignac/MiDi.
Key facts. Cited by 96 papers in the map (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 49 of its 59 references that are in the map. First posted on arXiv 17 Feb 2023.
Builds on
49 of 59 references.
Closest ancestors · 9
Same direction, newest first
- 2022Structure-based Drug Design with Equivariant Diffusion Models24 Oct
- 2022Equivariant 3D-Conditional Diffusion Models for Molecular Linker Design11 Oct
- 2022Diffusion Models for Graphs Benefit From Discrete State Spaces4 Oct
- 2022Protein structure generation via folding diffusion30 Sep
- 2022DiGress: Discrete Denoising diffusion for graph generation29 Sep
- 2022MDM: Molecular Diffusion Model for 3D Molecule Generation13 Sep
- 2022Equivariant Diffusion for Molecule Generation in 3D31 Mar
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation6 Mar
- 2022Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equations5 Feb
Other references · 40
- 2017Attention Is All You Needcited by 59,932
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2016Layer Normalizationcited by 5,579
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2022Video Diffusion Modelscited by 2,280
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2015Spatial Transformer Networkscited by 1,423
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
28 more
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 735
- 2020A Generalization of Transformer Networks to Graphscited by 653
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2020Principal Neighbourhood Aggregation for Graph Netscited by 477
- 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
- 2022DiffDock: Diffusion Steps, Twists, and Turns for Molecular Dockingcited by 307
- 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 258
- 2021Pre-training Molecular Graph Representation with 3D Geometrycited by 248
- 2020GEOM: Energy-annotated molecular conformations for property prediction and molecular generationcited by 238
- 2019Efficient Graph Generation with Graph Recurrent Attention Networkscited by 231
- 2021Geometric and Physical Quantities Improve E(3) Equivariant Message Passingcited by 225
- 2020Hierarchical Generation of Molecular Graphs using Structural Motifscited by 204
- 2018Constrained Graph Variational Autoencoders for Molecule Designcited by 182
- 2019Symmetry-adapted generation of 3d point sets for the targeted discovery of moleculescited by 171
- 2021E(n) Equivariant Normalizing Flowscited by 165
- 20213D Infomax improves GNNs for Molecular Property Predictioncited by 146
- 2019GraphNVP: An Invertible Flow Model for Generating Molecular Graphscited by 115
- 2022Equivariance with Learned Canonicalization Functionscited by 106
- 2022Equivariant Energy-Guided SDE for Inverse Molecular Designcited by 72
- 2021Learning to Extend Molecular Scaffolds with Structural Motifscited by 70
- 2022MolGenSurvey: A Systematic Survey in Machine Learning Models for Molecule Designcited by 66
- 2022Unified 2D and 3D Pre-Training of Molecular Representationscited by 45
- 2022Retrieval-based Controllable Molecule Generationcited by 44
- 2022GraphGDP: Generative Diffusion Processes for Permutation Invariant Graph Generationcited by 43
- 2021Top-N: Equivariant set and graph generation without exchangeabilitycited by 36
- 2022Protein Sequence and Structure Co-Design with Equivariant Translationcited by 31
- 2022Set Norm and Equivariant Skip Connections: Putting the Deep in Deep Setscited by 17
Built on by
Citing papers per year: 2023: 11; 2024: 23; 2025: 39; 2026: 23.
96 papers.
Sort:
By area:
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 294
- 2025All-atom Diffusion Transformers: Unified generative modelling of molecules and materialscited by 75
- 2023MolDiff: Addressing the Atom-Bond Inconsistency Problem in 3D Molecule Diffusion Generationcited by 65
- 2024Mixed Continuous and Categorical Flow Matching for 3D De Novo Molecule Generationcited by 53
- 2023Navigating the Design Space of Equivariant Diffusion-Based Generative Models for De Novo 3D Molecule Generationcited by 50
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 40
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matchingcited by 40
- 20233D molecule generation by denoising voxel gridscited by 38
- 2023Efficient and Scalable Graph Generation through Iterative Local Expansioncited by 35
- 2023Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Designcited by 29
- 2024A Survey of Generative AI for de novo Drug Design: New Frontiers in Molecule and Protein Generationcited by 29
- 2024Cometh: A continuous-time discrete-state graph diffusion modelcited by 29
38 more
- 2023Sparse Training of Discrete Diffusion Models for Graph Generationcited by 26
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structurescited by 20
- 2024UniGEM: A Unified Approach to Generation and Property Prediction for Moleculescited by 20
- 2025FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generationcited by 20
- 2024Structure-based drug design by denoising voxel gridscited by 18
- 2025FLOWR: Flow Matching for Structure-Aware De Novo, Interaction- and Fragment-Based Ligand Generationcited by 18
- 2024Generative Modelling of Structurally Constrained Graphscited by 17
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationcited by 17
- 2025Multi-domain Distribution Learning for De Novo Drug Designcited by 15
- 2024Geometric Representation Condition Improves Equivariant Molecule Generationcited by 13
- 2025Applications of Modular Co-Design for De Novo 3D Molecule Generationcited by 13
- 2025Equivariant Neural Diffusion for Molecule Generationcited by 13
- 2024Geometric-Facilitated Denoising Diffusion Model for 3D Molecule Generationcited by 12
- 2025Learning Diffusion Models with Flexible Representation Guidancecited by 12
- 2024Exploring Discrete Flow Matching for 3D De Novo Molecule Generationcited by 11
- 2025TABASCO: A Fast, Simplified Model for Molecular Generation with Improved Physical Qualitycited by 11
- 2023An Equivariant Generative Framework for Molecular Graph-Structure Co-Designcited by 10
- 2024Geometry Informed Tokenization of Molecules for Language Model Generationcited by 10
- 2023SILVR: Guided Diffusion for Molecule Generationcited by 9
- 20243D Vessel Graph Generation Using Denoising Diffusioncited by 9
- 2025Controlled Generation with Equivariant Variational Flow Matchingcited by 9
- 2024LayerDAG: A Layerwise Autoregressive Diffusion Model for Directed Acyclic Graph Generationcited by 8
- 2025Straight-Line Diffusion Model for Efficient 3D Molecular Generationcited by 8
- 2025GEOM-Drugs Revisited: Toward More Chemically Accurate Benchmarks for 3D Molecule Generationcited by 8
- 2025FreeScene: Mixed Graph Diffusion for 3D Scene Synthesis from Free Promptscited by 8
- 2025Unified Guidance for Geometry-Conditioned Molecular Generationcited by 7
- 2025Bures-Wasserstein Flow Matching for Graph Generationcited by 7
- 2024Zero Shot Molecular Generation via Similarity Kernelscited by 6
- 2024E(3)-equivariant models cannot learn chirality: Field-based molecular generationcited by 6
- 2025SynCoGen: Synthesizable 3D Molecule Generation via Joint Reaction and Coordinate Modelingcited by 5
- 2025Controllable Graph Generation with Diffusion Models via Inference-Time Tree Search Guidancecited by 5
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generationcited by 5
- 2025Graph Diffusion that can Insert and Deletecited by 4
- 2025MolGuidance: Advanced Guidance Strategies for Conditional Molecular Generation with Flow Matchingcited by 4
- 2024NEBULA: Neural Empirical Bayes Under Latent Representations for Efficient and Controllable Design of Molecular Librariescited by 3
- 2025Disentangling Total-Variance and Signal-to-Noise-Ratio Improves Diffusion Modelscited by 3
- 2025Clifford Group Equivariant Diffusion Models for 3D Molecular Generationcited by 3
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignmentcited by 3
50 most cited of 96 All 96 (JSON)
- 2026Zatom-2: Multitask Pretraining on Atomistic Data for Generative Modeling across Domains8 Oct 2026
- 2026TreeRef-BFN: Equivariance-Free De Novo Molecule Generation based on 2D Topology and Internal 3D Geometry26 Sep 2026
- 2026QALPA: Property-guided diffusion modeling for efficient exploration of chemical spaces of flexible molecules15 Sep 2026
- 2026A Systematic Evaluation of Molecule Generation Models for De Novo Drug Design: From Benchmarks to Practical Insights9 Sep 2026
- 2026PocketVE: Stable and Property-Guided Structure-Based Drug Design with Variance-Exploding Diffusion8 Sep 2026
- 2026Generation of High-Level Concepts in 3D Scene Graphs via Autoregressive Diffusion28 Aug 2026
- 2026DiverseDiT++: Quantifying, Analyzing, and Promoting Representation Diversity in Diffusion Transformers4 Aug 2026
- 2026Boltzmann-Expected Molecular Design with Decoupled Annealing Flows21 Jul 2026
- 2026DBMol: Design of High-Affinity, Target-Specific Small Molecules through Structure Prediction Models21 Jul 2026
- 2026MotifRole-Diff: Risk-Optimal Role-Aware Corruption for Masked Molecular Graph Diffusion20 Jul 2026
- 2026SinAE: A Single-Architecture Flow-Matching Autoencoder for Cross-Domain Atomic Systems14 Jul 2026
- 2026Autoregressive latent diffusion for 3D molecule generation10 Jul 2026
38 more
- 2026DiPhon: Diffusion on Graphons for Scalable Graph Generation8 Jul 2026
- 2026Sesame: Structure-Aware Molecular Generation via Spatial Density-Map Conditioning22 Jun 2026
- 2026Generative Molecular Morphing for Flexible-Size Design via Unbalanced Optimal Transport5 Jun 2026
- 2026SciCore-Mol: Augmenting Large Language Models with Pluggable Molecular Cognition Modules21 May 2026
- 2026Generating Physically Consistent Molecules with Energy-Based Models18 May 2026
- 2026Toward Better Geometric Representations for Molecule Generative Models8 May 2026
- 2026Permutation-Symmetrized Diffusion for Unconditional Molecular Generation24 Mar 2026
- 2026VesselTok: Tokenizing Vessel-like 3D Biomedical Graph Representations for Reconstruction and Generation19 Mar 2026
- 2026Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials24 Feb 2026
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation16 Feb 2026
- 2026MacroGuide: Topological Guidance for Macrocycle Generation16 Feb 2026
- 2025MolGuidance: Advanced Guidance Strategies for Conditional Molecular Generation with Flow Matching13 Dec 2025
- 2025MolSculpt: Sculpting 3D Molecular Geometries from Chemical Syntax9 Dec 2025
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025When Diffusion Breaks Constraints: Sequential Autoregressive Generation with RL and MCTS1 Dec 2025
- 2025FlexiFlow: decomposable flow matching for generation of flexible molecular ensemble21 Nov 2025
- 2025Pharmacophore-based design by learning on voxel grids19 Nov 2025
- 2025InertialAR: Autoregressive 3D Molecule Generation with Inertial Frames31 Oct 2025
- 2025HyperDiffusionFields (HyDiF): Diffusion-Guided Hypernetworks for Learning Implicit Molecular Neural Fields20 Oct 2025
- 2025Controllable Graph Generation with Diffusion Models via Inference-Time Tree Search Guidance12 Oct 2025
- 2025Predictive Feature Caching for Training-free Acceleration of Molecular Geometry Generation6 Oct 2025
- 2025FLOWR.root: A flow matching based foundation model for joint multi-purpose structure-aware 3D ligand generation and affinity prediction2 Oct 2025
- 2025Multi-domain Distribution Learning for De Novo Drug Design25 Aug 2025
- 2025FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generation18 Aug 2025
- 2025Data-driven particle dynamics: Structure-preserving coarse-graining for emergent behavior in non-equilibrium systems18 Aug 2025
- 2025SynCoGen: Synthesizable 3D Molecule Generation via Joint Reaction and Coordinate Modeling16 Jul 2025
- 2025Do we need equivariant models for molecule generation?13 Jul 2025
- 2025Learning Diffusion Models with Flexible Representation Guidance11 Jul 2025
- 2025Semantically Consistent Discrete Diffusion for 3D Biological Graph Modeling7 Jul 2025
- 2025TABASCO: A Fast, Simplified Model for Molecular Generation with Improved Physical Quality1 Jul 2025
- 2025Unraveling the Potential of Diffusion Models in Small Molecule Generation25 Jun 2025
- 2025Controlled Generation with Equivariant Variational Flow Matching23 Jun 2025
- 2025Generating Directed Graphs with Dual Attention and Asymmetric Encoding19 Jun 2025
- 2025Bures-Wasserstein Flow Matching for Graph Generation16 Jun 2025
- 2025Equivariant Neural Diffusion for Molecule Generation12 Jun 2025
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment11 Jun 2025
- 2025Graph Diffusion that can Insert and Delete6 Jun 2025
- 2025FreeScene: Mixed Graph Diffusion for 3D Scene Synthesis from Free Prompts3 Jun 2025
50 newest of 96 All 96 (JSON)
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 294
- 2025All-atom Diffusion Transformers: Unified generative modelling of molecules and materialscited by 75
- 2024Mixed Continuous and Categorical Flow Matching for 3D De Novo Molecule Generationcited by 53
- 2023Navigating the Design Space of Equivariant Diffusion-Based Generative Models for De Novo 3D Molecule Generationcited by 50
- 2024Cometh: A continuous-time discrete-state graph diffusion modelcited by 29
- 2023Sparse Training of Discrete Diffusion Models for Graph Generationcited by 26
- 2025FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generationcited by 20
- 2024Structure-based drug design by denoising voxel gridscited by 18
- 2024Generative Modelling of Structurally Constrained Graphscited by 17
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationcited by 17
- 2025Multi-domain Distribution Learning for De Novo Drug Designcited by 15
- 2025Equivariant Neural Diffusion for Molecule Generationcited by 13
12 most cited of 47 All 47 (JSON)
- 2026QALPA: Property-guided diffusion modeling for efficient exploration of chemical spaces of flexible molecules15 Sep 2026
- 2026DiverseDiT++: Quantifying, Analyzing, and Promoting Representation Diversity in Diffusion Transformers4 Aug 2026
- 2026MotifRole-Diff: Risk-Optimal Role-Aware Corruption for Masked Molecular Graph Diffusion20 Jul 2026
- 2026SinAE: A Single-Architecture Flow-Matching Autoencoder for Cross-Domain Atomic Systems14 Jul 2026
- 2026Autoregressive latent diffusion for 3D molecule generation10 Jul 2026
- 2026DiPhon: Diffusion on Graphons for Scalable Graph Generation8 Jul 2026
- 2026Permutation-Symmetrized Diffusion for Unconditional Molecular Generation24 Mar 2026
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation16 Feb 2026
- 2025MolGuidance: Advanced Guidance Strategies for Conditional Molecular Generation with Flow Matching13 Dec 2025
- 2025MolSculpt: Sculpting 3D Molecular Geometries from Chemical Syntax9 Dec 2025
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025When Diffusion Breaks Constraints: Sequential Autoregressive Generation with RL and MCTS1 Dec 2025
12 newest of 47 All 47 (JSON)
- 2023MolDiff: Addressing the Atom-Bond Inconsistency Problem in 3D Molecule Diffusion Generationcited by 65
- 2023Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Spacecited by 40
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matchingcited by 40
- 20233D molecule generation by denoising voxel gridscited by 38
- 2023Efficient and Scalable Graph Generation through Iterative Local Expansioncited by 35
- 2023Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Designcited by 29
- 2024A Survey of Generative AI for de novo Drug Design: New Frontiers in Molecule and Protein Generationcited by 29
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structurescited by 20
- 2024UniGEM: A Unified Approach to Generation and Property Prediction for Moleculescited by 20
- 2025FLOWR: Flow Matching for Structure-Aware De Novo, Interaction- and Fragment-Based Ligand Generationcited by 18
- 2024Geometric Representation Condition Improves Equivariant Molecule Generationcited by 13
- 2025Applications of Modular Co-Design for De Novo 3D Molecule Generationcited by 13
12 most cited of 48 All 48 (JSON)
- 2026Zatom-2: Multitask Pretraining on Atomistic Data for Generative Modeling across Domains8 Oct 2026
- 2026TreeRef-BFN: Equivariance-Free De Novo Molecule Generation based on 2D Topology and Internal 3D Geometry26 Sep 2026
- 2026A Systematic Evaluation of Molecule Generation Models for De Novo Drug Design: From Benchmarks to Practical Insights9 Sep 2026
- 2026PocketVE: Stable and Property-Guided Structure-Based Drug Design with Variance-Exploding Diffusion8 Sep 2026
- 2026Boltzmann-Expected Molecular Design with Decoupled Annealing Flows21 Jul 2026
- 2026DBMol: Design of High-Affinity, Target-Specific Small Molecules through Structure Prediction Models21 Jul 2026
- 2026Sesame: Structure-Aware Molecular Generation via Spatial Density-Map Conditioning22 Jun 2026
- 2026Generative Molecular Morphing for Flexible-Size Design via Unbalanced Optimal Transport5 Jun 2026
- 2026SciCore-Mol: Augmenting Large Language Models with Pluggable Molecular Cognition Modules21 May 2026
- 2026Generating Physically Consistent Molecules with Energy-Based Models18 May 2026
- 2026Toward Better Geometric Representations for Molecule Generative Models8 May 2026
- 2026VesselTok: Tokenizing Vessel-like 3D Biomedical Graph Representations for Reconstruction and Generation19 Mar 2026
12 newest of 48 All 48 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024Diffusion Models in textitDe Novo Drug Design
- 2023Coarse-to-Fine: a Hierarchical Diffusion Model for Molecule Generation in 3D
- 2023Geometry-Complete Diffusion for 3D Molecule Generation and Optimization
- 2025DMol: A Highly Efficient and Chemical Motif-Preserving Molecule Generation Platform
- 2025Diffusion Models at the Drug Discovery Frontier: A Review on Generating Small Molecules versus Therapeutic Peptides
- 2026Equivariant Asynchronous Diffusion: An Adaptive Denoising Schedule for Accelerated Molecular Conformation Generation
- 2025Sampling 3D Molecular Conformers with Diffusion Transformers
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generation