Continuous-Time Distribution Matching for Few-Step Diffusion Distillation
Summary
This paper introduces Continuous-Time Distribution Matching (CDM), a method for few-step diffusion distillation that migrates from discrete to continuous optimization to improve visual fidelity and preserve fine details.
View Cached Full Text
Cached at: 05/08/26, 07:03 AM
Paper page - Continuous-Time Distribution Matching for Few-Step Diffusion Distillation
Source: https://huggingface.co/papers/2605.06376 Published on May 7
·
Submitted byhttps://huggingface.co/byliutao
liutaoon May 8
Authors:
,
,
,
,
,
,
,
,
,
Abstract
Continuous-Time Distribution Matching migrates diffusion model distillation from discrete to continuous optimization, enabling arbitrary points along sampling trajectories and preserving fine visual details through dynamic scheduling and velocity field extrapolation.
Stepdistillationhas become a leading technique for acceleratingdiffusion models, among whichDistribution MatchingDistillation(DMD) andConsistency Distillationare two representative paradigms. While consistency methods enforce self-consistency along the full PF-ODE trajectory to steer it toward the clean data manifold, vanilla DMD relies on sparse supervision at a few predefined discrete timesteps. This restricted discrete-time formulation and mode-seeking nature of thereverse KL divergencetends to exhibit visual artifacts and over-smoothed outputs, often necessitating complex auxiliary modules -- such as GANs or reward models -- to restore visual fidelity. In this work, we introduceContinuous-TimeDistribution Matching(CDM), migrating the DMD framework from discrete anchoring to continuous optimization for the first time. CDM achieves this through twocontinuous-timedesigns. First, we replace the fixed discrete schedule with a dynamic continuous schedule of random length, so thatdistribution matchingis enforced at arbitrary points along sampling trajectories rather than only at a few fixed anchors. Second, we propose acontinuous-timealignment objective that performs active off-trajectory matching on latents extrapolated via the student’svelocity field, improving generalization and preserving fine visual details. Extensive experiments on different architectures, including SD3-Medium and Longcat-Image, demonstrate that CDM provides highly competitive visual fidelity for few-stepimage generationwithout relying on complex auxiliary objectives. Code is available at https://github.com/byliutao/cdm.
View arXiv pageView PDFProject pageGitHub1Add to collection
Get this paper in your agent:
hf papers read 2605\.06376
Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
Models citing this paper2
#### byliutao/stable-diffusion-3-medium-turbo Text-to-Image• Updated27 minutes ago • 4 • 1
#### byliutao/Longcat-Image-Turbo Updated26 minutes ago • 1
Datasets citing this paper0
No dataset linking this paper
Cite arxiv.org/abs/2605.06376 in a dataset README.md to link it from this page.
Spaces citing this paper0
No Space linking this paper
Cite arxiv.org/abs/2605.06376 in a Space README.md to link it from this page.
Collections including this paper0
No Collection including this paper
Add this paper to acollectionto link it from this page.
Similar Articles
Consistent Distribution Matching for Data-Free Diffusion Distillation
论文提出 Consistent Distribution Matching (CDMD),通过在单一学生网络内统一样本生成与速度估计,实现无需辅助 fake-score 网络的数据自由扩散蒸馏,在 ImageNet 256×256 上以 1 NFE 达到 2.04 FID,优于 DMD2 等基线,并给出基于 Wasserstein 收敛的理论分析。
D-OPSD: On-Policy Self-Distillation for Continuously Tuning Step-Distilled Diffusion Models
This paper introduces D-OPSD, a novel training paradigm for step-distilled diffusion models that enables on-policy self-distillation during supervised fine-tuning. It allows models to learn new concepts or styles without compromising their efficient few-step inference capabilities.
Multi-Bandwidth Distribution Matching Distillation: On the Equivalence of Distribution Matching Distillation and Drifting Models
本文证明了 Drifting Models 与 Distribution Matching Distillation (DMD/DMD2) 在优化目标上的等价性,并基于此视角提出了改进的 Multi-Bandwidth Distribution Matching Distillation (MBDMD) 方法,用于提升一步生成模型的性能。
Improved Distributional Diffusion Models
The paper presents Improved Distributional Diffusion Models (iDDM), which overcome scaling limits of DDMs via late particle expansion and time-dependent scoring rule schedules, achieving 4.48 FID at 4 steps and 2.38 FID at 50 steps on ImageNet-256 with DiT-XL/2 from a single-stage from-scratch training without teachers or self-distillation.
Trajectory as the Teacher: Few-Step Discrete Flow Matching via Energy-Navigated Distillation
This paper introduces Trajectory-Shaped Discrete Flow Matching (TS-DFM), which replaces blind stochastic jumps with guided navigation to significantly improve text generation efficiency and reduce computational costs. The method achieves superior perplexity and speed compared to traditional multi-step baselines while maintaining unchanged inference costs.