Distribution Matching Distillation Meets Reinforcement Learning
On the map: Diffusion models → Diffusion and flow policies for robotics
Abstract (arXiv)
Distribution Matching Distillation (DMD) facilitates efficient inference by distilling multi-step diffusion models into few-step variants. Concurrently, Reinforcement Learning (RL) has emerged as a vital tool for aligning generative models with human preferences. While both represent critical post-training stages for large-scale diffusion models, existing studies typically treat them as independent, sequential processes, leaving a systematic framework for their unification largely unexplored. In this work, we demonstrate that jointly optimizing these two objectives yields mutual benefits: RL enables more preference-aware and controllable distillation rather than uniformly compressing the full data distribution, while DMD serves as an effective regularizer to mitigate reward hacking during RL training. Building on these insights, we propose DMDR, a unified framework that incorporates Reward-Tilted Distribution Matching optimization alongside two dynamic distillation training strategies in the initial stage, followed by the joint DMD and RL optimization in the second stage. Extensive experiments demonstrate that DMDR achieves state-of-the-art visual quality and prompt adherence among few-step generation methods, even surpassing the performance of its multi-step teacher model.
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 79 of its 91 references that are in the map. First posted on arXiv 17 Nov 2025.
Builds on
79 of 91 references.
Closest ancestors · 9
Same direction, newest first
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Process19 Sep
- 2025Directly Aligning the Full Diffusion Trajectory with Fine-Grained Human Preference8 Sep
- 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
- 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
- 2024Diff-Instruct++: Training One-step Text-to-image Generator Model to Align with Human Preferences24 Oct
- 2024Deep Reward Supervisions for Tuning Text-to-Image Diffusion Models1 May
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023Training Diffusion Models with Reinforcement Learning22 May
- 2023Human Preference Score: Better Aligning Text-to-Image Models with Human Preference25 Mar
Other references · 70
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
38 more
- 2022DreamFusion: Text-to-3D using 2D Diffusioncited by 3,006
- 2016Improved Techniques for Training GANscited by 2,938
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,388
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
- 2023ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillationcited by 1,207
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
- 2023One-step Diffusion with Distribution Matching Distillationcited by 978
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
- 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 925
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
- 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inferencecited by 826
- 2023Adversarial Diffusion Distillationcited by 741
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 707
- 2018Which Training Methods for GANs do actually Converge?cited by 672
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 666
- 2024Improved Distribution Matching Distillation for Fast Image Synthesiscited by 648
- 2023Q-Align: Teaching LMMs for Visual Scoring via Discrete Text-Defined Levelscited by 489
- 2017Towards Principled Methods for Training Generative Adversarial Networkscited by 460
- 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 455
- 2024One Step Diffusion via Shortcut Modelscited by 373
- 2021Knowledge Distillation in Iterative Generative Models for Improved Sampling Speedcited by 344
- 2024Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillationcited by 276
- 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Modelscited by 276
- 2025Improving Video Generation with Human Feedbackcited by 267
- 2024SDXL-Lightning: Progressive Adversarial Diffusion Distillationcited by 258
- 2023Data Filtering Networkscited by 247
- 2025Seedream 4.0: Toward Next-generation Multimodal Image Generationcited by 233
- 2023Diff-Instruct: A Universal Approach for Transferring Knowledge From Pre-trained Diffusion Modelscited by 232
- 2025HPSv3: Towards Wide-Spectrum Human Preference Scorecited by 196
- 2025Unified Reward Model for Multimodal Understanding and Generationcited by 194
- 2024Hyper-SD: Trajectory Segmented Consistency Model for Efficient Image Synthesiscited by 153
- 2024VisionReward: Fine-Grained Multi-Dimensional Human Preference Learning for Image and Video Generationcited by 148
- 2025Diffusion Adversarial Post-Training for One-Step Video Generationcited by 142
- 2017Stabilizing Training of Generative Adversarial Networks through Regularizationcited by 118
- 2025ShareGPT-4o-Image: Aligning Multimodal Models with GPT-4o-Level Image Generationcited by 94
- 2025Lumina-Image 2.0: A Unified and Efficient Image Generative Frameworkcited by 93
- 2025SANA-Sprint: One-Step Diffusion with Continuous-Time Consistency Distillationcited by 85
50 most cited of 70 All 70 (JSON)
Built on by
Citing papers per year: 2025: 2; 2026: 41.
43 papers.
Sort:
By area:
- 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
- 2026Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Modelscited by 23
- 2025TwinFlow: Realizing One-step Generation on Large Models with Self-adversarial Flowscited by 17
- 2026D-OPSD: On-Policy Self-Distillation for Continuously Tuning Step-Distilled Diffusion Modelscited by 17
- 2026Manifold-Aware Exploration for Reinforcement Learning in Video Generationcited by 14
- 2026Hallo-Live: Real-Time Streaming Joint Audio-Video Avatar Generation with Asynchronous Dual-Stream and Human-Centric Preference Distillationcited by 13
- 2026DNA: Uncovering Universal Latent Forgery Knowledgecited by 10
- 2026Reinforcing Few-step Generators via Reward-Tilted Distribution Matchingcited by 6
- 2026ERNIE-Image Technical Reportcited by 5
- 2026AdaTSQ: Pushing the Pareto Frontier of Diffusion Transformers via Temporal-Sensitivity Quantizationcited by 4
- 2026Continuous-Time Distribution Matching for Few-Step Diffusion Distillationcited by 4
- 2026Lens: Rethinking Training Efficiency for Foundational Text-to-Image Modelscited by 4
31 more
- 2026MeanFlowNFT: Bringing Forward-Process RL to Average-Velocity Generatorscited by 4
- 2026Mamoda2.5: Enhancing Unified Multimodal Model with DiT-MoEcited by 3
- 2026CollectionLoRA: Collecting 50 Effects in 1 LoRA via Multi-Teacher On-Policy Distillationcited by 3
- 2026CoDMD: Copula-aware Distribution Matching Distillation for Fast Video Generationcited by 3
- 2026An Empirical Study of Training Pixel-Space Text-to-Image Diffusion Modelscited by 3
- 2026Duality Models: An Embarrassingly Simple One-step Generation Paradigmcited by 2
- 2026HybridStitch: Pixel and Timestep Level Model Stitching for Diffusion Accelerationcited by 2
- 2026Guiding Distribution Matching Distillation with Gradient-Based Reinforcement Learningcited by 2
- 2026Stream-R1: Reliability-Perplexity Aware Reward Distillation for Streaming Video Generationcited by 2
- 2026High-Fidelity Two-Step Image Generation via Teacher-Aligned End-to-End Distillationcited by 2
- 2026Cross-Resolution Distribution Matching for Diffusion Distillationcited by 1
- 2026Advances in GRPO for Generation Models: A Surveycited by 1
- 2026Rdm: Re-conceptualizing Distribution Matching as a Reward for Diffusion Distillationcited by 1
- 2026AdvDMD: Adversarial Reward Meets DMD For High-Quality Few-Step Generationcited by 1
- 2026FlashMol: High-Quality Molecule Generation in as Few as Four Stepscited by 1
- 2026Delta Forcing: Trust Region Steering for Interactive Autoregressive Video Generationcited by 1
- 2026Efficient Image Synthesis with Sphere Latent Encodercited by 1
- 2026Monocular Avatar Reconstruction via Cascaded Diffusion Priors and UV-Space Differentiable Shadingcited by 1
- 2026AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatarscited by 1
- 2026Joint Alignment and Distillation for Video Generation via Sample-Guided Distribution Matchingcited by 1
- 2026Qwen-Image-Flash: Rethinking the Training Recipe for Few-Step Distillationcited by 0
- 2026World Model Self-Distillation: Training World Models to Solve General Taskscited by 0
- 2026A Test-time Actor-Critic Approach to News Images Generationcited by 0
- 2026Reward Lightning: Fast Video Generation via Homologous Preference Distillationcited by 0
- 2026ScaleResfusion: Residual Rectified Flow based on Residual Vector Fieldcited by 0
- 2026RL-Native Distillation: Exploiting Scored Trajectories for Few-Step Image Generationcited by 0
- 2026DUET: A Diversity-Quality Duet of Distillation Experts for Two-Step Video Generationcited by 0
- 2026DyMD: Preserving Interaction Dynamics through Distribution Matching Distillation in Few-Step Video World Modelscited by 0
- 2026PDMD: Projected Distribution Matching Distillation for Video Diffusion Modelscited by 0
- 2026PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillationcited by 0
- 2026DMA2: Pixel-space Distribution Matching with Adversarial and Anchor Lossescited by 0
- 2026DMA2: Pixel-space Distribution Matching with Adversarial and Anchor Losses29 Sep 2026
- 2026PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation29 Sep 2026
- 2026PDMD: Projected Distribution Matching Distillation for Video Diffusion Models28 Sep 2026
- 2026DyMD: Preserving Interaction Dynamics through Distribution Matching Distillation in Few-Step Video World Models25 Sep 2026
- 2026Joint Alignment and Distillation for Video Generation via Sample-Guided Distribution Matching3 Sep 2026
- 2026An Empirical Study of Training Pixel-Space Text-to-Image Diffusion Models17 Aug 2026
- 2026DUET: A Diversity-Quality Duet of Distillation Experts for Two-Step Video Generation10 Aug 2026
- 2026RL-Native Distillation: Exploiting Scored Trajectories for Few-Step Image Generation10 Aug 2026
- 2026ScaleResfusion: Residual Rectified Flow based on Residual Vector Field28 Jul 2026
- 2026AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatars27 Jul 2026
- 2026MeanFlowNFT: Bringing Forward-Process RL to Average-Velocity Generators16 Jul 2026
- 2026Reward Lightning: Fast Video Generation via Homologous Preference Distillation4 Jul 2026
31 more
- 2026Monocular Avatar Reconstruction via Cascaded Diffusion Priors and UV-Space Differentiable Shading26 Jun 2026
- 2026Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Models24 Jun 2026
- 2026CoDMD: Copula-aware Distribution Matching Distillation for Fast Video Generation20 Jun 2026
- 2026A Test-time Actor-Critic Approach to News Images Generation19 Jun 2026
- 2026High-Fidelity Two-Step Image Generation via Teacher-Aligned End-to-End Distillation10 Jun 2026
- 2026World Model Self-Distillation: Training World Models to Solve General Tasks10 Jun 2026
- 2026Qwen-Image-Flash: Rethinking the Training Recipe for Few-Step Distillation2 Jun 2026
- 2026Reinforcing Few-step Generators via Reward-Tilted Distribution Matching25 May 2026
- 2026CollectionLoRA: Collecting 50 Effects in 1 LoRA via Multi-Teacher On-Policy Distillation25 May 2026
- 2026ERNIE-Image Technical Report25 May 2026
- 2026Lens: Rethinking Training Efficiency for Foundational Text-to-Image Models20 May 2026
- 2026Efficient Image Synthesis with Sphere Latent Encoder15 May 2026
- 2026Delta Forcing: Trust Region Steering for Interactive Autoregressive Video Generation14 May 2026
- 2026FlashMol: High-Quality Molecule Generation in as Few as Four Steps7 May 2026
- 2026Continuous-Time Distribution Matching for Few-Step Diffusion Distillation7 May 2026
- 2026D-OPSD: On-Policy Self-Distillation for Continuously Tuning Step-Distilled Diffusion Models6 May 2026
- 2026Stream-R1: Reliability-Perplexity Aware Reward Distillation for Streaming Video Generation5 May 2026
- 2026Mamoda2.5: Enhancing Unified Multimodal Model with DiT-MoE4 May 2026
- 2026AdvDMD: Adversarial Reward Meets DMD For High-Quality Few-Step Generation29 Apr 2026
- 2026Hallo-Live: Real-Time Streaming Joint Audio-Video Avatar Generation with Asynchronous Dual-Stream and Human-Centric Preference Distillation26 Apr 2026
- 2026Guiding Distribution Matching Distillation with Gradient-Based Reinforcement Learning21 Apr 2026
- 2026Rdm: Re-conceptualizing Distribution Matching as a Reward for Diffusion Distillation30 Mar 2026
- 2026Manifold-Aware Exploration for Reinforcement Learning in Video Generation23 Mar 2026
- 2026HybridStitch: Pixel and Timestep Level Model Stitching for Diffusion Acceleration8 Mar 2026
- 2026Cross-Resolution Distribution Matching for Diffusion Distillation6 Mar 2026
- 2026Advances in GRPO for Generation Models: A Survey21 Feb 2026
- 2026AdaTSQ: Pushing the Pareto Frontier of Diffusion Transformers via Temporal-Sensitivity Quantization10 Feb 2026
- 2026Duality Models: An Embarrassingly Simple One-step Generation Paradigm4 Feb 2026
- 2026DNA: Uncovering Universal Latent Forgery Knowledge30 Jan 2026
- 2025TwinFlow: Realizing One-step Generation on Large Models with Self-adversarial Flows3 Dec 2025
- 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformer27 Nov 2025
- 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
- 2026D-OPSD: On-Policy Self-Distillation for Continuously Tuning Step-Distilled Diffusion Modelscited by 17
- 2026Hallo-Live: Real-Time Streaming Joint Audio-Video Avatar Generation with Asynchronous Dual-Stream and Human-Centric Preference Distillationcited by 13
- 2026Reinforcing Few-step Generators via Reward-Tilted Distribution Matchingcited by 6
- 2026AdaTSQ: Pushing the Pareto Frontier of Diffusion Transformers via Temporal-Sensitivity Quantizationcited by 4
- 2026Continuous-Time Distribution Matching for Few-Step Diffusion Distillationcited by 4
- 2026CollectionLoRA: Collecting 50 Effects in 1 LoRA via Multi-Teacher On-Policy Distillationcited by 3
- 2026CoDMD: Copula-aware Distribution Matching Distillation for Fast Video Generationcited by 3
- 2026An Empirical Study of Training Pixel-Space Text-to-Image Diffusion Modelscited by 3
- 2026HybridStitch: Pixel and Timestep Level Model Stitching for Diffusion Accelerationcited by 2
- 2026Guiding Distribution Matching Distillation with Gradient-Based Reinforcement Learningcited by 2
- 2026Stream-R1: Reliability-Perplexity Aware Reward Distillation for Streaming Video Generationcited by 2
12 most cited of 23 All 23 (JSON)
- 2026DMA2: Pixel-space Distribution Matching with Adversarial and Anchor Losses29 Sep 2026
- 2026PE-OPSD: Internalizing Prompt Enhancement into Flow-matching Models via On-Policy Self-Distillation29 Sep 2026
- 2026PDMD: Projected Distribution Matching Distillation for Video Diffusion Models28 Sep 2026
- 2026An Empirical Study of Training Pixel-Space Text-to-Image Diffusion Models17 Aug 2026
- 2026DUET: A Diversity-Quality Duet of Distillation Experts for Two-Step Video Generation10 Aug 2026
- 2026ScaleResfusion: Residual Rectified Flow based on Residual Vector Field28 Jul 2026
- 2026Reward Lightning: Fast Video Generation via Homologous Preference Distillation4 Jul 2026
- 2026Monocular Avatar Reconstruction via Cascaded Diffusion Priors and UV-Space Differentiable Shading26 Jun 2026
- 2026CoDMD: Copula-aware Distribution Matching Distillation for Fast Video Generation20 Jun 2026
- 2026High-Fidelity Two-Step Image Generation via Teacher-Aligned End-to-End Distillation10 Jun 2026
- 2026Reinforcing Few-step Generators via Reward-Tilted Distribution Matching25 May 2026
- 2026CollectionLoRA: Collecting 50 Effects in 1 LoRA via Multi-Teacher On-Policy Distillation25 May 2026
12 newest of 23 All 23 (JSON)
- 2026Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Modelscited by 23
- 2026Delta Forcing: Trust Region Steering for Interactive Autoregressive Video Generationcited by 1
- 2026World Model Self-Distillation: Training World Models to Solve General Taskscited by 0
- 2026DyMD: Preserving Interaction Dynamics through Distribution Matching Distillation in Few-Step Video World Modelscited by 0
- 2026DyMD: Preserving Interaction Dynamics through Distribution Matching Distillation in Few-Step Video World Models25 Sep 2026
- 2026Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Models24 Jun 2026
- 2026World Model Self-Distillation: Training World Models to Solve General Tasks10 Jun 2026
- 2026Delta Forcing: Trust Region Steering for Interactive Autoregressive Video Generation14 May 2026
- 2025TwinFlow: Realizing One-step Generation on Large Models with Self-adversarial Flowscited by 17
- 2026DNA: Uncovering Universal Latent Forgery Knowledgecited by 10
- 2026ERNIE-Image Technical Reportcited by 5
- 2026Lens: Rethinking Training Efficiency for Foundational Text-to-Image Modelscited by 4
- 2026MeanFlowNFT: Bringing Forward-Process RL to Average-Velocity Generatorscited by 4
- 2026Duality Models: An Embarrassingly Simple One-step Generation Paradigmcited by 2
- 2026FlashMol: High-Quality Molecule Generation in as Few as Four Stepscited by 1
- 2026Efficient Image Synthesis with Sphere Latent Encodercited by 1
- 2026AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatarscited by 1
- 2026Qwen-Image-Flash: Rethinking the Training Recipe for Few-Step Distillationcited by 0
- 2026A Test-time Actor-Critic Approach to News Images Generationcited by 0
- 2026RL-Native Distillation: Exploiting Scored Trajectories for Few-Step Image Generationcited by 0
- 2026RL-Native Distillation: Exploiting Scored Trajectories for Few-Step Image Generation10 Aug 2026
- 2026AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatars27 Jul 2026
- 2026MeanFlowNFT: Bringing Forward-Process RL to Average-Velocity Generators16 Jul 2026
- 2026A Test-time Actor-Critic Approach to News Images Generation19 Jun 2026
- 2026Qwen-Image-Flash: Rethinking the Training Recipe for Few-Step Distillation2 Jun 2026
- 2026ERNIE-Image Technical Report25 May 2026
- 2026Lens: Rethinking Training Efficiency for Foundational Text-to-Image Models20 May 2026
- 2026Efficient Image Synthesis with Sphere Latent Encoder15 May 2026
- 2026FlashMol: High-Quality Molecule Generation in as Few as Four Steps7 May 2026
- 2026Duality Models: An Embarrassingly Simple One-step Generation Paradigm4 Feb 2026
- 2026DNA: Uncovering Universal Latent Forgery Knowledge30 Jan 2026
- 2025TwinFlow: Realizing One-step Generation on Large Models with Self-adversarial Flows3 Dec 2025
Similar, not cited
Close in wording, no citation either way
- 2025ReDiF: Reinforced Distillation for Few Step Diffusion
- 2026ParetoSlider: Diffusion Models Post-Training for Continuous Reward Control
- 2026On-Policy Self-Distillation in Diffusion Models
- 2026Diffusion Fine-tuning with Rewarded Moment Matching Distillation
- 2025Data-regularized Reinforcement Learning for Diffusion Models at Scale
- 2026Scaling Reinforcement Learning for Diffusion Models via Velocity Matching
- 2025Beyond Reward Margin: Rethinking and Resolving Likelihood Displacement in Diffusion Models via Video Generation
- 2026Consistent Distribution Matching for Data-Free Diffusion Distillation