arXiv AI

Bidirectional Centered Score Matching for Discrete Diffusion Distillation

Bidirectional Centered Score Matching (BCSM) is introduced as an efficient within-position distillation objective for discrete diffusion models, reducing the complexity from ≠|V|^2 to ≠K^2 by using an ≠K bidirectional active set. BCSM aligns relative logit geometry while maintaining a factorized ≠K objective, and it incorporates cross-timestep transition alignment. Experiments show significant efficiency gains, achieving up to 6× fewer neural function evaluations and notable improvements in HumanEval and MMMU metrics at single-step inference.

arXiv Machine Learning
Oct 28

Cumulative Marginal Matching for One-Step Diffusion

The paper introduces Cumulative Marginal Matching (CMM), a new objective for training few‑step diffusion and flow‑matching models. CMM treats flow‑matching as a sequence of intermediate marginal distributions and directly optimizes the cumulative Wasserstein‑2 discrepancy between the base and few‑step models across the entire trajectory. It derives a tractable surrogate for this objective, shows that Shortcut‑CSL is a first‑order approximation of CMM, and reports improved generation quality on standard benchmarks under the same training budget.

By Paribesh Regmi, Rui Li
arXiv Computer Vision
Oct 5

VDOT++: Unified Few-Step Video Generation via Unbalanced Optimal Transport Distillation

The paper introduces VDOT++, a unified distillation framework for few-step video generation across text-to-video, image-to-video, and condition-based tasks. It extends the previous VDOT method by incorporating an asymmetric unbalanced optimal transport distillation that allows student tokens to carry less mass, and an α1 ground cost that uses a weighted median to preserve modes. The approach combines distribution matching with adversarial refinement and cross-scale distillation, achieving competitive performance on multiple video benchmarks with only four sampling steps.

By Yutong Wang, Xingtong Ge, Enhuai Liu, Yunke Wang, Tianfan Xue, Yu Qiao, Yaohui Wang, Xinyuan Chen, Chang Xu
arXiv Machine Learning
Oct 8

Consistent Distribution Matching for Data-Free Diffusion Distillation

Consistent Distribution Matching is a data‑free distillation technique that accelerates diffusion and flow models by unifying sample generation and score estimation in a single student network. The method requires only a frozen teacher and a trainable student, optimizing a single objective that provably leads to Wasserstein convergence of the student’s flow‑map to the teacher’s marginals. On ImageNet 256×256, it achieves an FID of 2.04 with one function evaluation and 1.37 with four evaluations after just 40 training epochs, outperforming existing data‑free baselines.

By Yuxiang Fu, Qi Yan, Zike Wu, Yongxing Zhang, Purang Abolmaesumi, Lele Wang, Renjie Liao