cs.CVMar 10, 2026

FrameDiT: Diffusion Transformer with Matrix Attention for Efficient Video Generation

Authors: Minh Khoa Le, Kien Do, Duc Thanh Nguyen, Truyen Tran

Organizations: Applied Artificial Intelligence Initiative, Deakin University, Australia · FPT Smart Cloud, Vietnam · Deakin University, Australia

Abstract

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 spatio-temporal tokens which can be modeled using Diffusion Transformers (DiTs). However, this approach faces a trade-off between the strong but expensive Full 3D Attention and the efficient but temporally limited Local Factorized Attention. To resolve this trade-off, we propose Matrix Attention, a frame-level temporal attention mechanism that processes an entire frame as a matrix and generates query, key, and value matrices via matrix-native operations. By attending across frames rather than tokens, Matrix Attention effectively preserves global spatio-temporal structure and adapts to significant motion. We build FrameDiT-G, a DiT architecture based on MatrixAttention, and further introduce FrameDiT-H, which integrates Matrix Attention with Local Factorized Attention to capture both large and small motion. Extensive experiments show that FrameDiT-H achieves state-of-the-art results across multiple video generation benchmarks, offering improved temporal coherence and video quality while maintaining efficiency comparable to Local Factorized Attention.

Figures & tables

Appendix figures & tables3 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Mixture of Distributions Matters: Dynamic Sparse Attention for Efficient Video Diffusion Transformers

    Jan 14, 2026Yuxi Liu, Yipeng Hu, Zekun Zhang +2Video Diffusion ModelsDiffusion Transformer

  2. Where Compute Matters: Heterogeneous Attention for Efficient Video Diffusion

    Sep 25, 2026Olga Zatsarynna, Denis Korzhenkov, Juergen Gall +2Video Diffusion ModelsDiffusion Model Inference Acceleration