Papers
arxiv:2607.26811

DistillAlign: Coordinating Mode Covering and Mode Seeking in Autoregressive Video Distillation

Published on Jul 29
· Submitted by
Li Jiaxing
on Jul 30
Authors:
,
,
,
,
,
,
,
,
,

Abstract

Existing autoregressive video distillation methods commonly adopt a Distribution Matching Distillation (DMD)-based multi-stage pipeline. However, they typically decouple the initialization and DMD stages -- which then pursue different target distributions -- and judge the intermediate student mainly by visual scores such as VBench. In this paper, we revisit this design from a distributional perspective. Given the mode-seeking nature of the distribution matching loss, a good initialization should match the mode coverage of the target DMD teacher, rather than merely pursuing high quality. To analyze this, we introduce a distributional evaluation protocol that measures precision and coverage between student and teacher distributions in a shared latent space. It exposes differences hidden by visual scores: some initializations reach high precision but low coverage, leading to suboptimal refinement, while mode-covering ones preserve broader support. Furthermore, even when the target distributions are aligned, DMD's reverse-KL objective can still drive the student toward high-probability teacher regions in late training, reducing coverage and diversity. To address this, we propose joint distillation, which combines DMD's mode-seeking objective with a Consistency Distillation-based mode-covering constraint. Experiments show that our method improves generation quality, coverage, and diversity; notably, even with a Wan-1.3B DMD teacher, it outperforms baselines refined with Wan-14B, underscoring the importance of distributional alignment in autoregressive video distillation.

Community

Paper submitter

We take a distributional view of the multi-stage distillation pipeline and organize the paper around two questions:

1️⃣ For the intermediate initialization, what matters more — higher quality, or a better-matched distribution?
Our experiments give a clear (and slightly surprising) answer: when the initialization and DMD stages aim at the same distribution, even a weaker teacher or lower-quality data beats a stronger-teacher / better-data combo whose distributions are mismatched. Alignment beats raw quality. Motivated by this, we design a distribution-aware evaluation (teacher-normalized precision & coverage) that gives a far richer lens on intermediate distillation checkpoints than visual scores like VBench.

2️⃣ Once the distributions are matched, what happens inside the distillation stage?
We track how the student distribution evolves and find that pure reverse-KL DMD contracts it in late training — so even as visual quality stays high, diversity quietly collapses. To counter this, we propose joint distillation, which balances the mode-covering and mode-seeking objectives: broad coverage preserved, quality still sharpened.

Notably, with only a 1.3B DMD teacher, DistillAlign already surpasses baselines refined with a 14B teacher. We open-source our training data, models at every stage, and the full measured intermediate results — come give it a try! 🚀

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2607.26811
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2607.26811 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2607.26811 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2607.26811 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.