Continuous-Time Distribution Matching for Few-Step Diffusion Distillation

Hugging Face Daily Papers Papers

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.

Step distillation has become a leading technique for accelerating diffusion models, among which Distribution Matching Distillation (DMD) and Consistency Distillation are 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 the reverse KL divergence tends 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 introduce Continuous-Time Distribution Matching (CDM), migrating the DMD framework from discrete anchoring to continuous optimization for the first time. CDM achieves this through two continuous-time designs. First, we replace the fixed discrete schedule with a dynamic continuous schedule of random length, so that distribution matching is enforced at arbitrary points along sampling trajectories rather than only at a few fixed anchors. Second, we propose a continuous-time alignment objective that performs active off-trajectory matching on latents extrapolated via the student's velocity 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-step image generation without relying on complex auxiliary objectives. Code is available at https://github.com/byliutao/cdm.
Original Article
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

Hugging Face Daily Papers

论文提出 Consistent Distribution Matching (CDMD),通过在单一学生网络内统一样本生成与速度估计,实现无需辅助 fake-score 网络的数据自由扩散蒸馏,在 ImageNet 256×256 上以 1 NFE 达到 2.04 FID,优于 DMD2 等基线,并给出基于 Wasserstein 收敛的理论分析。

Improved Distributional Diffusion Models

Hugging Face Daily Papers

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

Hugging Face Daily Papers

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.