what_they_did
The authors propose a hybrid linear-softmax attention mechanism for video diffusion transformers. They combine gated linear attention (O(N) complexity) with periodic gated-softmax anchors at a 3:1 ratio to restore full-rank token interactions. They also introduce Block Attention Residuals (AttnRes) to propagate refreshed representations across layers, boosting deep-layer effective rank by ~12%. The model is trained from scratch on video data, with proxy studies at reduced resolution to determine the optimal 25% softmax ratio. They evaluate at 5B and 14B scales, using 40-step sampling and a compiled DiT forward pass.
key_findings
- The hybrid attention achieves quality comparable to full-softmax video DiTs (VBench 84.30 at 480p) while maintaining linear attention's favorable long-sequence scaling.
- The compiled DiT forward pass is 3.2x faster than a matched full-softmax baseline at 720p/60 frames, and the gap expands with longer videos.
- With full-stack Sol-Engine optimization (kernel fusion, caching, sparse attention), the 5B pipeline runs in 13.06 seconds at 720p/5 seconds, making it 120x faster than Wan 2.2-A14B on a single H100.
- The block attention residuals boost effective rank in deep layers by ~12%, improving representational capacity.
why_it_matters
This work demonstrates that hybrid attention can recover the expressiveness of full softmax attention at substantially lower computational cost, enabling scalable long, high-resolution video generation on limited hardware. The approach challenges the assumption that quadratic attention is necessary for high-quality video diffusion, offering a practical path for deployment.
caveats
The evaluation is limited to a single GPU (H100) and specific benchmarks. The proxy studies for the softmax ratio were done at reduced resolution and may not perfectly generalize to full resolution. The comparison to Wan 2.2-A14B is a single speed benchmark; quality comparisons are limited. The model is trained from scratch, so the method may not directly apply to fine-tuning existing pretrained models.