Accelerating Text-to-Video Generation with Calibrated Sparse Attention

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Yehezkel, Shai, Yadin, Shahar, Elata, Noam, Ostrovsky-Berman, Yaron, Kawar, Bahjat
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912945754079232
author Yehezkel, Shai
Yadin, Shahar
Elata, Noam
Ostrovsky-Berman, Yaron
Kawar, Bahjat
author_facet Yehezkel, Shai
Yadin, Shahar
Elata, Noam
Ostrovsky-Berman, Yaron
Kawar, Bahjat
contents Recent diffusion models enable high-quality video generation, but suffer from slow runtimes. The large transformer-based backbones used in these models are bottlenecked by spatiotemporal attention. In this paper, we identify that a significant fraction of token-to-token connections consistently yield negligible scores across various inputs, and their patterns often repeat across queries. Thus, the attention computation in these cases can be skipped with little to no effect on the result. This observation continues to hold for connections among local token blocks. Motivated by this, we introduce CalibAtt, a training-free method that accelerates video generation via calibrated sparse attention. CalibAtt performs an offline calibration pass that identifies block-level sparsity and repetition patterns that are stable across inputs, and compiles these patterns into optimized attention operations for each layer, head, and diffusion timestep. At inference time, we compute the selected input-dependent connections densely, and skip the unselected ones in a hardware-efficient manner. Extensive experiments on Wan 2.1 14B, Mochi 1, and few-step distilled models at various resolutions show that CalibAtt achieves up to 1.58x end-to-end speedup, outperforming existing training-free methods while maintaining video generation quality and text-video alignment.
format Preprint
id arxiv_https___arxiv_org_abs_2603_05503
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Accelerating Text-to-Video Generation with Calibrated Sparse Attention
Yehezkel, Shai
Yadin, Shahar
Elata, Noam
Ostrovsky-Berman, Yaron
Kawar, Bahjat
Computer Vision and Pattern Recognition
Recent diffusion models enable high-quality video generation, but suffer from slow runtimes. The large transformer-based backbones used in these models are bottlenecked by spatiotemporal attention. In this paper, we identify that a significant fraction of token-to-token connections consistently yield negligible scores across various inputs, and their patterns often repeat across queries. Thus, the attention computation in these cases can be skipped with little to no effect on the result. This observation continues to hold for connections among local token blocks. Motivated by this, we introduce CalibAtt, a training-free method that accelerates video generation via calibrated sparse attention. CalibAtt performs an offline calibration pass that identifies block-level sparsity and repetition patterns that are stable across inputs, and compiles these patterns into optimized attention operations for each layer, head, and diffusion timestep. At inference time, we compute the selected input-dependent connections densely, and skip the unselected ones in a hardware-efficient manner. Extensive experiments on Wan 2.1 14B, Mochi 1, and few-step distilled models at various resolutions show that CalibAtt achieves up to 1.58x end-to-end speedup, outperforming existing training-free methods while maintaining video generation quality and text-video alignment.
title Accelerating Text-to-Video Generation with Calibrated Sparse Attention
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2603.05503