Radial Attention: O(n log n) Sparse Attention with Energy Decay — Generating “Long Videos” Cheaply
One-Line Summary (TL;DR)
Based on the observation that post-softmax attention energy decays exponentially with distance (in time/space), the authors design a static mask that decays compute density in the same way, substantially cutting the training/inference cost of long-video generation. (source: §4.1–§4.2, Eq.3–Eq.4)
Key Numbers (Summary)
Params: N/A (method) | Evaluation models: Mochi 1 10B Params / HunyuanVideo 13B Params / Wan2.1 14B Params (source: §5.1)
Context: e.g., generating a 5s 720p clip with HunyuanVideo needs roughly 115K tokens (source: §3)
Architecture: Sparse self-attention mask (softmax preserved) replacing dense self-attention (source: §3–§4, Eq.2)
Complexity: Attention O(n^2) → O(n log n) (source: Abstract, Fig.2)
Serving (end-to-end latency, single H100)
- Default length: HunyuanVideo 1649s → 876s (1.88×) / Wan2.1-14B 1630s → 917s (1.77×) (source: Tab.1)
- 4× length (Hunyuan): 2895s → 781s (3.71×) (source: Tab.2)
Training (LoRA-based length extension)
- 4× length (Hunyuan): 93.6h → 21.4h (4.37×) (source: Tab.2)
Attention fidelity: output MSE 3.9e-3 (lower than SVG’s 4.4e-3 and STA’s 1.5e-2) (source: §5/§6 Discussion)
Terminology: TPOT is not directly reported as a headline metric in this paper; instead the authors compare end-to-end latency (s) and training time (h) for video diffusion, Vision Reward, etc. (source: Tab.1–Tab.2)
Core Idea
The paper’s central claim is that attention in video diffusion models exhibits spatiotemporal energy decay — post-softmax scores drop as the temporal/spatial distance between tokens grows — and that converting this into a reduction in compute density via a static sparse mask (Radial Attention) makes it possible to cut training/inference on long videos to O(n log n) while keeping softmax, maintaining or improving quality. (source: Abstract, §4.1–§4.3)
Background: The Problem They Address
Why Do “Long Videos” Get Expensive?
- Video adds a temporal axis, so the token count explodes, and because 3D dense attention builds an n×n matrix, time/memory blow up to O(n^2). (source: §3)
- For example, generating a 5-second 720p clip in HunyuanVideo involves roughly 115K tokens, so increasing resolution/frames translates directly into an attention bottleneck. (source: §3)
Where Did Prior SOTA Stand?
- The latest video-generation SOTA adopts 3D dense attention on top of a DiT backbone to capture long-range dependencies, but cost grows quadratically with the number of frames, straining both training and serving. (source: Related Work)
- Video-specific sparse methods (STA, SVG, etc.) improve efficiency, but STA suffers from a fixed receptive field, and SVG relies on runtime profiling, which is unreliable on long-video distributions and hard to use directly in training. (source: §1, §2, Fig.3)
- Linear-attention alternatives provide a global receptive field but can miss local detail and degrade quality, leaving a design space between “dense (O(n^2))” and “linear (O(n))”. (source: Related Work)
New Approach: Radial Attention
Radial Attention assumes/observes “distance-based energy decay” and builds a sparse mask accordingly: temporal direction halves density band by band, and spatial direction keeps only a diagonal neighborhood within frame-to-frame blocks. (source: §4.1–§4.2, Fig.5, Eq.4)
1) Observation: Spatiotemporal Energy Decay
- Post-softmax attention scores decay as temporal/spatial distance grows, and the authors argue via regression that an exponential function fits well. (source: §4.1, Fig.4, Fig.9)
- This is modeled in the form of Eq.3 (an exp decay over a linear combination of distances). (source: Eq.3)
2) Design: “Energy Decay → Compute-Density Decay”
- Temporal direction: the main diagonal neighborhood is split into several bands; as you move to outer bands the compute density halves, while the band width doubles, so total computation per band stays bounded by a constant. (source: §4.2, Fig.5(a))
- Spatial direction: reflecting the observation that interactions at the same (or nearby) spatial location dominate within a frame-i↔j block, the diagonal width is shrunk as the temporal distance grows. (source: §4.2)
- When the diagonal width would drop below 1, instead of narrowing further the authors keep the diagonal in only selected frame-pair blocks — a “frequency reduction” rule. (source: §4.2, Fig.5(b))
- An attention sink (every token attends to the 1st frame) is added to shore up quality. (source: §4.2, Fig.5(c))
3) Result: O(n log n) + Hardware Friendly
- O(n log n) is derived from an upper bound on the number of non-zero entries (allowed connections) of the mask. (source: §4.2, Eq.5–Eq.6, Appx A.1)
- The implementation uses 128×128 block sparsity so it runs on modern GPUs. (source: §4.2 Hardware-friendly block sparsity)
How It Works: A Concrete Walkthrough
Take a tiny toy video and walk through, step by step, how the Radial Attention mask realizes “dense nearby, sparse far away”. (source: §4.2)
Toy setting
- Let the number of frames be f=4 (frame 0..3) and tokens per frame be s=4 (spatial position 0..3). (source: §4.2 notation n=fs)
- A query token is denoted (frame i, pos k) and a key token (frame j, pos l). (source: Eq.4 definition)
Step 1) Assign a band by temporal distance
- |i−j|=0 is band 0 (100% density); as |i−j| grows, the pair is classified into outer bands, and density is halved per band. (source: §4.2 Temporal density decay)
For example, when i=0, j=0 falls in band 0, j=1 in band ±1, and j=2 in a further band. (source: §4.2 band indexing)
Step 2) Inside a frame-to-frame block, keep “only the diagonal neighborhood”
- When i and j are close (k≈l), keep several diagonals around it; the farther apart they are, the smaller the allowed diagonal width. (source: §4.2 Spatial density decay)
For the toy, with “width=1”, only (k=l) pairs survive. (source: the k=l case of Eq.4)
Step 3) When the pair is so far that width < 1, connect “only intermittently”
- For far-apart frame pairs, instead of keeping every (k=l), keep diagonals only in selected (i,j) blocks so the average density matches. (source: §4.2 diagonal frequency rule)
Step 4) Always attend to the 1st frame via the attention sink
- Every token can attend to tokens in frame 0, providing a stable global anchor. (source: §4.2 attention sink)
The diagram below simplifies the “dense → radial” flow. (source: Fig.3 concept)
flowchart LR A["Dense 3D Attention all pairs O(n^2)"] -->|"observation: energy decay"| B["Compute density decay design"] B --> C["Static Mask M (temporal bands + spatial diagonals)"] C --> D["Masked Softmax Attention O(n log n)"] D --> E["Length extension via LoRA fine-tuning"]
Performance Validation: Key Results
1) Default length: “quality maintained + speedup”
- On HunyuanVideo (117 frames), Ours reports PSNR 27.3, LPIPS 0.114, latency 876s, a 1.88× speedup. (source: Tab.1)
- On Wan2.1-14B (69 frames), Ours reports PSNR 23.9, LPIPS 0.163, latency 917s, a 1.77× speedup. (source: Tab.1)
- In the same table, speed is similar to SVG (e.g., 1.90× vs 1.88×), but the authors stress that the static mask also targets training/length extension. (source: Tab.1, Fig.3 discussion)
2) 2×/4× length extension: “extend cheaply with LoRA”
- On HunyuanVideo 4× (509 frames), Ours reports sparsity 88.3%, training 21.4h (4.37×), inference 781s (3.71×), Vision Reward 0.134. (source: Tab.2)
- On Mochi 1 4× (667 frames), Ours reports sparsity 85.5%, training 17.4h (2.83×), inference 386s (2.57×), Vision Reward 0.113. (source: Tab.2)
3) Empirical validity of the “exponential decay” assumption
- Fig.9 reports that fitting the decay curve with exp(−ax+b) yields R^2 > 0.985. (source: Fig.9)
Our Take: Strengths, Limitations, and Why This Work Matters
Strengths
- Because it “prunes only unimportant relations” while keeping softmax attention, it is more amenable than linear-attention families (which change the architecture substantially) to preserving pretrained weights + light tuning. (source: Abstract, §4.3)
- Being a static mask, it avoids SVG’s runtime-profiling errors and its inapplicability to training, fitting the goal of adapting directly to the “long-video distribution”. (source: §1, §2, Fig.3)
- On the LoRA front, the authors argue that with dense attention LoRA lags full fine-tuning until it catches up at 4×, whereas under Radial Attention LoRA can match or surpass full fine-tuning. (source: Fig.9(a) discussion)
Limitations
- The authors state that modeling attention scores as “exponential decay” (Eq.3) simplifies the complex structure of natural video. (source: Limitations)
- As Eq.6 shows, complexity is O(n log n) in the number of frames, but they acknowledge a quadratic component remains in resolution. (source: Limitations, Eq.6 reference)
Why It Matters?
- In video generation, “length extension” immediately translates into a practical cost problem (tuning/serving); this paper carefully lays out a path — mask + LoRA — that increases length without substantially changing the pretrained model. (source: §4.3, Tab.2)
What’s Next? The Road Ahead
- The authors currently use Radial Attention only for “length-extension fine-tuning”, but — like NSA/MoBA — they leave introducing it at the pre-training stage for “native long-video support” as future work. (source: Limitations/Discussion)
- They also propose as follow-up a more scalable attention mechanism that reduces the quadratic bottleneck in resolution. (source: Limitations)
One “Secret Weapon”: “Radial (integrated) Mask” vs Spatial-only / Temporal-only
On HunyuanVideo 4× (509 frames), the “integrated mask (Ours)” lifts Vision Reward far above variants that keep only the spatial or temporal direction. (source: Tab.2)
| Variant | Sparsity | Train Time (h) | Train Speedup | Infer Time (s) | Infer Speedup | Vision Reward | Δ Vision Reward vs Ours |
|---|---|---|---|---|---|---|---|
| Spatial | 88.3% | 20.7h | 4.52× | 755s | 3.83× | 0.112 | -0.022 |
| Temporal | 88.2% | 21.1h | 4.44× | 774s | 3.74× | 0.083 | -0.051 |
| Ours | 88.3% | 21.4h | 4.37× | 781s | 3.71× | 0.134 | 0.000 |
- Read on, spatial-only captures the same-position (diagonal) structure across frames but under-uses the temporal-band design that spends more budget on “nearby frames”; temporal-only, conversely, keeps the cross-frame flow but captures less intra-frame local detail, so the quality loss is larger. (source: §4.2 design intent, Tab.2 results)
Click the toggle below to expand a detailed LLM Q&A session on the paper.
▶️Click to expand
Prompt 1.1.1 (Research Gap)
Analyze the paper's 'Introduction' and 'Related Work' sections and explain the core research gap, decisive limitations of prior work, or unresolved questions this study explicitly sets out to address. Summarize what the 'state of the art' looked like at the time of publication, as the authors describe it.Problem Setting: Why the Bottleneck Shifts to Attention for “Long Videos” (source: §1, §3)
- Video adds a temporal axis, so the number of latent tokens explodes; for example, generating a 5s, 720p video in HunyuanVideo requires about 115K tokens. (source: §3)
- Over-compressing latents degrades quality, so token reduction effectively has a floor. (source: §3)
- Recent video diffusion models use 3D dense attention to capture spatiotemporal correlations, and because the attention matrix is $n \times n$, time/memory complexity grows to $O(n^2)$. (source: §3)
- FlashAttention-style methods reduce memory but leave the quadratic bottleneck in time intact, which becomes the key constraint for high-resolution/long videos. (source: §3)
Decisive Limitations of Prior Methods: The Clash Between “Efficiency” and “Long-Video Generalization/Trainability” (source: §1, §2, §3)
- SVG profiles heads online as spatial/temporal to pick masks, but on unseen length distributions such as longer videos, the head classification can be wrong, and that error can be amplified during training, leading to degraded performance. (source: §1)
- The authors also contrast that dynamic masking of the SVG type “cannot go beyond the length limit itself” and is hard to use reliably for training on long-video distributions. (source: Fig.3, §2)
- Linear-attention replacement families change softmax attention, so the structural change is large, and recovering original quality with only “weak” fine-tuning is difficult. (source: §1)
- Static local-window families such as STA have a fixed receptive field, which constrains long-range dependency and is fragile on long videos. (source: §2)
- Sparse patterns borrowed from LLMs (e.g., power-of-two distances) ignore video’s spatial/temporal structure and may be suboptimal for video generation. (source: §2)
- On the “long-video generation” axis, the authors also mention training-free extensions such as RIFLEx, which does 2× extrapolation via RoPE frequency adjustment, but note it is not presented as a family that fundamentally solves the quadratic attention cost problem itself. (source: §2)
The Core Research Gap the Paper Explicitly Targets (source: §3)
- The open question the authors pose is: “Can we design an attention pattern that has the expressiveness of dynamic sparse masking, but is also applicable to training as a static pattern and runs in a hardware-friendly way?” (source: §3)
- In other words, the prior options were split between (1) 3D dense — good quality but blocked at $O(n^2)$ for training/serving, (2) dynamic sparse — faster inference but hard to train/generalize reliably on long-video distributions, and (3) linearization / large-scale architectural change — hard to recover quality. This, in the authors’ view, is the “gap”. (source: §1, §2, §3)
Summary of the SOTA at Publication, as the Authors Describe It (source: §2)
- Since Sora, video diffusion models have moved to DiT backbones, and on the thread where Latte proposed spatial/temporal decoupled attention, recent SOTA has chosen 3D dense attention to model “joint spatiotemporal dynamics” more strongly. (source: §2)
- But this SOTA choice (3D dense) can be orders-of-magnitude heavier than decoupled attention and scales quadratically in the number of frames, creating a major barrier for both training and deployment. (source: §2)
- On the efficiency axis, distillation/caching/quantization/distributed inference etc. also apply to video, but because video depends on 3D dense attention, the bottleneck has moved not to the FFN but to the attention layers. (source: §2)
- Representative video-specific sparse methods cited are STA (sliding 3D window) and SVG (dynamic per-head pattern selection); both improve efficiency but each leaves problems on “long videos” (STA: limited long-range; SVG: profiling reliability degrades at unseen lengths). (source: §2)
Key Numbers (minimum summary related to the gap) (source: §3, Fig.2)
- Token-scale example: 115K tokens at 5s, 720p (HunyuanVideo). (source: §3)
- Bottleneck complexity contrast: Dense attention $O(n^2)$ → the target is a transition to $O(n \log n)$. (source: Fig.2, §1)
- For long videos (e.g., 500-frame, 720p), the paper presents as motivation a 9× reduction in attention compute, 3.7× faster inference, and 4.6× lower tuning cost. (source: Fig.2)
Prompt 1.1.2 (Central Hypothesis)
What is this paper's central hypothesis or core claim? State it as one clear, concise sentence of the form: 'The authors hypothesize that, by using [proposed technique], they can achieve [concrete outcome] that overcomes [existing limitation].'The authors hypothesize that, by using a static $O(n \log n)$ Radial Attention mask that reflects video’s spatiotemporal structure, they can (source: §2) avoid the $O(n^2)$ cost bottleneck of 3D dense attention and (source: §2) the long-video-distribution instability and quality degradation of existing sparse/linear families, (source: §2) while keeping generation quality close to dense, (source: §5.2, Fig.6) achieving ~1.9×/1.8× end-to-end speedups at default length on HunyuanVideo and Wan2.1-14B, and efficiently enabling extension to longer videos (source: §5.2).
Prompt 1.2.1 (Identifying Novelty)
Based on the full paper, list the 1-3 most important and original contributions as distinct items. For each, clearly state whether it is a new architectural component, a new training technique, a new theoretical insight, a new dataset, or a new application of an existing methodology.1) Radial Attention: a static $O(n \log n)$ sparse-mask design that reflects video’s spatiotemporal structure (new architectural component)
- Radial Attention proposes a static sparse attention mechanism that lowers the $O(n^2)$ bottleneck of 3D dense attention to $O(n \log n)$. (source: Fig.2, §1)
- The mask is designed with a band structure (varying diagonal width/density) so that attention compute density decays exponentially with temporal distance (gap between frames): nearby frames are seen densely, distant frames sparsely. (source: Fig.5, §4)
- An attention sink is also added so that every token can reference the first frame, reinforcing quality. (source: §4)
- As a motivating case, the paper highlights a 9× reduction in attention compute and 3.7× end-to-end latency speedup at 500-frame, 720p, underscoring practical efficiency on long videos. (source: Fig.2)
2) Spatiotemporal Energy Decay: modeling how attention scores decrease with distance (new theoretical insight)
- The authors define the drop in post-softmax attention scores as spatial/temporal distance grows as Spatiotemporal Energy Decay, and model it as exponential decay in spatial and temporal distance. (source: §1)
- Translating this observation into “computation-density decay” and realizing it with a simple but hardware-friendly static mask is the key design principle. (source: §1, §4)
3) A LoRA-based fine-tuning framework that cheaply adapts existing video diffusion models via “length extension” (new training technique / new application of an existing methodology)
- Because Radial Attention keeps softmax attention intact and only “prunes unimportant token relations”, it lets existing pretrained video diffusion models be adapted to longer sequences with lightweight tuning such as LoRA. (source: §1)
- The authors state that the length-extension LoRA is also compatible with existing style LoRA, presenting scalability for real workflows (controlling style and length simultaneously). (source: §1, §5.2)
- For 4× longer video generation they claim a 4.4× reduction in tuning cost and 3.7× faster inference (with quality maintained), targeting “training and serving scaling together”. (source: §1, Fig.2)
Prompt 1.2.2 (Strengths from the Authors’ Perspective)
From the authors' perspective, why is their approach superior to previous methods? Quote or explain in plain terms the key arguments they use to support the novelty and strengths of their work.1.2.2 Authors’ Strengths: “Dynamic expressiveness with a static mask + training and serving scaling together”
Strength 1 : Resolving the dynamic (sparse) family’s “untrainability/instability” with a static pattern (source: §1, Fig.3)
- The authors believe SVG-type dynamic masking causes head-profiling misclassification on unseen length distributions, and that this error is amplified during optimization, leading to degraded performance. (source: §1)
- They also cite as a limitation that “dynamic masking decides patterns on the fly — an overhead — and is not applied to training”. (source: §4)
- As a remedy, Radial Attention implements “energy decay → compute-density decay” with a static sparse mask, targeting a pattern that can be used directly in training as well. (source: §1, §4)
Strength 2 : Anchoring the design principle in the observed “energy decay”, justifying a simple yet effective mask (source: §1, Fig.4)
- The authors argue they define the drop in post-softmax attention scores with increasing spatial/temporal distance as Spatiotemporal Energy Decay and model it as exponential decay in distance. (source: §1, Fig.4)
- On this basis, they fold SVG’s spatial/temporal head split into a single Radial Attention and lower the computational complexity to $O(n \log n)$. (source: §1, Fig.2)
Strength 3 : Hardware friendliness + minimal structural change (softmax preserved) favors “real deployment/tuning” (source: §2, §1)
- After summarizing that existing $O(n \log n)$ families (e.g., LSH/hierarchical/multiresolution/fast multipole) are hardware-unfriendly and scale poorly at large sizes, the authors contrast that they obtain both hardware friendliness and scale with a “simple static attention mask”. (source: §2)
- And because they keep softmax attention and only “prune unimportant token relations”, existing pretrained video diffusion models can be adapted to long sequences with lightweight tuning such as LoRA. (source: §1)
Strength 4 : Quantifying the quality-efficiency tradeoff under the same compute budget (source: Tab.1, §2)
- At default length, the authors summarize that under the same compute budget, Radial Attention is consistently better than STA/PA on PSNR (dB), SSIM, and LPIPS, and shows video fidelity similar to SVG. (source: Tab.1)
- On HunyuanVideo (117 frames), Ours reports 339 TFLOPs, 876 s for a 1.88× speedup, with PSNR 27.3 dB, SSIM 0.886, LPIPS 0.114. (source: Tab.1)
- On Wan2.1-14B (69 frames), Ours reports 323 TFLOPs, 917 s for a 1.77× speedup, with PSNR 23.9 dB, SSIM 0.842, LPIPS 0.163. (source: Tab.1)
- As an additional long-video example, they state that at 509 frames, 720p, attention computation is reduced 9× versus dense. (source: §2, Fig.2)
- For 4× longer video generation they claim a 4.4× reduction in tuning cost and 3.7× faster inference (quality maintained), emphasizing “training and serving scaling together”. (source: §1)
Prompt 1.3.1 (Step-by-Step Algorithm Explanation)
Explain the core algorithm, model architecture, or main methodology step by step. Assume the reader is a graduate student in AI. In particular, create a very simple toy example and sample inputs — such as simple sentences, a 3×3 pixel image, a small state space — and show through the example how the input is transformed into the output through each step. Define every key term and variable the moment it appears.1.3.1 Step-by-Step Algorithm: Build the Radial Attention mask → apply Sparse Attention → extend length with LoRA
Terms and variables (source: §3, §4)
- $f$ : number of frames. (source: §4)
- $s$ : tokens per frame. (source: §4)
- $n$ : total number of tokens, $n = f s$. (source: §4)
- $Q,K,V \in \mathbb{R}^{n\times d}$ : the query/key/value matrices of attention; $d$ is the embedding dimension. (source: §3)
- $M \in {-\infty, 0}^{n\times n}$ : sparsity mask (added to logits). (source: §3)
- $\tilde{M} \in {-\infty, 0}^{f\times f\times s\times s}$ : (frame, frame, spatial, spatial) 4D mask. (source: §4.2)
Step 0 Basic Attention / SparseAttention formulas (source: §3)
Dense attention is computed as follows. (source: §3)
$$ \mathrm{Attention}(Q,K,V)=\mathrm{softmax}!\left(\frac{QK^\top}{\sqrt{d}}\right)V $$Sparse attention removes some pairs by adding a mask $M$ to the logits. (source: §3)
$$ \mathrm{SparseAttention}(Q,K,V)=\mathrm{softmax}!\left(\frac{QK^\top+M}{\sqrt{d}}\right)V $$Step 1 Observation: model Spatiotemporal Energy Decay as exponential decay (source: Fig.4, Eq.3)
The authors report that post-softmax attention scores decrease as temporal distance and spatial distance grow, and define this as Spatiotemporal Energy Decay. (source: Fig.4, §4.1)
For a given query token (frame $i_0$, position $k_0$), the probability $p_{js+l}$ of attending to the token at position $l$ of frame $j$ is assumed to be bounded as follows. (source: Eq.3)
$$ p_{js+l} \le C_{\mathrm{rel}} e^{-\alpha|j-i_0|-\beta|l-k_0|}, p_{i_0 s + k_0} $$- $\alpha$ : temporal decay rate (per frame). (source: Eq.3)
- $\beta$ : spatial decay rate (per spatial index). (source: Eq.3)
Step 2 Core design: convert “energy decay” into “compute-density decay” (source: §4.2, Fig.5)
Step 2-A Temporal axis: split into bands and halve the compute density for each band farther away (source: §4.2, Fig.5a)
A rule sets the compute density according to the temporal distance between frames $i$ and $j$ as follows. (source: §4.2)
$$ \rho_t(i,j)=\left(\frac{1}{2}\right)^{\left\lfloor \log_2(\max(|i-j|,1))\right\rfloor} $$This rule partitions the attention map into diagonal-centered bands and halves each band’s density as it moves away from the center. (source: §4.2, Fig.5a)
Step 2-B Spatial axis: inside frame-to-frame blocks keep only “diagonals (similar positions)”, shrinking the width as distance grows (source: §4.2, Fig.5b)
The authors find energy concentrates on “similar spatial locations”, so within frame blocks they keep diagonal-form interactions. (source: §4.2)
The diagonal width kept in the block between frames $i$ and $j$ (specifically, the half-diagonal width) is set as follows. (source: §4.2)
$$ w(i,j)=\left\lfloor \frac{s}{2^{\left\lfloor \log_2(\max(|i-j|,1))\right\rfloor}} \right\rfloor $$Step 2-C When the diagonal width drops below 1: sample by reducing the diagonal “frequency” (source: Fig.5 caption, Eq.4)
When the diagonal width becomes too small (intuitively “no diagonal width left to shrink”), the authors keep the diagonals but subsample the frame blocks to match the “same amortized density decay”. (source: Fig.5 caption)
Step 3 Define the formal mask $\tilde{M}$ and flatten it into the 2D mask $M$ (source: Eq.4, §4.2)
Radial Attention’s 4D mask is defined by the following conditions. (source: Eq.4)
- Allowed (0): (i) near-diagonal pairs within a temporal band that satisfy the diagonal-width condition, or (ii) for distant frames, diagonal samples that satisfy the condition. (source: Eq.4)
- Blocked ($-\infty$): all other token pairs. (source: Eq.4)
Since the actual attention op (Eq. (2)) needs the 2D mask $M$, the frame/spatial indices are flattened as follows. (source: §4.2)
$$ M_{is+k,; js+l} = \tilde{M}_{i,j,k,l} $$Step 4 Quality reinforcement: Attention Sink (everyone attends to the first frame) (source: §4.2, Eq.8)
To improve video quality, an attention sink is added so every token attends to the tokens of the first frame (frame 0). (source: §4.2)
The appendix defines it as a separate mask that includes allowance toward $j=0$. (source: Appx A.1, Eq.8)
Step 5 Execution view: compute block-sparsely in blocks, not token by token (source: §4.2)
For modern-hardware efficiency, attention is computed block-wise, and the implementation uses a 128×128 (tokens × tokens) block size. (source: §4.2)
Step 6 Adapting to long videos: insert Radial Attention + LoRA into the attention projections (source: §4.3)
Pretrained models were trained on short-video distributions, so training directly on long videos is assumed to be expensive. (source: §4.3)
Radial Attention lowers fine-tuning cost to $O(n\log n)$, and combining LoRA reduces overhead further. (source: §4.3)
LoRA is applied to the Q/K/V/O projections of the attention layers. (source: §4.3)
Walking Through a Toy Example (3×3-token frames, 4 frames)
Setup
- Number of frames: $f=4$
- Tokens per frame: $s=9$ → a 3×3 grid flattened in row-major order
- Total tokens: $n=fs=36$
The spatial index of each frame is laid out as follows.
(0) (1) (2)
(3) (4) (5)
(6) (7) (8)Goal
Take the query token as frame $i_0=2$, position $k_0=4$ (center) and see step by step which key tokens are allowed (0) by the mask.
(A) Same frame ($j=2$)
- temporal distance: $|2-2|=0$
- $\left\lfloor \log_2(\max(0,1))\right\rfloor = 0$
- diagonal width: $w = \left\lfloor \frac{s}{2^0} \right\rfloor = 9$
Intuitively, “nearby, so it attends to almost everything”. → Inside frame 2, (k=4) can attend to many positions in frame 2.
(B) Neighboring frame ($j=1$ or $j=3$)
- temporal distance: $|2-j|=1$
- $\left\lfloor \log_2(1)\right\rfloor = 0$
- diagonal width: $w = \left\lfloor \frac{9}{1} \right\rfloor = 9$
Neighboring frames still get high density. → In frame 1/3 too, (k=4) can attend fairly broadly.
(C) A farther frame ($j=0$)
- temporal distance: $|2-0|=2$
- $\left\lfloor \log_2(2)\right\rfloor = 1$
- diagonal width: $w = \left\lfloor \frac{9}{2} \right\rfloor = 4$
Now “the farther away, the more it centers on similar positions” kicks in. Reading the near-diagonal condition of the paper’s Eq. (4) simply as “|k−l| must be small”, → in frame 0, (k=4) attends to roughly some of l=1..7 (near the diagonal), because the width has shrunk.
(D) Attention Sink bonus
With the attention sink on, any query additionally gets an attend path to the first-frame (frame 0) tokens, so → it partially recovers the connections cut in (C) as “global anchors for quality reinforcement”. (source: §4.2)
Prompt 1.3.2 (‘Secret Weapon’ Identification)
Pick one core component and, for removal/replacement/scale change, present Δ(metric) in a table and explain the mechanism behind the change (e.g., gating load balance, rotary vs ALiBi, replacing sparse attention's half-window).1.3.2 The ‘Secret Weapon’: The “decay form” of the $O(n\log n)$ mask (Radial vs Harmonic Series)
The core component chosen is the sparsity pattern (mask pattern) (source: Appx C.3). The authors run an ablation that replaces Radial Attention with a Harmonic Series Decay (HS) pattern, in which the diagonal width is inversely proportional to the distance from the main diagonal (source: Appx C.3).
Δ(metric) on removal/replacement (HunyuanVideo, 117 frames, Default Length)
| Setting | PSNR (dB) ↑ | ΔPSNR (dB) | SSIM (—) ↑ | ΔSSIM | LPIPS (—) ↓ | ΔLPIPS | VisionReward (—) ↑ | ΔVR |
|---|---|---|---|---|---|---|---|---|
| HS (replace) | 27.0 dB (source: Tab. A) | — | 0.881 (source: Tab. A) | — | 0.119 (source: Tab. A) | — | 0.136 (source: Tab. A) | — |
| Radial (Ours) | 27.3 dB (source: Tab. A) | +0.3 dB | 0.886 (source: Tab. A) | +0.005 | 0.114 (source: Tab. A) | −0.005 | 0.139 (source: Tab. A) | +0.003 |
Why this change happens: the “energy decay vs compute-budget reallocation” mechanism
- The authors’ assumption is that post-softmax attention scores decrease with spatial/temporal distance, and that this decay is well fit by an exponential form (source: §1, §4.1).
- Radial translates this observation directly into “let compute density also decay exponentially”, halving the band-wise compute density as temporal distance $|i-j|$ grows (source: §4.2, Fig.5).
- It also shrinks the diagonal width inside frame-to-frame blocks as the inter-frame distance grows, and when the width would fall below 1 it lowers the diagonal frequency to sample, forcing “less budget on far frames” (source: §4.2, Fig.5).
- HS, by contrast, sets the diagonal width with inverse-distance decay (source: Appx C.3), so relative to the exponential decay Radial assumes, it allocates relatively more compute to distant frames / low-energy regions (source: Appx C.3, §4.2).
- Consequently, under the same $O(n\log n)$ constraint, Radial concentrates budget on temporally nearby tokens (high-energy interactions) (source: §4.2), which is the logic behind the observed quality gains over HS: higher PSNR/SSIM, lower LPIPS, higher VisionReward (source: §4.2, Tab. A).
Prompt 1.4.1 (Core Results Analysis)
Analyze the key results, including the tables/figures in 'Experiments' or 'Results'. What are the key performance metrics? On which benchmarks were they reported? Summarize the results the authors most emphasize as evidence of success.What are the key performance metrics?
- For video quality/consistency, the authors use PSNR (dB) / SSIM (unitless) / LPIPS (unitless) and Vision Reward (unitless) (source: Tab.1).
- For efficiency, they report attention FLOPs (TFLOPs), end-to-end latency (s), and speedup (×) (source: Tab.1).
- For longer videos (2×/4× length), they additionally report Vision Reward (unitless) and VBench submetrics (Subject Consistency / Aesthetic Quality / Image Quality, all unitless), plus LoRA tuning time (h) and training/inference speedup (×) (source: Tab.2).
On which benchmarks/models was it measured?
- The experiments cover three text-to-video diffusion models: Mochi 1 (10B Params), HunyuanVideo (13B Params), and Wan2.1-14B (14B Params) (source: §5.1).
- Quantitative “default length” comparisons run on HunyuanVideo (117 frames) and Wan2.1-14B (69 frames) under a single-H100 condition (source: Tab.1).
- Quantitative “length extension” comparisons run at 2× and 4× frame counts; Wan2.1-14B reports only 2× due to cost/memory constraints (source: Tab.2; §Long video generation).
Default-length quantitative results: ~1.8× speedup while maintaining quality
HunyuanVideo (117 frames) (source: Tab.1)
| Method | PSNR (dB) | SSIM | LPIPS | Vision Reward | TFLOPs | Latency (s) | Speedup |
|---|---|---|---|---|---|---|---|
| STA(FA3) | 26.7 dB | 0.866 | 0.167 | 0.132 | 331 TFLOPs | 719 s | 2.29× |
| PA | 22.1 dB | 0.764 | 0.256 | 0.140 | 339 TFLOPs | 1002 s | 1.65× |
| SVG | 27.2 dB | 0.895 | 0.114 | 0.144 | 340 TFLOPs | 867 s | 1.90× |
| Ours | 27.3 dB | 0.886 | 0.114 | 0.139 | 339 TFLOPs | 876 s | 1.88× |
- Authors’ claim (gist of the table caption): under the same computation budget, Ours consistently improves PSNR·SSIM·LPIPS over STA/PA, shows fidelity similar to SVG, and achieves about 1.8× speedup on HunyuanVideo/Wan2.1-14B (source: Tab.1).
Wan2.1-14B (69 frames) (source: Tab.1)
| Method | PSNR (dB) | SSIM | LPIPS | Vision Reward | TFLOPs | Latency (s) | Speedup |
|---|---|---|---|---|---|---|---|
| STA(FA3) | 22.9 dB | 0.830 | 0.171 | 0.132 | 322 TFLOPs | 812 s | 2.01× |
| PA | 22.4 dB | 0.790 | 0.176 | 0.126 | 324 TFLOPs | 978 s | 1.67× |
| SVG | 23.2 dB | 0.825 | 0.202 | 0.114 | 324 TFLOPs | 949 s | 1.71× |
| Ours | 23.9 dB | 0.842 | 0.163 | 0.128 | 323 TFLOPs | 917 s | 1.77× |
- At default length, Ours reports PSNR 23.9 dB, latency 917 s, and a 1.77× speedup on Wan2.1-14B (source: Tab.1).
Length-extension (2×/4×) quantitative results: “lower tuning cost + faster inference + maintained quality” as the core evidence of success
HunyuanVideo: at 4× (509 frames), training 4.37× + inference 3.71×
- At 4× (509 frames), Ours reports sparsity 88.3%, training time 21.4 h, training speedup 4.37×, inference time 781 s, inference speedup 3.71×, Vision Reward 0.134 (source: Tab.2).
- At the same 4× (509 frames), Dense (Full) reports training time 93.6 h, inference time 2895 s, Vision Reward 0.133, so by the speedup column Ours delivers large acceleration while Vision Reward stays close (source: Tab.2).
- In the Figure 7 visual comparison at 4× length, Radial Attention (LoRA) reports a higher average Vision Reward than the “Dense Attention (LoRA) baseline” (e.g., Dense Attention 0.133 vs Ours 0.134, unitless) (source: Fig.7).
At 2× (253 frames), the “quality vs efficiency” tradeoff is also highlighted in a table
- At 2× (253 frames), Ours reports sparsity 80.8%, training time 16.2 h, training speedup 2.78×, inference time 339 s, inference speedup 2.35×, Vision Reward 0.126 (source: Tab.2).
- At the same 2× (253 frames), RIFLEx reports sparsity 0.00%, inference time 797 s, Vision Reward 0.128 (in a form interpretable as a “training-free” entry), and the authors state that RIFLEx’s quality degrades at 4× (source: Tab.2; §Long video generation).
Mochi 1: at 4× (667 frames), Ours also posts the highest Vision Reward
- At 4× (667 frames), Ours reports sparsity 85.5%, training time 17.4 h, training speedup 2.83×, inference time 386 s, inference speedup 2.57×, Vision Reward 0.113, VBench S.C. 0.958 / A.Q. 0.618 / I.Q. 0.638 (source: Tab.2).
- At the same 4× (667 frames), PA reports Vision Reward 0.107, VBench S.C. 0.956 / A.Q. 0.633 / I.Q. 0.650, and the authors stress in Appendix Figure D that “Radial Attention achieves the highest Vision Reward” (source: Tab.2; Appx Fig.D).
Wan2.1-14B: at 2× (161 frames), ~2× speedup (presented via table/example)
- At Wan2.1-14B 2× (161 frames), Ours reports sparsity 73.6%, training time 14.5 h, training speedup 1.93×, inference time 2847 s, inference speedup 2.01× (source: Tab.2).
- In the Appendix example, Wan2.1-14B’s Dense Attention (LoRA) shows latency 5735 s and Vision Reward 0.094, whereas Ours shows latency 2847 s and Vision Reward 0.165 (source: Appx Fig.D discussion).
The message the authors most emphasize as “evidence of success” (numbers focused)
- At default length: 1.88× speedup on HunyuanVideo (876 s vs 1649 s), with PSNR/LPIPS reported at levels similar to SVG (source: Tab.1).
- At long length: HunyuanVideo 4× (509 frames) delivers training 4.37× (21.4 h vs 93.6 h) + inference 3.71× (781 s vs 2895 s), maintaining Vision Reward around 0.134 while “cutting tuning and inference cost simultaneously” (source: Tab.2; Fig.1).
- As a scaling trend: at 500-frame 720p, Figure 2 summarizes a 9× reduction in attention compute, 3.7× end-to-end speedup, and 4.6× lower tuning cost (source: Fig.2).
Prompt 1.4.2 (Critical Comparison)
How does the proposed method perform against the main baselines and SOTA models mentioned in the paper? Identify the specific comparison points that most strongly support the superiority claims. Conversely, summarize any results it did not beat, or where improvement was marginal, and why.1) Comparison at default length (training-free, default frames)
Strongest superiority point: “quality maintained at the same compute (TFLOPs) + better quality than STA/PA”
HunyuanVideo (117 frames):
- Ours has PSNR 27.3, higher than STA/PA, and LPIPS 0.114, on par with SVG (lower is better).
- Speed is 1.88×, nearly on par with SVG (1.90×).
Wan2.1-14B (69 frames):
- Ours comes out broadly better than STA/PA/SVG with PSNR 23.9 / SSIM 0.842 / LPIPS 0.163.
- Speed is 1.77×.
Paper’s interpretation (why): even at O(n log n), PA ignores video’s spatiotemporal locality, so real quality drops; STA may be fast, but quality degradation is observed.
Points it did not beat, or that are ambiguous
- At HunyuanVideo default length, Vision Reward for Ours (0.139) is lower than SVG (0.144) or Original (0.141). That is, it reads less as “default-length quality completely dominating SOTA” and more as “roughly on par with SVG”.
- Speed too is less “clearly faster than SVG” and more nearly equal (1.88× vs 1.90×).
- STA is faster in raw speed at 2.29× on Hunyuan, but the paper instead points to visual quality degradation as its downside, and notes STA uses FA-3 while Ours is FA-2-based, providing a fair-comparison context.
2) Comparison at length extension (2×/4×)
Strongest superiority point: “quality at the Dense (Full/LoRA) level at 4× while cutting cost/latency significantly”
HunyuanVideo 4× (509 frames)
- Ours: Vision Reward 0.134, at or above Full dense fine-tuning (0.133), while
- Training 21.4h (4.37×) and Inference 781s (3.71×) cut cost/latency substantially.
Mochi 1 4× (667 frames)
- Ours: Vision Reward 0.113, higher than Full (0.099), with large reductions in speed/training cost.
Paper’s interpretation (why):
- Spatial/Temporal masks have a limited receptive field,
- LongLoRA/PA see globally but miss spatiotemporal correlation, degrading quality,
- SANA (linear attention replacement) needs large-scale retraining and is ill-suited to length-extension fine-tuning.
- Ours, by contrast, is a static mask that keeps softmax and “only prunes less important relations”, so length extension works well even with LoRA.
Points it did not beat, or where improvement was marginal
At HunyuanVideo 2× (253 frames), RIFLEx posts Vision Reward 0.128, slightly higher than Ours (0.126) (a very small margin).
- The paper also says “at 2× RIFLEx improves, but it hits a limit when going longer”.
At Wan2.1-14B 2× (161 frames), Ours’s Vision Reward (0.145) is lower than Full (0.150), so Full leads on peak quality. Ours’s advantages instead are 2.01× inference speedup and 1.93× training-time savings.
3) Auxiliary comparisons that further support the “claims” (evidence of quality preservation)
- In the attention-output MSE comparison, Ours shows lower error than SVG/STA (3.9e-3 vs 4.4e-3 / 1.5e-2), adding evidence that “even after sparsification, it approximates the original attention better”.
Prompt 1.5.1 (Acknowledged and Potential Limitations)
What limitations/weaknesses/failure cases do the authors explicitly acknowledge? Based on your analysis, what do you see as potential limitations (strong assumptions, scalability, computational cost, generalization limits, social impact, etc.)?1.5.1 Acknowledged and potential limitations
Limitations/weaknesses the authors explicitly acknowledge
| Category | Acknowledged limitation | Meaning/impact | Follow-up direction suggested by the authors |
|---|---|---|---|
| Modeling assumption | The exponential decay assumption on attention scores (Eq.3) simplifies the complex spatiotemporal dependencies of natural video. (source: §6, Eq.3) | If the decay shape varies with data/head/situation, the rationale for the mask weakens and the efficiency–quality tradeoff can worsen. (source: §6) | They say there is room to improve efficiency and performance by “understanding/modeling the underlying data structure more deeply”. (source: §6) |
| Computational complexity | It achieves $O(n\log n)$ in frame length, but complexity in resolution remains quadratic (Eq.6). (source: §6, Eq.6) | In “long video (f↑) + high resolution (s↑)”, the resolution axis becomes the bottleneck, so the gains may be limited for high-resolution long-form generation. (source: §6, Eq.6) | They state they will explore “more scalable attention for high-resolution video generation”. (source: §6) |
| Training/applicability scope | Currently Radial Attention is used only for length-extension fine-tuning; pre-training application was not done. (source: §6) | For “native long-video” support, stability/generalization at the pre-training stage remains to be validated. (source: §6) | It could be applied at pre-training like NSA and MoBA, but they leave it as future work. (source: §6) |
| Social impact | Malicious users could exploit it for deepfakes/disinformation, and the access gap to minimal compute could worsen the digital divide. (source: §D) | Efficiency gains can directly mean “wider spread of generation tools”, so deployment without safeguards can raise risk. (source: §D) | They recommend responsible deployment, adherence to ethical standards, and development of detection methods, and say they will specify code/model licenses. (source: §D) |
Potential limitations (analysis-based)
Radial Attention uses a static mask grounded in “decay with distance (time/space)”, designed so tokens attend more to “similar spatial positions”. (source: §1, Fig.3(b))
- Therefore, in videos with many “semantically important interactions that are far apart in spacetime (e.g., long-running narrative threads, reappearances after scene changes)”, aggressive pruning could potentially lead to quality degradation. (source: §1, Fig.3(b))
Because the authors themselves note that the exponential decay assumption of Eq.3 is a simplification, in data distributions whose decay deviates from exponential, the relations “cut away by the mask” could potentially be genuinely important. (source: §6, Eq.3)
The text clearly says the complexity bound is read as $O(n\log n)$ when frame count $f$ is large and spatial-resolution token count $s$ is fixed. (source: §4.2, Eq.6)
- That is, in scenarios where resolution (= $s$) grows along with frames (high-resolution, long-form), the advantage shrinks, and eventually the “resolution quadratic” noted in the authors’ own limitations could potentially dominate. (source: §6, Eq.6)
For hardware efficiency the implementation uses block sparsity with a 128×128 block size. (source: §4.2)
- In that case, actual speed can depend heavily on “block alignment / kernel efficiency and density of the sparse pattern” on top of the theoretical $O(n\log n)$, so the gains could potentially be smaller for certain resolution/frame combinations. (source: §4.2)
An attention sink (every token attends to the first frame) is added to improve quality. (source: §4.2, Fig.5(c))
- This can bias toward strongly anchoring the “initial-frame condition”, so it could potentially act as a constraint in videos whose content changes drastically over time. (source: §4.2)
Experiments center on three T2V diffusion models (Mochi 1: 10B Params, HunyuanVideo: 13B Params, Wan2.1: 14B Params) and default resolutions (480p/720p). (source: §5.1)
- It is therefore natural to see generalization to video generators with very different “architecture/resolution/length distributions” as needing further validation. (source: §5.1)
On licensing, the paper notes OpenVid-1M has “no explicit license”. (source: §E)
- This can be a legal/operational risk for reproducibility or derivative distribution. (source: §E)
Prompt 1.5.2 (Future Research Trajectories)
What future research directions do the authors propose? In light of the limitations, suggest reasonable next steps or alternative directions.1.5.2 Future research trajectories
Future directions the authors propose (explicit)
- Refining the decay assumption: since the exponential decay assumption on attention scores (Eq.3) simplifies the complex spatiotemporal dependencies of natural video, they say that understanding/modeling the data structure more deeply could improve both efficiency and performance. (source: §6)
- High-resolution scaling: because the current method still has quadratic complexity in resolution (Eq.6), they say they will explore more scalable attention for high-resolution video generation. (source: §6)
- Pre-training application: currently Radial Attention is used only for length-extension fine-tuning, but applying it at pre-training — like NSA/MoBA — to support “native long video” is left as future work. (source: §6)
- Mitigating LoRA-merging quality (style bias): merging length-extension LoRA with style LoRA can subtly change the style, possibly a bias from the relatively small length-extension LoRA training data, which more comprehensive datasets could mitigate. (source: §5.3)
Reasonable next steps / alternative directions in light of the limitations (analysis)
Making the decay (energy→density) rule “learnable”: the regression fit ($R^2 > 0.985$) supporting exponential decay is strong observational evidence (source: Fig.9), but the truly optimal mask can vary by layer/head/data. (source: Fig.9)
- As a next step, learn $\alpha,\beta$ (temporal/spatial decay rates) or band density per layer/head, or use a piecewise (near=dense, far=sparse) mixed schedule, to reduce the risk of a “fixed rule”. (analysis)
Combining 2D/multiresolution approaches to remove the resolution-axis bottleneck: as in the authors’ limitation, breaking the quadratic in resolution (source: §6) needs more than the temporal-axis $O(n\log n)$. (source: §6)
- Natural alternatives: (i) hierarchically merge spatial tokens (multiscale), or (ii) factorize: local/window sparse in space + radial in time, aiming for “subquadratic in both time and space”. (analysis)
Stability strategy when applied at pre-training: the authors leave pre-training application open but unexplored. (source: §6)
- A reasonable next step is a gradual dense→radial curriculum (dense at first, sparsity increasing later) or starting with only some layers/heads radial and checking training stability. (analysis)
Stronger evaluation of long-range semantic dependency: Radial’s design spends “less budget on farther frames” (source: §4.2), so it can be vulnerable to “distant but important” interactions such as scene changes/reappearances. (source: §4.2)
- As a next step, it is reasonable to examine failure-mode breakdowns on prompts/benchmarks where long-range dependency is central (e.g., recurring objects, long-term goal maintenance) and sink alternatives (e.g., learned memory tokens). (analysis)
Better data and regularization for LoRA merging (style/length): since the authors attribute style bias to data scale (source: §5.3), beyond larger data, follow-ups such as regularization at merge time / adapter routing to reduce mutual interference are reasonable. (source: §5.3)
![[Paper Review] Radial Attention: O(n log n) Sparse Attention with Energy Decay for Long Video Generation](https://moonlight-paper-snapshot.s3.ap-northeast-2.amazonaws.com/arxiv/radial-attention-onlog-n-sparse-attention-with-energy-decay-for-long-video-generation-2.png)
Comments