English
Related papers

Related papers: CHAI: CacHe Attention Inference for text2video

200 papers

To efficiently process visual data at scale, researchers have proposed two techniques for lowering the computational overhead associated with the underlying deep learning models. The first approach consists of leveraging a specialized,…

Databases · Computer Science 2021-02-18 Jiashen Cao , Ramyad Hadidi , Joy Arulraj , Hyesoon Kim

Transformer-based language models have found many diverse applications requiring them to process sequences of increasing length. For these applications, the causal self-attention -- which is the only component scaling quadratically w.r.t.…

Machine Learning · Computer Science 2023-06-05 Matteo Pagliardini , Daniele Paliotta , Martin Jaggi , François Fleuret

Diffusion Transformers are fundamental for video and image generation, but their efficiency is bottlenecked by the quadratic complexity of attention. While block sparse attention accelerates computation by attending only critical key-value…

Computer Vision and Pattern Recognition · Computer Science 2026-02-04 Haopeng Li , Shitong Shao , Wenliang Zhong , Zikai Zhou , Lichen Bai , Hui Xiong , Zeke Xie

Diffusion models can synthesize realistic co-speech video from audio for various applications, such as video creation and virtual agents. However, existing diffusion-based methods are slow due to numerous denoising steps and costly…

Computer Vision and Pattern Recognition · Computer Science 2025-10-06 Beijia Lu , Ziyi Chen , Jing Xiao , Jun-Yan Zhu

The success of deep neural networks (DNN) in machine perception applications such as image classification and speech recognition comes at the cost of high computation and storage complexity. Inference of uncompressed large scale DNN models…

Machine Learning · Computer Science 2020-07-06 Yihao Fang , Shervin Manzuri Shalmani , Rong Zheng

This paper presents a method to accelerate the inference process of diffusion transformer (DiT)-based text-to-speech (TTS) models by applying a selective caching mechanism to transformer layers. Specifically, I integrate SmoothCache into…

Audio and Speech Processing · Electrical Eng. & Systems 2025-09-11 Siratish Sakpiboonchit

Despite the promise of synthesizing high-fidelity videos, Diffusion Transformers (DiTs) with 3D full attention suffer from expensive inference due to the complexity of attention computation and numerous sampling steps. For example, the…

Computer Vision and Pattern Recognition · Computer Science 2025-02-18 Hangliang Ding , Dacheng Li , Runlong Su , Peiyuan Zhang , Zhijie Deng , Ion Stoica , Hao Zhang

Graphics rendering applications increasingly leverage neural networks in tasks such as denoising, supersampling, and frame extrapolation to improve image quality while maintaining frame rates. The temporal coherence inherent in these tasks…

Graphics · Computer Science 2025-06-18 Lufei Liu , Tor M. Aamodt

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…

Computer Vision and Pattern Recognition · Computer Science 2026-03-06 Shai Yehezkel , Shahar Yadin , Noam Elata , Yaron Ostrovsky-Berman , Bahjat Kawar

Recent advances in diffusion models have demonstrated remarkable capabilities in video generation. However, the computational intensity remains a significant challenge for practical applications. While feature caching has been proposed to…

Computer Vision and Pattern Recognition · Computer Science 2025-07-29 Xuran Ma , Yexin Liu , Yaofu Liu , Xianfeng Wu , Mingzhe Zheng , Zihao Wang , Ser-Nam Lim , Harry Yang

Recent advances in diffusion models have enabled high-quality video generation, but the additional temporal dimension significantly increases computational costs, making training and inference on long videos prohibitively expensive. In this…

Computer Vision and Pattern Recognition · Computer Science 2025-12-09 Xingyang Li , Muyang Li , Tianle Cai , Haocheng Xi , Shuo Yang , Yujun Lin , Lvmin Zhang , Songlin Yang , Jinbo Hu , Kelly Peng , Maneesh Agrawala , Ion Stoica , Kurt Keutzer , Song Han

Diffusion Transformers (DiTs) with 3D full attention power state-of-the-art video generation, but suffer from prohibitive compute cost -- when generating just a 5-second 720P video, attention alone takes 800 out of 945 seconds of total…

Computer Vision and Pattern Recognition · Computer Science 2025-06-06 Peiyuan Zhang , Yongqi Chen , Runlong Su , Hangliang Ding , Ion Stoica , Zhengzhong Liu , Hao Zhang

Diffusion-based talking head models generate high-quality, photorealistic videos but suffer from slow inference, limiting practical applications. Existing acceleration methods for general diffusion models fail to exploit the temporal and…

Graphics · Computer Science 2026-01-21 Jianzhi Long , Wenhao Sun , Rongcheng Tu , Dacheng Tao

In video and image generation tasks, Diffusion Transformer (DiT) models incur extremely high computational costs due to attention mechanisms, which limits their practical applications. Furthermore, with hardware advancements, a wide range…

Computer Vision and Pattern Recognition · Computer Science 2026-04-21 Aiyue Chen , Yaofu Liu , Junjian Huang , Guang Lian , Yiwu Yao , Wangli Lan , Jing Lin , Zhixin Ma , Tingting Zhou

Video-language models (VLMs) learn to reason about the dynamic visual world through natural language. We introduce a suite of open datasets, benchmarks, and recipes for scalable oversight that enable precise video captioning. First, we…

Computer Vision and Pattern Recognition · Computer Science 2026-04-28 Zhiqiu Lin , Chancharik Mitra , Siyuan Cen , Isaac Li , Yuhan Huang , Yu Tong Tiffany Ling , Hewei Wang , Irene Pi , Shihang Zhu , Ryan Rao , George Liu , Jiaxi Li , Ruojin Li , Yili Han , Yilun Du , Deva Ramanan

The quadratic complexity of attention remains the central bottleneck in long-context inference for large language models. Prior acceleration methods either sparsify the attention map with structured patterns or permanently evict tokens at…

Computation and Language · Computer Science 2026-05-04 Dongwon Jo , Beomseok Kang , Jiwon Song , Jae-Joon Kim

High-fidelity video generation remains challenging for diffusion models due to the difficulty of modeling complex spatio-temporal dynamics efficiently. Recent video diffusion methods typically represent a video as a sequence of…

Computer Vision and Pattern Recognition · Computer Science 2026-04-21 Minh Khoa Le , Kien Do , Duc Thanh Nguyen , Truyen Tran

Text-to-image generation using diffusion models has seen explosive popularity owing to their ability in producing high quality images adhering to text prompts. However, production-grade diffusion model serving is a resource intensive task…

Computer Vision and Pattern Recognition · Computer Science 2023-12-08 Shubham Agarwal , Subrata Mitra , Sarthak Chakraborty , Srikrishna Karanam , Koyel Mukherjee , Shiv Saini

We explore the role of attention mechanism during inference in text-conditional diffusion models. Empirical observations suggest that cross-attention outputs converge to a fixed point after several inference steps. The convergence time…

Computer Vision and Pattern Recognition · Computer Science 2025-02-27 Haozhe Liu , Wentian Zhang , Jinheng Xie , Francesco Faccio , Mengmeng Xu , Tao Xiang , Mike Zheng Shou , Juan-Manuel Perez-Rua , Jürgen Schmidhuber

We propose Chunk-wise Attention Transducer (CHAT), a novel extension to RNN-T models that processes audio in fixed-size chunks while employing cross-attention within each chunk. This hybrid approach maintains RNN-T's streaming capability…

Machine Learning · Computer Science 2026-03-02 Hainan Xu , Vladimir Bataev , Travis M. Bartley , Jagadeesh Balam