Trans-Dimensional Generative Modeling via Jump Diffusion Models
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
We propose a new class of generative models that naturally handle data of varying dimensionality by jointly modeling the state and dimension of each datapoint. The generative process is formulated as a jump diffusion process that makes jumps between different dimensional spaces. We first define a dimension destroying forward noising process, before deriving the dimension creating time-reversed generative process along with a novel evidence lower bound training objective for learning to approximate it. Simulating our learned approximation to the time-reversed generative process then provides an effective way of sampling data of varying dimensionality by jointly generating state values and dimensions. We demonstrate our approach on molecular and video datasets of varying dimensionality, reporting better compatibility with test-time diffusion guidance imputation tasks and improved interpolation capabilities versus fixed dimensional models that generate state values and dimensions separately.
Key facts. Cited by 33 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 27 of its 47 references that are in the map. First posted on arXiv 25 May 2023.
Builds on
27 of 47 references.
Closest ancestors · 10
Same direction, newest first
- 2022Dimensionality-Varying Diffusion Process29 Nov
- 2022From Denoising Diffusions to Denoising Markov Models7 Nov
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2022A Continuous Time Framework for Discrete Denoising Models30 May
- 2022Subspace Diffusion Generative Models3 May
- 2021A Variational Perspective on Diffusion-Based Generative Models and Score Matching5 Jun
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Maximum Likelihood Training of Score-Based Diffusion Models22 Jan
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 17
- 2017Attention Is All You Needcited by 59,932
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2022Video Diffusion Modelscited by 2,280
- 2022Imagen Video: High Definition Video Generation with Diffusion Modelscited by 1,765
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2018Model Cards for Model Reportingcited by 1,068
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 666
5 more
- 2022Flexible Diffusion Modeling of Long Videoscited by 294
- 2022Equivariant 3D-Conditional Diffusion Models for Molecular Linker Designcited by 121
- 2021Top-N: Equivariant set and graph generation without exchangeabilitycited by 36
- 2023A Control-Centric Benchmark for Video Predictioncited by 25
- 2023Towards Controllable Diffusion Models via Reward-Guided Explorationcited by 8
Built on by
Citing papers per year: 2023: 2; 2024: 2; 2025: 17; 2026: 12.
33 papers.
Sort:
By area:
- 2024Pyramidal Flow Matching for Efficient Video Generative Modelingcited by 354
- 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 91
- 2023On the Design Fundamentals of Diffusion Models: A Surveycited by 59
- 2025Edit Flows: Flow Matching with Edit Operationscited by 53
- 2025Any-Order Flexible Length Masked Diffusioncited by 48
- 2023Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Designcited by 29
- 2025OneFlow: Concurrent Mixed-Modal and Interleaved Generation with Edit Flowscited by 17
- 2026DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvascited by 16
- 2025Multi-domain Distribution Learning for De Novo Drug Designcited by 15
- 2025Insertion Language Models: Sequence Generation with Arbitrary-Position Insertionscited by 11
- 2025Scalable Autoregressive 3D Molecule Generationcited by 7
- 2025Compositional Flows for 3D Molecule and Synthesis Pathway Co-designcited by 6
21 more
- 2025A Diffusion Model to Shrink Proteins While Maintaining Their Functioncited by 6
- 2025Generative modelling with jump-diffusionscited by 4
- 2025Graph Diffusion that can Insert and Deletecited by 4
- 2025Pyramidal Patchification Flow for Visual Generationcited by 4
- 2025Branching Flows: Discrete, Continuous, and Manifold Flow Matching with Splits and Deletionscited by 4
- 2025Spatiotemporal Pyramid Flow Matching for Climate Emulationcited by 4
- 2026PyramidalWan: On Making Pretrained Video Model Pyramidal for Efficient Inferencecited by 3
- 2024Equivariant Blurring Diffusion for Hierarchical Molecular Conformer Generationcited by 2
- 2025A Jump-Diffusion Framework for Irregular Time Series Generationcited by 2
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generationcited by 2
- 2026Generative Molecular Morphing for Flexible-Size Design via Unbalanced Optimal Transportcited by 2
- 2025Discrete Bayesian Sample Inference for Graph Generationcited by 0
- 2026Beyond Two-stage Diffusion TTS: Joint Structure and Content Refinement via Jump Diffusioncited by 0
- 2026Unified Number-Free Text-to-Motion Generation Via Flow Matchingcited by 0
- 2026A Continuous-Time Markov Chain Framework for Insertion Language Modelscited by 0
- 2026Variable-Length Generative Protein Design via Generalized Poisson Flowcited by 0
- 2026Sticky Jump Diffusions: A Unifying View of Masked, Continuous, and Hybrid Diffusioncited by 0
- 2026Existence-Field Diffusion Model for Spatial Point Processes with Variable Cardinalitycited by 0
- 2026HorizonFlow: Variable-Length Planning for Offline Goal-Conditioned RLcited by 0
- 2026RetroGEF: Dynamic Graph Edit Flow for Single-Step Retrosynthesiscited by 0
- 2026Grand Canonical Generatorscited by 0
- 2026Grand Canonical Generators30 Sep 2026
- 2026RetroGEF: Dynamic Graph Edit Flow for Single-Step Retrosynthesis29 Sep 2026
- 2026HorizonFlow: Variable-Length Planning for Offline Goal-Conditioned RL29 Sep 2026
- 2026Existence-Field Diffusion Model for Spatial Point Processes with Variable Cardinality29 Jul 2026
- 2026Sticky Jump Diffusions: A Unifying View of Masked, Continuous, and Hybrid Diffusion12 Jul 2026
- 2026Variable-Length Generative Protein Design via Generalized Poisson Flow10 Jul 2026
- 2026A Continuous-Time Markov Chain Framework for Insertion Language Models8 Jun 2026
- 2026Generative Molecular Morphing for Flexible-Size Design via Unbalanced Optimal Transport5 Jun 2026
- 2026Unified Number-Free Text-to-Motion Generation Via Flow Matching27 Mar 2026
- 2026Beyond Two-stage Diffusion TTS: Joint Structure and Content Refinement via Jump Diffusion14 Mar 2026
- 2026DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas1 Feb 2026
- 2026PyramidalWan: On Making Pretrained Video Model Pyramidal for Efficient Inference8 Jan 2026
21 more
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025Spatiotemporal Pyramid Flow Matching for Climate Emulation1 Dec 2025
- 2025Branching Flows: Discrete, Continuous, and Manifold Flow Matching with Splits and Deletions12 Nov 2025
- 2025A Diffusion Model to Shrink Proteins While Maintaining Their Function10 Nov 2025
- 2025Discrete Bayesian Sample Inference for Graph Generation4 Nov 2025
- 2025OneFlow: Concurrent Mixed-Modal and Interleaved Generation with Edit Flows3 Oct 2025
- 2025Any-Order Flexible Length Masked Diffusion31 Aug 2025
- 2025Multi-domain Distribution Learning for De Novo Drug Design25 Aug 2025
- 2025Pyramidal Patchification Flow for Visual Generation30 Jun 2025
- 2025Edit Flows: Flow Matching with Edit Operations10 Jun 2025
- 2025Graph Diffusion that can Insert and Delete6 Jun 2025
- 2025A Jump-Diffusion Framework for Irregular Time Series Generation29 May 2025
- 2025Scalable Autoregressive 3D Molecule Generation20 May 2025
- 2025Insertion Language Models: Sequence Generation with Arbitrary-Position Insertions9 May 2025
- 2025Compositional Flows for 3D Molecule and Synthesis Pathway Co-design10 Apr 2025
- 2025PixelFlow: Pixel-Space Generative Models with Flow10 Apr 2025
- 2025Generative modelling with jump-diffusions9 Mar 2025
- 2024Equivariant Blurring Diffusion for Hierarchical Molecular Conformer Generation26 Oct 2024
- 2024Pyramidal Flow Matching for Efficient Video Generative Modeling8 Oct 2024
- 2023Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Design9 Oct 2023
- 2023On the Design Fundamentals of Diffusion Models: A Survey7 Jun 2023
- 2024Pyramidal Flow Matching for Efficient Video Generative Modelingcited by 354
- 2023On the Design Fundamentals of Diffusion Models: A Surveycited by 59
- 2025Edit Flows: Flow Matching with Edit Operationscited by 53
- 2025Any-Order Flexible Length Masked Diffusioncited by 48
- 2026DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvascited by 16
- 2025Multi-domain Distribution Learning for De Novo Drug Designcited by 15
- 2025Scalable Autoregressive 3D Molecule Generationcited by 7
- 2025Compositional Flows for 3D Molecule and Synthesis Pathway Co-designcited by 6
- 2025A Diffusion Model to Shrink Proteins While Maintaining Their Functioncited by 6
- 2025Generative modelling with jump-diffusionscited by 4
- 2025Graph Diffusion that can Insert and Deletecited by 4
- 2025Pyramidal Patchification Flow for Visual Generationcited by 4
12 most cited of 22 All 22 (JSON)
- 2026Existence-Field Diffusion Model for Spatial Point Processes with Variable Cardinality29 Jul 2026
- 2026Sticky Jump Diffusions: A Unifying View of Masked, Continuous, and Hybrid Diffusion12 Jul 2026
- 2026A Continuous-Time Markov Chain Framework for Insertion Language Models8 Jun 2026
- 2026Unified Number-Free Text-to-Motion Generation Via Flow Matching27 Mar 2026
- 2026Beyond Two-stage Diffusion TTS: Joint Structure and Content Refinement via Jump Diffusion14 Mar 2026
- 2026DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas1 Feb 2026
- 2025NEAT: Neighborhood-Guided, Efficient, Autoregressive Set Transformer for 3D Molecular Generation5 Dec 2025
- 2025Spatiotemporal Pyramid Flow Matching for Climate Emulation1 Dec 2025
- 2025Branching Flows: Discrete, Continuous, and Manifold Flow Matching with Splits and Deletions12 Nov 2025
- 2025A Diffusion Model to Shrink Proteins While Maintaining Their Function10 Nov 2025
- 2025Discrete Bayesian Sample Inference for Graph Generation4 Nov 2025
- 2025Any-Order Flexible Length Masked Diffusion31 Aug 2025
12 newest of 22 All 22 (JSON)
- 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 91
- 2023Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Designcited by 29
- 2025OneFlow: Concurrent Mixed-Modal and Interleaved Generation with Edit Flowscited by 17
- 2025Insertion Language Models: Sequence Generation with Arbitrary-Position Insertionscited by 11
- 2026PyramidalWan: On Making Pretrained Video Model Pyramidal for Efficient Inferencecited by 3
- 2025A Jump-Diffusion Framework for Irregular Time Series Generationcited by 2
- 2026Generative Molecular Morphing for Flexible-Size Design via Unbalanced Optimal Transportcited by 2
- 2026Variable-Length Generative Protein Design via Generalized Poisson Flowcited by 0
- 2026RetroGEF: Dynamic Graph Edit Flow for Single-Step Retrosynthesiscited by 0
- 2026Grand Canonical Generatorscited by 0
- 2026Grand Canonical Generators30 Sep 2026
- 2026RetroGEF: Dynamic Graph Edit Flow for Single-Step Retrosynthesis29 Sep 2026
- 2026Variable-Length Generative Protein Design via Generalized Poisson Flow10 Jul 2026
- 2026Generative Molecular Morphing for Flexible-Size Design via Unbalanced Optimal Transport5 Jun 2026
- 2026PyramidalWan: On Making Pretrained Video Model Pyramidal for Efficient Inference8 Jan 2026
- 2025OneFlow: Concurrent Mixed-Modal and Interleaved Generation with Edit Flows3 Oct 2025
- 2025A Jump-Diffusion Framework for Irregular Time Series Generation29 May 2025
- 2025Insertion Language Models: Sequence Generation with Arbitrary-Position Insertions9 May 2025
- 2025PixelFlow: Pixel-Space Generative Models with Flow10 Apr 2025
- 2023Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Design9 Oct 2023
Similar, not cited
Close in wording, no citation either way
- 2024Breaking the Curse of Dimensionality: Diffusion Models Efficiently Learn Low-Dimensional Distributions
- 2024GUD: Generation with Unified Diffusion
- 2025A solvable model of learning generative diffusion: theory and insights
- 2023Infinite-Dimensional Diffusion Models
- 2025Diffusion Models with Double Guidance: Generate with aggregated datasets
- 2022Let us Build Bridges: Understanding and Extending Diffusion Generative Models
- 2026Entropy-Based Dimension-Free Convergence and Loss-Adaptive Schedules for Diffusion Models
- 2025Geometric Regularity in Deterministic Sampling Dynamics of Diffusion-based Generative Models