Geometric Latent Diffusion Models for 3D Molecule Generation
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Generative models, especially diffusion models (DMs), have achieved promising results for generating feature-rich geometries and advancing foundational science problems such as molecule design. Inspired by the recent huge success of Stable (latent) Diffusion models, we propose a novel and principled method for 3D molecule generation named Geometric Latent Diffusion Models (GeoLDM). GeoLDM is the first latent DM model for the molecular geometry domain, composed of autoencoders encoding structures into continuous latent codes and DMs operating in the latent space. Our key innovation is that for modeling the 3D molecular geometries, we capture its critical roto-translational equivariance constraints by building a point-structured latent space with both invariant scalars and equivariant tensors. Extensive experiments demonstrate that GeoLDM can consistently achieve better performance on multiple molecule generation benchmarks, with up to 7% improvement for the valid percentage of large biomolecules. Results also demonstrate GeoLDM's higher capacity for controllable generation thanks to the latent modeling. Code is provided at urlhttps://github.com/MinkaiXu/GeoLDM.
Key facts. Cited by 186 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 49 of its 62 references that are in the map. First posted on arXiv 2 May 2023.
Builds on
49 of 62 references.
Closest ancestors · 5
Same direction, newest first
- 2022Diffusion probabilistic modeling of protein backbones in 3D for the motif-scaffolding problem8 Jun
- 2022Protein Structure and Sequence Generation with Equivariant Denoising Diffusion Probabilistic Models26 May
- 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 · 44
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,937
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,387
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
32 more
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2021E(n) Equivariant Graph Neural Networkscited by 965
- 2016Pixel Recurrent Neural Networkscited by 816
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 746
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 708
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 658
- 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 652
- 2021Score-based Generative Modeling in Latent Spacecited by 636
- 2020SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networkscited by 607
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
- 2022LION: Latent Point Diffusion Models for 3D Shape Generationcited by 553
- 20183D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Datacited by 354
- 2020Learning from Protein Structure with Geometric Vector Perceptronscited by 313
- 2020GraphAF: a Flow-based Autoregressive Model for Molecular Graph Generationcited by 305
- 2019Cormorant: Covariant Molecular Neural Networkscited by 263
- 2019Diagnosing and Enhancing VAE Modelscited by 193
- 2018Constrained Graph Variational Autoencoders for Molecule Designcited by 185
- 2020MoFlow: An Invertible Flow Model for Generating Molecular Graphscited by 185
- 2022Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pocketscited by 183
- 2019Symmetry-adapted generation of 3d point sets for the targeted discovery of moleculescited by 172
- 2019From Variational to Deterministic Autoencoderscited by 133
- 2022Diffusion-based Molecule Generation with Informative Prior Bridgescited by 106
- 2021D2C: Diffusion-Denoising Models for Few-shot Conditional Generationcited by 96
- 2021Inverse design of 3d molecular structures with conditional generative neural networkscited by 93
- 2019FlowSeq: Non-Autoregressive Conditional Sequence Generation with Generative Flowcited by 89
- 2020ATOM3D: Tasks On Molecules in Three Dimensionscited by 74
- 2020A Contrastive Learning Approach for Training Variational Autoencoder Priorscited by 66
- 2022DiffBP: Generative Diffusion of 3D Molecules for Target Protein Bindingcited by 58
- 2022Unsupervised Learning of Group Invariant and Equivariant Representationscited by 39
- 2021Learning to design drug-like molecules in three-dimensional space using deep generative modelscited by 37
- 2020Generating 3D Molecular Structures Conditional on a Receptor Binding Site with Deep Generative Modelscited by 33
- 2021Auto-Encoding Molecular Conformationscited by 8
Built on by
Citing papers per year: 2023: 11; 2024: 68; 2025: 66; 2026: 41.
186 papers.
Sort:
By area:
- 2024Derivative-Free Guidance in Continuous and Discrete Diffusion Models with Soft Value-Based Decodingcited by 136
- 2024TFG: Unified Training-Free Guidance for Diffusion Modelscited by 107
- 2024Discrete-state Continuous-time Diffusion for Graph Generationcited by 76
- 2025All-atom Diffusion Transformers: Unified generative modelling of molecules and materialscited by 75
- 2023Equivariant Flow Matching with Hybrid Probability Transportcited by 60
- 2024Minimax Optimality of Score-based Diffusion Models: Beyond the Density Lower Bound Assumptionscited by 47
- 2024Graph Diffusion Transformers for Multi-Conditional Molecular Generationcited by 45
- 2024Efficient Video Diffusion Models via Content-Frame Motion-Latent Decompositioncited by 42
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matchingcited by 41
- 20233D molecule generation by denoising voxel gridscited by 40
- 2024Equivariant Graph Neural Operator for Modeling 3D Dynamicscited by 38
- 2023UGG: Unified Generative Graspingcited by 30
38 more
- 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuningcited by 30
- 2023UniGS: Unified Representation for Image Generation and Segmentationcited by 29
- 2024A Survey of Generative AI for de novo Drug Design: New Frontiers in Molecule and Protein Generationcited by 29
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2025Scenario Dreamer: Vectorized Latent Diffusion for Generating Driving Simulation Environmentscited by 28
- 2024SLEDGE: Synthesizing Driving Environments with Generative Models and Rule-Based Trafficcited by 27
- 2024Diffusion Models in textitDe Novo Drug Designcited by 26
- 2024Unifying Generation and Prediction on Graphs with Latent Graph Diffusioncited by 25
- 2024Training Free Guided Flow Matching with Optimal Controlcited by 25
- 2023MOFDiff: Coarse-grained Diffusion for Metal-Organic Framework Designcited by 24
- 2024UniGEM: A Unified Approach to Generation and Property Prediction for Moleculescited by 21
- 2024Hyperbolic Geometric Latent Diffusion Model for Graph Generationcited by 20
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structurescited by 20
- 2024Structure-based drug design by denoising voxel gridscited by 19
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 19
- 2024Unified Generative Modeling of 3D Molecules via Bayesian Flow Networkscited by 18
- 2024Are High-Degree Representations Really Unnecessary in Equivariant Graph Neural Networks?cited by 18
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationcited by 17
- 2025A Periodic Bayesian Flow for Material Generationcited by 16
- 2024Geometric Trajectory Diffusion Modelscited by 15
- 2025Multi-domain Distribution Learning for De Novo Drug Designcited by 15
- 2024Geometric Representation Condition Improves Equivariant Molecule Generationcited by 13
- 2025Electron flow matching for generative reaction mechanism prediction obeying conservation lawscited 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
- 2024Latent Diffusion for Neural Spiking Datacited by 12
- 2025TABASCO: A Fast, Simplified Model for Molecular Generation with Improved Physical Qualitycited by 12
- 2024Accelerating 3D Molecule Generation via Jointly Geometric Optimal Transportcited by 11
- 2024LDMol: A Text-to-Molecule Diffusion Model with Structurally Informative Latent Space Surpasses AR Modelscited by 11
- 2024Context-Guided Diffusion for Out-of-Distribution Molecular and Protein Designcited by 11
- 2025DIFFVSGG: Diffusion-Driven Online Video Scene Graph Generationcited by 11
- 2023Discrete Graph Auto-Encodercited by 10
- 2024Geometry Informed Tokenization of Molecules for Language Model Generationcited by 10
- 2025Platonic Transformers: A Solid Choice For Equivariancecited by 10
- 2023Towards Joint Sequence-Structure Generation of Nucleic Acid and Protein Complexes with SE(3)-Discrete Diffusioncited by 9
- 2024Molecular relaxation by reverse diffusion with time step predictioncited by 9
- 2025Unified Guidance for Geometry-Conditioned Molecular Generationcited by 9
50 most cited of 186 All 186 (JSON)
- 2026OctMole: Multiscale 3D Molecular Generation with Hierarchical Spatial Flows12 Oct 2026
- 2026GLASS: Global Latent Aggregation with Slot-based Set Decoding for Scalable All-Atom Crystal Generation29 Sep 2026
- 2026ReGDiff: Guided Diffusion in Regulated Latent Space for Exploring Metamaterial Voxel Geometry28 Sep 2026
- 2026TreeRef-BFN: Equivariance-Free De Novo Molecule Generation based on 2D Topology and Internal 3D Geometry26 Sep 2026
- 2026PocketVE: Stable and Property-Guided Structure-Based Drug Design with Variance-Exploding Diffusion8 Sep 2026
- 2026SwiftExplorer: Training-free Diffusion Model Alignment with Swift Diversity Exploration6 Sep 2026
- 2026SinAE: A Single-Architecture Flow-Matching Autoencoder for Cross-Domain Atomic Systems14 Jul 2026
- 2026Autoregressive latent diffusion for 3D molecule generation10 Jul 2026
- 2026SynLaD: Latent Diffusion for Generating Synthesizable Molecules Conditioned on 3D Pharmacophore Profiles1 Jul 2026
- 2026MolSafeEval: A Benchmark for Uncovering Safety Risks in AI-Generated Molecules1 Jul 2026
- 2026Uncertainty Estimation for Molecular Diffusion Models11 Jun 2026
- 2026Uncertainty-Calibrated Diffusion for Reliable 3D Molecular Graph Generation1 Jun 2026
38 more
- 2026Genotype-Conditioned Molecular Generation via Evidence-Grounded Multi-Objective Latent Perturbation in Diffusion Models31 May 2026
- 2026Efficient Synthetic Network Generation via Latent Embedding Reconstruction31 May 2026
- 2026The Score Hamiltonian: Mapping Diffusion Models to Adiabatic Transport28 May 2026
- 2026Domain-Gated Latent Diffusion: Generative Inverse Design of HMX-Class Energetic Materials with First-Principles Validation26 May 2026
- 2026SciCore-Mol: Augmenting Large Language Models with Pluggable Molecular Cognition Modules21 May 2026
- 2026Generative Pseudo-Force Fields for Molecular Generation18 May 2026
- 2026Generating Physically Consistent Molecules with Energy-Based Models18 May 2026
- 2026Composable Crystals: Controllable Materials Discovery via Concept Learning14 May 2026
- 2026Crys-JEPA: Accelerating Crystal Discovery via Embedding Screening and Generative Refinement14 May 2026
- 2026From Single-Step Edit Response to Multi-Step Molecular Optimization11 May 2026
- 2026MolWorld: Molecule World Models for Actionable Molecular Optimization9 May 2026
- 2026Toward Better Geometric Representations for Molecule Generative Models8 May 2026
- 2026FlashMol: High-Quality Molecule Generation in as Few as Four Steps7 May 2026
- 2026From Knowledge to Action: Outcomes of the 2025 Large Language Model (LLM) Hackathon for Applications in Materials Science and Chemistry4 May 2026
- 2026Quotient-Space Diffusion Models23 Apr 2026
- 2026Contraction and Hourglass Persistence for Learning on Graphs, Simplices, and Cells19 Apr 2026
- 2026End-to-End Training for Unified Tokenization and Latent Denoising23 Mar 2026
- 2026VecMol: Vector-Field Representations for 3D Molecule Generation13 Mar 2026
- 2026Equivariant Asynchronous Diffusion: An Adaptive Denoising Schedule for Accelerated Molecular Conformation Generation10 Mar 2026
- 2026Latent Diffusion-Based 3D Molecular Recovery from Vibrational Spectra6 Mar 2026
- 2026Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials24 Feb 2026
- 2026Drift Estimation for Stochastic Differential Equations with Denoising Diffusion Models19 Feb 2026
- 2026MolHIT: Advancing Molecular-Graph Generation with Hierarchical Discrete Diffusion Models19 Feb 2026
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation16 Feb 2026
- 2026All-Atom GPCR-Ligand Dynamics Simulation via a Residual Latent Flow Mode3 Feb 2026
- 2026RMFlow: Refined Mean Flow by a Noise-Injection Step for Multimodal Generation31 Jan 2026
- 2026Elign: Equivariant Diffusion Model Alignment from Foundational Machine Learning Force Fields29 Jan 2026
- 2026CORDS: Continuous Representations of Discrete Structures29 Jan 2026
- 20263D Molecule Generation from Rigid Motifs via SE(3) Flows23 Jan 2026
- 2025CoPHo: Classifier-guided Conditional Topology Generation with Persistent Homology17 Dec 2025
- 2025MolGuidance: Advanced Guidance Strategies for Conditional Molecular Generation with Flow Matching13 Dec 2025
- 2025MolSculpt: Sculpting 3D Molecular Geometries from Chemical Syntax9 Dec 2025
- 2025Latent Nonlinear Denoising Score Matching for Enhanced Learning of Structured Distributions7 Dec 2025
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025Atomic Diffusion Models for Small Molecule Structure Elucidation from NMR Spectra2 Dec 2025
- 2025Mofasa: A Step Change in Metal-Organic Framework Generation1 Dec 2025
- 2025FlexiFlow: decomposable flow matching for generation of flexible molecular ensemble21 Nov 2025
- 2025GeoSceneGraph: Geometric Scene Graph Diffusion Model for Text-guided 3D Indoor Scene Synthesis18 Nov 2025
50 newest of 186 All 186 (JSON)
- 2024Derivative-Free Guidance in Continuous and Discrete Diffusion Models with Soft Value-Based Decodingcited by 136
- 2024TFG: Unified Training-Free Guidance for Diffusion Modelscited by 107
- 2024Discrete-state Continuous-time Diffusion for Graph Generationcited by 76
- 2025All-atom Diffusion Transformers: Unified generative modelling of molecules and materialscited by 75
- 2023Equivariant Flow Matching with Hybrid Probability Transportcited by 60
- 2024Minimax Optimality of Score-based Diffusion Models: Beyond the Density Lower Bound Assumptionscited by 47
- 2024Graph Diffusion Transformers for Multi-Conditional Molecular Generationcited by 45
- 2024Efficient Video Diffusion Models via Content-Frame Motion-Latent Decompositioncited by 42
- 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuningcited by 30
- 2023UniGS: Unified Representation for Image Generation and Segmentationcited by 29
- 2025Scenario Dreamer: Vectorized Latent Diffusion for Generating Driving Simulation Environmentscited by 28
- 2024Diffusion Models in textitDe Novo Drug Designcited by 26
12 most cited of 91 All 91 (JSON)
- 2026ReGDiff: Guided Diffusion in Regulated Latent Space for Exploring Metamaterial Voxel Geometry28 Sep 2026
- 2026SwiftExplorer: Training-free Diffusion Model Alignment with Swift Diversity Exploration6 Sep 2026
- 2026SinAE: A Single-Architecture Flow-Matching Autoencoder for Cross-Domain Atomic Systems14 Jul 2026
- 2026Autoregressive latent diffusion for 3D molecule generation10 Jul 2026
- 2026SynLaD: Latent Diffusion for Generating Synthesizable Molecules Conditioned on 3D Pharmacophore Profiles1 Jul 2026
- 2026Uncertainty Estimation for Molecular Diffusion Models11 Jun 2026
- 2026Genotype-Conditioned Molecular Generation via Evidence-Grounded Multi-Objective Latent Perturbation in Diffusion Models31 May 2026
- 2026The Score Hamiltonian: Mapping Diffusion Models to Adiabatic Transport28 May 2026
- 2026Domain-Gated Latent Diffusion: Generative Inverse Design of HMX-Class Energetic Materials with First-Principles Validation26 May 2026
- 2026Generative Pseudo-Force Fields for Molecular Generation18 May 2026
- 2026Quotient-Space Diffusion Models23 Apr 2026
- 2026End-to-End Training for Unified Tokenization and Latent Denoising23 Mar 2026
12 newest of 91 All 91 (JSON)
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matchingcited by 41
- 20233D molecule generation by denoising voxel gridscited by 40
- 2024Equivariant Graph Neural Operator for Modeling 3D Dynamicscited by 38
- 2023UGG: Unified Generative Graspingcited by 30
- 2024A Survey of Generative AI for de novo Drug Design: New Frontiers in Molecule and Protein Generationcited by 29
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2024SLEDGE: Synthesizing Driving Environments with Generative Models and Rule-Based Trafficcited by 27
- 2023MOFDiff: Coarse-grained Diffusion for Metal-Organic Framework Designcited by 24
- 2024UniGEM: A Unified Approach to Generation and Property Prediction for Moleculescited by 21
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structurescited by 20
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 19
- 2024Are High-Degree Representations Really Unnecessary in Equivariant Graph Neural Networks?cited by 18
12 most cited of 95 All 95 (JSON)
- 2026OctMole: Multiscale 3D Molecular Generation with Hierarchical Spatial Flows12 Oct 2026
- 2026GLASS: Global Latent Aggregation with Slot-based Set Decoding for Scalable All-Atom Crystal Generation29 Sep 2026
- 2026TreeRef-BFN: Equivariance-Free De Novo Molecule Generation based on 2D Topology and Internal 3D Geometry26 Sep 2026
- 2026PocketVE: Stable and Property-Guided Structure-Based Drug Design with Variance-Exploding Diffusion8 Sep 2026
- 2026MolSafeEval: A Benchmark for Uncovering Safety Risks in AI-Generated Molecules1 Jul 2026
- 2026Uncertainty-Calibrated Diffusion for Reliable 3D Molecular Graph Generation1 Jun 2026
- 2026Efficient Synthetic Network Generation via Latent Embedding Reconstruction31 May 2026
- 2026SciCore-Mol: Augmenting Large Language Models with Pluggable Molecular Cognition Modules21 May 2026
- 2026Generating Physically Consistent Molecules with Energy-Based Models18 May 2026
- 2026Composable Crystals: Controllable Materials Discovery via Concept Learning14 May 2026
- 2026Crys-JEPA: Accelerating Crystal Discovery via Embedding Screening and Generative Refinement14 May 2026
- 2026From Single-Step Edit Response to Multi-Step Molecular Optimization11 May 2026
12 newest of 95 All 95 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2023Geometry-Complete Diffusion for 3D Molecule Generation and Optimization
- 2022MDM: Molecular Diffusion Model for 3D Molecule Generation
- 2025Towards Unified and Lossless Latent Space for 3D Molecular Latent Diffusion Modeling
- 2023Navigating the Design Space of Equivariant Diffusion-Based Generative Models for De Novo 3D Molecule Generation
- 2025D3MES: Diffusion Transformer with multihead equivariant self-attention for 3D molecule generation
- 2024Accelerating the Generation of Molecular Conformations with Progressive Distillation of Equivariant Latent Diffusion Models
- 2022DiffMD: A Geometric Diffusion Model for Molecular Dynamics Simulations
- 2025Exploring Molecule Generation Using Latent Space Graph Diffusion