Skip to content

Diffusion Distillation

Diffusion distillation compresses a multi-step pretrained diffusion / flow teacher into a few-step (1–4 NFE) student. The 2025–2026 thread on the wiki spans four orthogonal axes: (1) objective design — linear-shortcut (DMD, consistency, rectified flow) vs non-linear (ArcFlow’s analytic mixture-of-momentums); (2) role separation across student steps — DP-DMD’s v-prediction first step + DMD later steps for diversity preservation; TMD’s outer/inner backbone-flow-head decoupling; RMD’s resolution-stage separation (low-res for high-noise semantic stage, high-res for low-noise refinement); (3) train-vs-inference alignment — TMD’s flow-head rollout inside DMD, Causal Forcing’s AR-teacher ODE init to satisfy frame-level injectivity, RMD’s logSNR-invariance mapping that distills over identical denoising states regardless of resolution; (4) what to post-train with after distillation — TDM-R1’s non-differentiable-reward RL using deterministic-trajectory unbiased per-step reward estimates. Common substrate: distillation is now the practical production path to 1–4 NFE on production-scale models (SD3.5-M, FLUX, Qwen-Image-20B, Wan2.1 14B, Z-Image 6B), while teacher-free recipes like SoFlow / MF hold the from-scratch frontier on smaller image benchmarks.

  • [2026-07-20] Introducing Cosmos 3 Edge: Alongside Cosmos 3 Edge, NVIDIA releases a Cosmos 3 Super 4-Step Distillation checkpoint (T2I + I2V) — distribution-matching distillation reducing 35–50 denoising steps to 4 for up to 25× faster inference, with the post-training script released so developers can rerun the distillation on their own domains.
  • [2026-07-16] FlowDAgger: Human-in-the-Loop Adaptation of Generative Robot Policies in Latent Space: FlowDAgger ports the flow-matching inversion primitive (reverse-time ODE integration + local refinement to recover a source noise from a target sample) out of the image-editing setting and into robot-policy adaptation: each human corrective action a* is mapped to the noise ẑ* that the frozen flow-matching policy would have integrated forward into it, and those (o, ẑ*) pairs supervise a small latent policy that steers the base sampler at deployment. A wiki-first datapoint that noise-space inversion is a stable supervision signal beyond diffusion editing — with the frozen base policy playing the role of the frozen image DiT and the latent policy playing the role of a tiny steering adapter.
  • [2026-07-10] LingBot-VA 2.0: Native Video-Action Pretraining for Generalizable Robot Control: Few-step consistency distillation is applied to both the video expert and the action expert in a Mixture-of-Transformers video-action model, and stacked with quantization, sparse MoE routing, and asynchronous execution to reach a reported 225 Hz peak inference frequency on real robot hardware — one of the strongest filed cases of distillation as a load-bearing lever for real-time closed-loop VLA/WFM deployment (not just for image/video generation latency).
  • [2026-07-09] LingBot-World 2.0 / LingBot-World-Infinity — Infinite Worlds with Versatile Interactions: LingBot-World 2.0 distills a causal-fast real-time student from a causal-pretrained base to hit 720p / 60 fps interactive throughput — a step up from v1’s 480p / 16 fps LingBot-World-Fast. Method details not yet disclosed (tech report URL unresolvable at filing); positioned as the throughput datapoint that lets the open Wan-based interactive-WFM stack quote higher fps/resolution than DeepMind’s closed Project Genie surface.
  • [2026-07-04] RoboWorld: Fast and Reliable Neural Simulators for Generalist Robot Policy Evaluation: Step Forcing distills an AR video diffusion world model to a 4-step schedule by supervising denoising from one-step self-forwarded priors under the same schedule used at inference, avoiding the expense and action-following degradation of Self Forcing.
  • [2026-06-29] Qwen-Image-2.0-RL Technical Report: Qwen-Image-2.0-RL uses on-policy distillation with trajectory-level velocity matching as the final merge step of a multi-task RL pipeline — consolidating a T2I GRPO specialist and an edit GRPO specialist into a single deployable student. New use case for OPD in the diffusion stack: not step-count reduction or quality lift over a single larger teacher, but multi-policy consolidation after task-specific RL.
  • [2026-06-23] Causal World Modeling for Robot Control (LingBot-VA): Noisy History Augmentation is a non-distillation route to inference-step reduction: instead of training a separate few-step student, the model is trained with video history corrupted to a sampled flow-time t (using the same flow-matching interpolation x_t = (1-t)·noise + t·x_0) so the inverse-dynamics action decoder learns to read actions from partially denoised video latents. At inference the video need only be denoised to t* (not 0), roughly halving per-step denoising work — without retraining a student. Structurally analogous to Causal Forcing’s distillation trick but applied at the action-decoder readout rather than the video velocity field.
  • [2026-06-18] Data-Forcing Distillation: Restoring Diversity and Fidelity in Few-Step Video Generation: DFD (NVIDIA + Michigan + Toronto): one-line post-training fix for DMD/DMD2 that mixes real-data samples into the teacher-score evaluation roughly half the time, turning DMD’s reverse-KL student-only gradient into a teacher-score discrepancy that pulls the student toward missed modes and away from spurious ones — restores diversity and resolves over-saturation in 100–300 finetune steps on Wan2.1-1.3B and Cosmos-Predict2.5-2B (T2V, I2V, and AR video), with the claim that explicit real-data regularization is stronger than DMD2’s implicit GAN regularization. A pure-objective counterpart to DP-DMD’s role-separated-student fix for the same DMD failure mode.
  • [2026-06-03] Bootstrap Your Generator: Unpaired Visual Editing with Flow Matching: BYG (NVIDIA, ICML 2026): bootstraps an unpaired editor out of a pretrained flow-matching generator by manipulating its velocity field around the target-style manifold, rather than fitting on paired (source, edit) data — reframes editing as a flow-trajectory bootstrap problem.
  • [2026-06-02] Stable-Layers: Fine-Tuning Image Layer Decomposition Models with VLM-Scored Reinforcement Learning: Stable-Layers post-trains a diffusion-based layer decomposer with Flow-GRPO + LoRA using VLM-only rewards — a flow-matching RL fine-tune (no distillation step shortening) but in the same Flow-GRPO family that ties this cluster to RL alignment for flow models.
  • [2026-06-01] Pre-training under infinite compute: Stanford CRFM — LLM-pretraining analog of the ensemble-and-distill recipe. Distilling an 8-ensemble of 300M models into a single 300M student via sequence-level KD retains 83% of the ensemble benefit, outperforming the regularized recipe’s NN\to\infty asymptote at finite inference cost. Self-distillation (300M teacher → 300M student of identical architecture, real + synthetic token mixture) beats the teacher — concrete recipe for the Distillation Scaling Laws thesis that distillation pushes the loss-vs-parameter curve to the left.
  • [2026-06-01] RenderFlow: Single-Step Neural Rendering via Flow Matching: RenderFlow (Disney Research, CVPR 2026): single-step neural rendering trained directly with a flow-matching objective on (G-buffer, photoreal RGB) pairs — a from-scratch single-step generator targeting graphics rendering rather than image synthesis. Complements teacher-student distillation recipes (TMD, DMD, ArcFlow) with a domain where deterministic single-step inference is the requirement, not a speedup target, because stochastic samplers produce temporal jitter when applied frame-by-frame.
  • [2026-05-30] Mamoda2.5: Enhancing Unified Multimodal Model with DiT-MoE: Mamoda2.5 reports a joint few-step distillation + RL framework compressing a 30-step video-editing teacher to a 4-step student with up to 95.9× inference speedup over open-source baselines, applied to a 25B/3B-active DiT-MoE.
  • [2026-05-29] MRT: Masked Region Transformer for Layered Image Generation and Editing at Scale: MRT distills its 20B masked region diffusion model for multi-layer transparent image generation down to 8 steps with “minimal quality degradation,” contributing a 20B-scale data point for distillation in the layered-image regime.
  • [2026-05-25] PiD: Fast and High-Resolution Latent Decoding with Pixel Diffusion (PiD, NVIDIA): DMD2-distill a pixel-diffusion decoder (not a generator) from multi-step to 4 NFE. The decoder operates in pixel space at the target upsampled resolution (e.g., 2048×2048) and conditions on a noise-corrupted latent via a sigma-aware adapter. Distillation outcome: 4-step inference, ~210 ms on GB200 / sub-1 s on a consumer RTX 5090 at 13 GB peak memory for the 512² → 2048² decode, ~6× faster than cascaded diffusion-based super-resolution pipelines (and 5.9× over SeedVR2 on the same task) with better visual fidelity per project-page side-by-sides. New target for DMD2 distillation: the decode+upsample step rather than the generative diffusion itself.
  • [2026-05-23] Streaming Autoregressive Video Generation via Diagonal Distillation (DiagDistill, ICLR 2026): per-chunk asymmetric step schedule for AR video distillation (more steps early, fewer late, e.g. 4-3-2-2-2-2-2 over 7 chunks). Three pieces: (a) Diagonal Denoising schedule itself; (b) Diagonal Forcing — conditioning the next chunk on a noised version of the previous chunk’s output during training (t_d≈100 / 1000 optimal) to match inference-time KV-cache state and eliminate the implicit “next-noise-level prediction” that drives over-saturation drift; (c) Flow Distribution Matching — a flow-field reverse-KL on top of DMD using a learnable latent-frame-difference module, restoring motion amplitude lost at low NFE. On Wan2.1-T2V-1.3B: 31 FPS, 0.37 s first-frame latency, VBench Total 84.48 — 277.3× faster than the Wan2.1 teacher while slightly exceeding it on Total and Semantic. Ablations show Diagonal Denoising buys the speedup (quality stays at Self-Forcing level when removed), while Diagonal Forcing + Flow Distribution Matching buy the quality.
  • [2026-05-23] Cross-Resolution Distribution Matching for Diffusion Distillation (RMD, Huawei Cloud + NJUST): cross-resolution DMD — partitions the teacher’s logSNR trajectory into resolution-specific segments (low-res for high-noise / semantic stage, high-res for low-noise / refinement), derives student timestep intervals via an SD3-style logSNR-shift compensation so distillation occurs over matched denoising states across resolutions, and upsamples the student state into the teacher’s high-res space with a mixture of predicted noise (inherits teacher ODE trajectory) and Gaussian noise (bridges large resolution gaps). Two-stage (lo+hi) recipe on SDXL @ 4 NFE: 33.4× speedup and beats DMD2 / TDM / SDXL-Turbo / SDXL-Lightning across HPS/AeS/CLIP. On Wan2.1-T2V-14B: 25.6× speedup and VBench 82.51 vs 6-NFE TDM 80.48 / 6-NFE DMD2 80.30 — the first wiki datapoint where cascading resolution beats fixed-resolution distillation at fewer effective NFE on a 14B video teacher. The component ablation shows the cross-resolution RM objective (not the cascaded upsampling alone) is what does the lifting; UP without RM underperforms the baseline.
  • [2026-05-23] TDM-R1: Reinforcing Few-Step Diffusion Models with Non-Differentiable Reward (TDM-R1, ICML 2026): RL post-training of a TDM few-step student with non-differentiable rewards (GenEval / OCR / HPSv3). Hinges on the unbiased-per-step- reward property of deterministic ODE trajectories; introduces a diffusion-parameterized Surrogate Reward with group-BT learning and EMA reference, and a marginal-level (not trajectory-level) reverse-KL to the TDM teacher. SD3.5-M @ 4 NFE → GenEval 0.92; Z-Image @ 4 NFE strictly dominates 100-NFE Z-Image and Z-Image-Turbo on every reported metric.
  • [2026-05-23] Self-Flow: Self-Supervised Flow Matching for Scalable Multi-Modal Synthesis: internal self-distillation (EMA teacher inside the same generator, Dual- Timestep Scheduling) as an alternative to external DINO-based REPA. Not a teacher-student distillation between two diffusion models, but the same conceptual lever applied within one model; filed here because the “self-distillation as supervision” framing overlaps with the multi-step→few-step thread.
  • [2026-05-23] ArcFlow: Unleashing 2-Step Text-to-Image Generation via High-Precision Non-Linear Flow Distillation (ArcFlow): non-linear-flow 2-step distillation; analytic mixture-of-continuous- momentum velocity gives closed-form ODE integration. <5% of parameters trained (rank-256 LoRA on FFN + output head). Beats pi-Flow / TwinFlow / SenseFlow / Qwen-Image-Lightning at 2 NFE on FLUX.1-dev and Qwen-Image-20B; +85.7% diversity over Qwen-Image- Lightning on OneIG-Bench.
  • [2026-05-23] Diversity-Preserved Distribution Matching Distillation for Fast Visual Synthesis (DP-DMD): role-separated student where step 1 uses v-prediction (diversity) and steps 2..k use DMD (quality), gradient stop between them. No perceptual / adversarial / extra-real-data machinery — beats LPIPS-DMD and GAN-DMD on the diversity-quality Pareto at 4 NFE.
  • [2026-05-23] Causal Forcing++: Scalable Few-Step Autoregressive Diffusion Distillation for Real-Time Interactive Video Generation (Causal Forcing++): replaces offline PF-ODE trajectory generation with causal consistency distillation; scales the Causal Forcing pipeline to 1- and 2-step frame-wise real-time interactive video.
  • [2026-05-23] Causal Forcing: Autoregressive Diffusion Distillation Done Right for High-Quality Real-Time Interactive Video Generation (Causal Forcing): theoretical correction to Self-Forcing / CausVid — bidirectional teacher → AR student violates frame-level injectivity required by ODE distillation. Three-stage fix: teacher-forced AR diffusion fine-tune → causal ODE distillation from the AR teacher → asymmetric DMD. +19.3% Dynamic Degree on real-time-streaming T2V at matched compute.
  • [2026-05-22] Transition Matching Distillation for Fast Video Generation (TMD, NVIDIA + NYU): distills Wan2.1 1.3B / 14B T2V into ≤4-NFE students. Two ideas: (a) decouple the student into a heavy semantic backbone (run once per outer transition) and a recurrent flow head (run for several inner flow updates); (b) close DMD’s train/inference gap by rolling out the flow head during DMD gradient computation. MeanFlow-style stage-1 init for the flow head outperforms vanilla flow-matching init.
  • Is the “deterministic trajectory” property the actual gating condition for non-differentiable-reward RL, or just the cleanest one? TDM-R1 leans on TDM specifically. ArcFlow, TMD, DP-DMD all use deterministic ODE sampling at inference — can the TDM-R1 recipe drop in on top of each? Predicted yes (Eq. 6 of TDM-R1: Reinforcing Few-Step Diffusion Models with Non-Differentiable Reward is generic in the trajectory’s determinism), but untested. RMD’s cascaded trajectory adds resolution transitions to this question.
  • What is the right marginal-KL anchor for from-scratch few-step models (SoFlow: Solution Flow Models for One-Step Generative Modeling, Self-Evaluation Unlocks Any-Step Text-to-Image Generation)? TDM-R1’s KL is to a pretrained TDM teacher; from-scratch generators have no such anchor. Either the recipe needs a teacher-free regularizer (the generator’s own EMA?) or it doesn’t transfer cleanly.
  • Linear vs non-linear vs role-separated vs resolution-separated vs chunk-asymmetric objective — which matters most at each NFE? ArcFlow (non-linear), DP-DMD (role-separated across student steps), RMD (resolution-separated across trajectory segments), and now DiagDistill (asymmetric across AR chunks) all report wins on FLUX-class or Wan-class models at 2–4 effective NFE. None of them compare against the others directly; the field is missing a clean ablation that decouples trajectory parameterization from per-step objective from resolution schedule from per-chunk schedule.
  • Are the cascade ideas composable? RMD cascades resolution along the noise axis; DiagDistill cascades step count along the AR-chunk axis. The two cascades are orthogonal and should compose trivially on Wan2.1-T2V-14B; no one has tried.
  • Architectural-gap fixes (Causal Forcing’s three-stage AR teacher), train/inference-gap fixes (TMD’s flow-head rollout in DMD; DiagDistill’s Diagonal Forcing for the AR-chunk-specific version), trajectory-parameterization fixes (ArcFlow’s closed-form non-linear ODE), and resolution-cascade fixes (RMD’s logSNR-invariance mapping) all appear to attack independent failure modes. Are they composable? In principle yes; nobody has combined two yet. RMD + TMD’s flow head on Wan2.1-14B is an obvious next target; DiagDistill + Causal Forcing/++ on AR video is another.
  • Does TDM-R1 generalize to video? Z-Image is image-class, 6B. The natural target is Wan2.1 14B via TMD’s flow head + a video reward (VBench, ViewCrafter geometry score). Open whether the unbiased-per-step-reward argument survives the much higher-dimensional reward landscape of video. DiagDistill’s Flow Distribution Matching is a non-RL prior for the motion component of that reward and may compose.
  • Distillation + RL vs distillation-of-RL. TDM-R1 ablates against “first RL the many-step teacher with DGPO, then distill with TDM”: it converges fast but plateaus because the teacher caps the student. The clean composition appears to be RL during distillation, with a jointly-trained reward — but this is only one ablation, on SD3.5-M.
  • Does distilling the decoder (PiD) compose with distilling the generator? PiD’s 4-step DMD2 decoder runs on top of an otherwise normal LDM. The total wall-clock saving is largest if the generator is also distilled (DMD2 / pi-Flow / ArcFlow). PiD additionally allows the LDM to exit early via the sigma-aware adapter, which interacts nontrivially with the generator’s distilled step schedule. Whether the combined system beats either component in isolation, and where the right exit point is, has not been measured.