Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chen, Aiyue, Dong, Bin, Li, Jingru, Lin, Jing, Tian, Kun, Yao, Yiwu, Wang, Gongyi
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:https://arxiv.org/abs/2505.21036
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910994467389440
author Chen, Aiyue
Dong, Bin
Li, Jingru
Lin, Jing
Tian, Kun
Yao, Yiwu
Wang, Gongyi
author_facet Chen, Aiyue
Dong, Bin
Li, Jingru
Lin, Jing
Tian, Kun
Yao, Yiwu
Wang, Gongyi
contents Video generation using diffusion models is highly computationally intensive, with 3D attention in Diffusion Transformer (DiT) models accounting for over 80\% of the total computational resources. In this work, we introduce {\bf RainFusion}, a novel training-free sparse attention method that exploits inherent sparsity nature in visual data to accelerate attention computation while preserving video quality. Specifically, we identify three unique sparse patterns in video generation attention calculations--Spatial Pattern, Temporal Pattern and Textural Pattern. The sparse pattern for each attention head is determined online with negligible overhead (\textasciitilde\,0.2\%) with our proposed {\bf ARM} (Adaptive Recognition Module) during inference. Our proposed {\bf RainFusion} is a plug-and-play method, that can be seamlessly integrated into state-of-the-art 3D-attention video generation models without additional training or calibration. We evaluate our method on leading open-sourced models including HunyuanVideo, OpenSoraPlan-1.2 and CogVideoX-5B, demonstrating its broad applicability and effectiveness. Experimental results show that RainFusion achieves over {\bf 2\(\times\)} speedup in attention computation while maintaining video quality, with only a minimal impact on VBench scores (-0.2\%).
format Preprint
id arxiv_https___arxiv_org_abs_2505_21036
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RainFusion: Adaptive Video Generation Acceleration via Multi-Dimensional Visual Redundancy
Chen, Aiyue
Dong, Bin
Li, Jingru
Lin, Jing
Tian, Kun
Yao, Yiwu
Wang, Gongyi
Computer Vision and Pattern Recognition
Artificial Intelligence
Video generation using diffusion models is highly computationally intensive, with 3D attention in Diffusion Transformer (DiT) models accounting for over 80\% of the total computational resources. In this work, we introduce {\bf RainFusion}, a novel training-free sparse attention method that exploits inherent sparsity nature in visual data to accelerate attention computation while preserving video quality. Specifically, we identify three unique sparse patterns in video generation attention calculations--Spatial Pattern, Temporal Pattern and Textural Pattern. The sparse pattern for each attention head is determined online with negligible overhead (\textasciitilde\,0.2\%) with our proposed {\bf ARM} (Adaptive Recognition Module) during inference. Our proposed {\bf RainFusion} is a plug-and-play method, that can be seamlessly integrated into state-of-the-art 3D-attention video generation models without additional training or calibration. We evaluate our method on leading open-sourced models including HunyuanVideo, OpenSoraPlan-1.2 and CogVideoX-5B, demonstrating its broad applicability and effectiveness. Experimental results show that RainFusion achieves over {\bf 2\(\times\)} speedup in attention computation while maintaining video quality, with only a minimal impact on VBench scores (-0.2\%).
title RainFusion: Adaptive Video Generation Acceleration via Multi-Dimensional Visual Redundancy
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2505.21036