Search papers, labs, and topics across Lattice.
This paper introduces ScalingAttention, a training-free framework that addresses the computational bottleneck of 3D full attention in Diffusion Transformers (DiTs) by discovering an Intrinsic Sparse Topology. The method leverages a weight-encoded, scale-invariant topology that allows for efficient extraction and decouples topology discovery from sparsity control, utilizing techniques like WEST and FAST to optimize performance. Experimental results demonstrate that ScalingAttention achieves up to 1.90X end-to-end speedup while maintaining superior fidelity compared to existing state-of-the-art methods.
Achieving up to 1.90X speedup in video generation without sacrificing fidelity, ScalingAttention redefines efficiency in Diffusion Transformers.
While Diffusion Transformers (DiTs) have revolutionized high-fidelity video generation, their reliance on 3D full attention creates a quadratic computational bottleneck. Existing sparse methods face a dilemma: dynamic pruning suffers from prohibitive runtime overhead and memory fragmentation, while static heuristics fail to capture fine-grained dependencies. In this work, we propose ScalingAttention, a training-free framework grounded in a key inductive bias: while individual activations are input-dependent, the high-mass attention regions for each head rapidly converge to a stable, prompt-agnostic Intrinsic Sparse Topology. This topology is weight-encoded, scale-invariant, and efficient to extract. ScalingAttention decouples topology discovery from sparsity control via: (1) WEST (Weight-Encoded Sparse Topology), which extracts a robust block-sparse prior mask offline to eliminate runtime search; (2) FAST (Fidelity-Aware Sensitivity Tuning), which adaptively tunes head-wise sparsity based on diffusion fidelity requirements. To ensure practical acceleration, we co-design a hardware-aligned bit-wise block-sparse kernel. Experiments on Wan2.1 show up to 1.90X end-to-end speedup with superior fidelity, establishing a new Pareto frontier over state-of-the-art baselines.