English

Head Forcing: Long Autoregressive Video Generation via Head Heterogeneity

Computer Vision and Pattern Recognition 2026-05-15 v1 Artificial Intelligence

Abstract

Autoregressive video diffusion models support real-time synthesis but suffer from error accumulation and context loss over long horizons. We discover that attention heads in AR video diffusion transformers serve functionally distinct roles as local heads for detail refinement, anchor heads for structural stabilization, and memory heads for long-range context aggregation, yet existing methods treat them uniformly, leading to suboptimal KV cache allocation. We propose Head Forcing, a training-free framework that assigns each head type a tailored KV cache strategy: local and anchor heads retain only essential tokens, while memory heads employ a hierarchical memory system with dynamic episodic updates for long-range consistency. A head-wise RoPE re-encoding scheme further ensures positional encodings remain within the pretrained range. Without additional training, Head Forcing extends generation from 5 seconds to minute-level duration, supports multi-prompt interactive synthesis, and consistently outperforms existing baselines. Project Page: https://jiahaotian-sjtu.github.io/headforcing.github.io/.

Keywords

Cite

@article{arxiv.2605.14487,
  title  = {Head Forcing: Long Autoregressive Video Generation via Head Heterogeneity},
  author = {Jiahao Tian and Yiwei Wang and Gang Yu and Chi Zhang},
  journal= {arXiv preprint arXiv:2605.14487},
  year   = {2026}
}
R2 v1 2026-07-22T07:11:48.131Z