Preprint
Large Language Models

Context-Matched Distillation: Teacher Causality for Autoregressive Video Distillation

Hmrishav Bandyopadhyay, Xuanchi Ren, Zijian Huang, Jay Zhangjie Wu, Tianshi Cao, Ruilong Li, Bryan Chu, Sanja Fidler, Yi-Zhe Song, Zian Wang
August 13, 2026

0

Citations

0

Influential Citations

Venue

2026

Year

Abstract

Interactive autoregressive video generation demands both low-latency rollouts and precise online control. Few-step distillation accelerates generation by reducing denoising steps, while online control imposes a causal constraint: frames and blocks should depend on history and controls available during generation. Existing video distribution matching distillation (DMD) pipelines, however, often supervise causal few-step students using bidirectional teachers that score complete clips. The score for a target can therefore depend on future frames and controls that were unavailable when the student generated it, misaligning teacher supervision with the student's causal information set. We introduce Context-Matched Distillation (CMD), a causal DMD framework that aligns teacher supervision with the information available when each target is generated. CMD replaces bidirectional full-clip scoring with a causal teacher that evaluates each target without access to future frames or controls. The same causal teacher initializes the few-step student, establishing a consistent causal formulation across teacher training, student distillation, and inference. Beyond aligning the temporal information boundary, Prefix Scoring matches supervision to the student's realized rollout context by evaluating each target under the cached student-generated prefix that produced it. Prefix Corruption further stabilizes training by perturbing unreliable prefixes produced early in training while preserving this target-context alignment. With a simple causal formulation, CMD naturally extends to frame-wise and chunk-wise generation, long video distillation, and camera-conditioned distillation. Experiments demonstrate state-of-the-art aggregate performance among autoregressive methods on both short- and long-video benchmarks, together with substantially improved adherence to time-varying camera controls.

Analysis

Why This Paper Matters

Interactive autoregressive video generation is a critical frontier for AI applications like real-time content creation, virtual environments, and robotics. The demand for low-latency rollouts and precise online control has driven interest in few-step distillation, but existing methods suffer from a fundamental misalignment: they use bidirectional teachers that score complete clips, violating the causal constraints of autoregressive generation. This paper identifies this issue and proposes a solution that aligns teacher supervision with the student's causal information set, which is essential for reliable and controllable generation.

The significance lies in bridging the gap between training and inference. By ensuring that the teacher never uses future information, CMD creates a consistent causal formulation across teacher training, student distillation, and inference. This not only improves performance but also enables natural extensions to frame-wise and chunk-wise generation, long videos, and camera conditioning—capabilities that are crucial for practical deployment. The paper's focus on causal alignment is a step toward more trustworthy and controllable generative models.

Technical Contributions

  • Causal Teacher: Replaces bidirectional full-clip scoring with a teacher that evaluates each target without access to future frames or controls, ensuring supervision matches the student's causal information set.
  • Prefix Scoring: Matches supervision to the student's realized rollout context by evaluating each target under the cached student-generated prefix that produced it, improving alignment.
  • Prefix Corruption: Stabilizes training by perturbing unreliable prefixes from early training while preserving target-context alignment, preventing training instability.
  • Unified Framework: The same causal teacher initializes the few-step student, establishing consistency across teacher training, student distillation, and inference.
  • Extensions: Naturally supports frame-wise and chunk-wise generation, long video distillation, and camera-conditioned distillation, making it versatile for various autoregressive video tasks.

Results

The paper reports state-of-the-art aggregate performance among autoregressive methods on both short- and long-video benchmarks. Specifically, CMD demonstrates substantially improved adherence to time-varying camera controls, which is a key requirement for interactive applications. While exact metrics are not provided in the abstract, the claim of state-of-the-art performance suggests significant gains over existing autoregressive distillation methods. The improvements in camera control adherence indicate that the causal alignment directly translates to better controllability, a critical factor for user-facing systems.

Significance

CMD addresses a core challenge in autoregressive video generation: maintaining causality while leveraging powerful distillation techniques. By aligning teacher supervision with the student's information set, it enables more efficient and controllable generation, which is essential for real-time interactive applications. This work could influence future research in causal generative modeling, not only for video but also for other sequential domains like audio and text. The framework's extensibility to various generation granularities and conditioning signals makes it a valuable contribution to the field, potentially accelerating the deployment of interactive AI systems.