← 返回 2026-08-24

划分支撑集,重构残差:面向视频生成与世界模型的免训练稀疏注意力 Partition the Support, Reconstruct the Residual: Training-Free Sparse Attention for Video Generation and World Models

Pardis Taghavi, Reza Langari, Gaurav Pandey 📅 2026-08-19 👍 9 2026-08-29 18:30
世界模型 免训练方法 推理加速 稀疏注意力 视频生成

响应耦合分区+探针残差重构,22–26%密度下免训练稀疏注意力加速视频生成1.48–2.61倍

前置知识

块稀疏注意力 (Block-Sparse Attention)

把 query 与 key/value token 各自分组,注意力只对选中的\u201c组对\u201d(cell)精确计算,其余交互被跳过。分区块化让稀疏模式能映射到 GPU 友好的连续内存布局,配合 FlashInfer 等变长块内核获得真实加速。路由密度 $\rho_{\text{route}}$ 按真实代价 $|G_a^Q||G_b^{KV}|$ 加权统计被计算的 query-key pair 占比。

本文把可执行稀疏注意力拆解为分区/路由/跳过处理三个决策来分析,全部方法与消融都建立在这个算子框架上。

归一化硬丢弃 (Renormalized Hard Drop)

稀疏注意力对被跳过的 key 不做补偿,只把 softmax 权重在保留支撑集 $S_i$ 上重新归一化:$O_i^{\text{sp}}=\text{softmax}(q_iK_{S_i}^\top/\sqrt{d})V_{S_i}$。其相对稠密输出的残差满足恒等式 $R_i=p_{U,i}(O_{U,i}-O_i^{\text{sp}})$,同时依赖被丢弃质量 $p_{U,i}$ 与被丢/被留两个支撑诱导输出之差。

这是观察 O2 的数学基础,也是探针拟合残差重构所瞄准的目标对象——重构的正是这个 post-softmax 残差。

谱嵌入与 k-means 聚类

k-means 在欧氏空间按距离把向量分成若干组。谱嵌入先求度量矩阵 $M$ 的前 $r$ 个特征向量(特征值归一化到单位均值后按开方加权)得到投影 $F$,使原度量下的距离近似等于低维欧氏距离。文中 $M_K=Q_s^\top Q_s/M_s$ 的秩-$r$ 谱嵌入把\u201c两个 key 在采样 query 上响应相似\u201d变成可直接 k-means 的几何关系。

响应耦合分区的两步分组(K/V 组与 query 组)都靠“构造响应度量+谱嵌入+k-means”实现,是方法的计算骨架。

加权岭回归与子空间投影

岭回归在最小二乘上加 $\ell_2$ 惩罚,闭式解为 $(X^\top WX+\lambda I)^{-1}X^\top WY$,用于小样本稳定拟合;子空间投影取残差矩阵的右奇异向量 $\Psi_r$,把预测限制在 $\Psi_r\Psi_r^\top$ 张成的实测方向上,防止向未见过的输出方向外推。

探针拟合残差重构的估计器正是“覆盖加权岭回归 + 探针残差 SVD 子空间约束”,看不懂这两点就看不懂方法第 4.2 节。

视频扩散 Transformer 与世界模型

视频 DiT(HunyuanVideo、Wan2.2 等)在潜空间用 3D 全注意力对数万级时空 token 去噪生成视频(如 HunyuanVideo 每头 $N_q=118{,}800$);世界模型(NVIDIA Cosmos 系)则按文本/图像条件预测物理世界未来帧,注意力结构因条件模态不同而异。

论文在四类异构模型上验证通用性,必须理解它们序列长度与注意力集中模式的差异才能读懂观察 O1–O3 的模型间对比。

质量指标 PSNR/SSIM/LPIPS 与 VBench/PBench

PSNR 衡量像素级误差(dB,越高越好);SSIM 衡量结构相似性(0–1,越高越好);LPIPS 用深度特征衡量感知差异(越低越好)。VBench/VBench++ 的 ImgQual 与 SubCons 评生成语义质量,NVIDIA PBench Quality 专评物理世界合理性。

表 1 与各消融的结论全部由这些指标定义,不清楚方向性就无法判断“谁更好”。

研究动机

视频生成模型(HunyuanVideo、Wan2.2)与物理世界模型(NVIDIA Cosmos 系)在高分辨率、长时长下要处理数万级时空 token,自注意力的二次复杂度成为推理瓶颈。预训练视频 DiT 的注意力呈结构化、头相关的集中分布,免训练稀疏注意力因此成为主流加速路线,代表工作有 SVG2、SpargeAttn、SVG-EAR、PISA。但作者指出\u201c行级注意力集中\u201d不足以定义可执行的稀疏算子。观察 O1:共享路由会放大支撑需求——Wan2.2 单 query 保留 90% 注意力质量只需中位 6.2% 的 key,8 个 query 共享一条路由时涨到 22.9%;Cosmos-Predict2.5 单行要 56.5%、池化后 77.7%,行级稀疏度无法预测组级密度。观察 O2:保留质量不决定 post-softmax 误差——由 $R_i=p_{U,i}(O_{U,i}-O_i^{\text{sp}})$ 可知残差还依赖两个支撑诱导输出的差,Cosmos3-Nano 在相近保留质量下误差可差一个量级。现有方法默认两者由保留质量决定,导致密度失控或误差不可控。

本文的目标是本文的目标是把免训练块稀疏注意力当作一个完整的可执行算子来系统设计,而不是只回答\u201c选哪些块\u201d。具体包括四点:(1) 显式分离 query 分区、配对 K/V 分区、路由策略与被跳过交互处理四个决策,并量化它们之间的结构性依赖(O1–O3);(2) 构造让共享路由的 query 组具有真实重叠支撑、且残差结构便于后续修正的分区几何;(3) 对跳过交互造成的 post-softmax 残差做显式、逐注意力调用的统计重构,而非简单忽略或块级近似;(4) 在四个异构模型(HunyuanVideo-13B 文生视频、Wan2.2-I2V-A14B 图生视频、Cosmos-Predict2.5-14B 与 Cosmos3-Nano-16B 物理世界生成)上,用把探针行计入的精确 pair 核算与把全部在线开销计入的端到端计时验证:在 22%–26% 实际执行对密度下逼近稠密基准质量,同时取得 1.48×–2.61× 的真实加速。

与已有工作不同的是,已有工作的切入要么是静态结构先验(STA、SVG、Radial Attention 利用时空/层/头结构),要么是在线块重要性估计(SpargeAttn、AdaSpa、XAttention),要么是离线拓扑先验(HASTE、ScalingAttention);补偿被跳过交互则多用块级近似或跨时间步统计复用(Re-ttention、SVG-EAR 的 K/V 质心补偿、PISA 分块泰勒展开、Sol-Attn 代理分数)。聚类类方法中,SVG2 按 embedding 语义邻近聚类,SVOO 交替执行 query 感知的 key 聚类与反向聚类,AdaCluster 对 q/k 用不同相似度准则。SparsePR 的独特之处有二:其一,分区几何不基于激活邻近而基于当前调用的响应几何——key 的相似度定义为在采样 query 下的 logit 轮廓相似度,query 分组由 K/V 组质心一次性单向诱导、不做交替细化;其二,首次用仿射可表示性分解(式 5)证明分区选择会改变残差能被稀疏输出仿射解释的比例(+3.2–14.9 个百分点),把分区设计与残差重构耦合进统一框架。

核心方法

可执行稀疏算子要回答:谁与谁共享路由、哪些 key 被一起选中、算哪些 cell、没算的怎么办。SparsePR 用两个在线免训练组件作答。响应耦合分区:采样 $M_s$ 个 query 行,构造 key 响应度量 $M_K=\frac{1}{M_s}Q_s^\top Q_s$,该度量下 key 距离即 logit 轮廓差异;取前 $r_K=48$ 个特征方向得 $F_K$,key 坐标 $\phi_K(k_j)=F_K^\top k_j$ 经 k-means 得配对 K/V 组(value 随 key)。再用各非空组质心构造 query 响应度量,取 $r_Q=64$ 个方向得 $F_Q$,query 坐标 RMS 归一化后聚类成共享路由组。探针拟合残差重构:每 query 头取 $M=64$ 个分层探针行做精确注意力,拟合 $R_i\approx b+x_iB$($x_i=O_i^{\text{sp}}$),并把修正限制在探针残差的前 $r=16$ 个右奇异方向上:$\hat R_i=\mu_R+\bar x_iB_\lambda\Psi_r\Psi_r^\top$。

核心创新之一是一次性的非对称响应耦合:不同于 SVOO 的交替迭代聚类、SVG2 的 embedding 语义聚类或 AdaCluster 的双准则聚类,SparsePR 只做一次单向耦合——采样 query 定义 key 的响应坐标用于 K/V 分组,K/V 组质心再诱导 query 的响应坐标用于共享路由,全程只依赖当前注意力调用,无离线训练、无交替细化。核心创新之二是把残差处理从\u201c块级近似\u201d升级为\u201c行级统计学习\u201d:用 $M=64$ 个精确探针行在每个注意力调用内拟合从稀疏输出到 post-softmax 残差的仿射映射。其理论根基是观察 O3:在匹配的稀疏执行下,响应耦合分区使残差的仿射可解释能量 $\|P_{X_g}R_g\|_F^2/\|R_g\|_F^2$ 提升 3.2–14.9 个百分点、把仿射正交能量降到语义分区的 0.285×–0.653×——即分区几何本身决定了\u201c残差能否被简单修正\u201d。与已有方法的本质区别在于:分区不再是独立的预处理步骤,而是为残差可重构性服务的耦合设计。

方法步骤详情

(1) K/V 分组:采样 $M_s$ 个 query 行,由响应度量谱嵌入得 key 坐标 $\phi_K$($r_K=48$),k-means 得配对 K/V 组,value 随 key。(2) query 分组:对各非空组求质心、减均值堆成 $\tilde K$,构造 query 响应度量取 $r_Q=64$ 个特征方向得 $F_Q$,query 坐标 RMS 归一化后 k-means 得 $G^Q$。(3) 路由与执行:按 cell 代价 $|G_a^Q||G_b^{KV}|$ 的 pair 预算选 cell,$Q,K,V$ 重排为组主序布局,交 FlashInfer 变长块稀疏核精确计算,每行重归一化(硬丢弃),输出恢复原序。(4) 探针与拟合:每 query 头按组分层、组内按到质心距离径向排序轮转选 $M=64$ 个探针行(算法 A.1),精确计算得残差 $R_p$;按覆盖权重 $w_p=|G_a^Q|/m_a$ 加权,解岭回归得 $B_\lambda$($\lambda=0.1$),对探针残差 SVD 取 $r=16$ 个右奇异向量 $\Psi_r$。(5) 输出:未探针行 $\hat O_i=O_i^{\text{sp}}+\hat R_i$,探针行用稠密输出;$\rho_{\text{exec}}=\rho_{\text{route}}+M/N_q$,探针开销 0.054%–0.145%。

技术新颖性

新颖性有四层。(1) 问题刻画:O1/O2/O3 首次量化了共享路由的支撑膨胀(Wan2.2 的 6.2%→22.9%)、保留质量与 post-softmax 误差的脱钩、以及分区几何对残差仿射可表示性的决定作用(可解释能量 +3.2–14.9 pp,正交能量 0.285×–0.653×)。(2) 度量构造:key 响应度量 $M_K$ 的秩-$r$ 谱嵌入等价于对响应轮廓的归一化谱聚类,query 坐标的 RMS 归一化让 k-means 只比响应方向不比尺度,是有几何意义的构造而非启发式。(3) 估计器设计:修正被限制在实测探针残差的 SVD 子空间内($\Psi_r\Psi_r^\top$),不假设残差全局低秩、不做范数截断,是对小样本回归外推风险的结构化控制。(4) 评测口径:探针行计入执行密度 $\rho_{\text{exec}}$、全部在线开销计入端到端计时,避免\u201c算子快但路由贵\u201d的口径套利。相比 SVG-EAR 的误差感知路由与 PISA/Sol-Attn 的泰勒/代理近似,SparsePR 的修正逐调用校准于输出子空间。

Overview of SparsePR. Response-Coupled Partitioning builds executable K/V and query groups, and Probe-Fitted Residual Reconstruction uses exact probe rows to correct the sparse output.
Figure 2: Overview of SparsePR. Response-Coupled Partitioning builds executable K/V and query groups, and Probe-Fitted Residual Reconstruction uses exact probe rows to correct the sparse output.

实验结果

核心结果见表 1(单卡 H100、BF16、720p)。HunyuanVideo-13B:PSNR 31.844、LPIPS 0.087,语义指标与稠密持平(0.850/0.976);密度 21.92%,端到端 2.61×,超过 SVG-EAR(31.043,1.93×)与 SVG2(30.452,2.30×)。Wan2.2-I2V:PSNR 30.658、LPIPS 0.044,密度 21.97%,1.80×。Cosmos-Predict2.5-14B:PSNR 26.328,PBench 77.75 对稠密 77.76 几乎无损,1.51×(SVG2 仅 1.24× 且 PSNR 20.075)。Cosmos3-Nano-16B:PSNR 24.417 对 SVG2 的 22.458(+1.96 dB),密度 25.96%、1.48×。消融(表 2)显示探针拟合贡献主要下降:HunyuanVideo 上语义+硬丢弃 0.0887/0.7136 → 语义+探针 0.0527/0.3562 → SparsePR 0.0330/0.2285;响应耦合再把 Cosmos-Predict2.5 从 0.2622/0.8260 降到 0.0954/0.5769。密度扫描(表 3,12%–35%)全档最优。运行时(图 4b):Wan2.2 全生成 1650s→917s(1.80×),探针拟合仅占 1.1%。

Quality–efficiency comparison across content-generation and physical video world models.
Table 1: Quality–efficiency comparison across content-generation and physical video world models.
Partition and Reconstruction ablation at matched routed-pair density. Entries report mean/p99 normalized attention-output error; lower is better.
Table 2: Partition and Reconstruction ablation at matched routed-pair density. Entries report mean/p99 normalized attention-output error; lower is better.
Cosmos3-Nano sensitivity to total executed-pair density ρ. Entries report mean/p99.
Table 3: Cosmos3-Nano sensitivity to total executed-pair density ρ. Entries report mean/p99.
Probe-selection and estimator ablations with the response partition, M = 64, r = 16.
Table 4: Probe-selection and estimator ablations with the response partition, M = 64, r = 16.
Qualitative Results. Cosmos2.5-14B model.
Figure 1: Qualitative Results. Cosmos2.5-14B model.
Partitioning and runtime analysis. (a) Error reduction from response-coupled vs. semantic partitioning at 22% density. (b) Wan2.2 full generation latency for Dense, SVG2, and SparsePR.
Figure 4: Partitioning and runtime analysis. (a) Error reduction from response-coupled vs. semantic partitioning at 22% density. (b) Wan2.2 full generation latency for Dense, SVG2, and SparsePR.
查看结构化数据
任务指标本文基线提升
HunyuanVideo-13B 720p 文生视频 PSNR (dB) 31.844(密度 21.92%) SVG-EAR 31.043 / SVG2 30.452 / SpargeAttn 24.589 较最强基线 +0.80 dB,且加速更高
HunyuanVideo-13B 720p 文生视频 端到端加速 2.61×(255.95 PFLOPs) SVG2 2.30× / SVG-EAR 1.93× / SpargeAttn 1.38× 较 SVG2 提升 13.5%
Wan2.2-I2V-A14B 720p 图生视频 LPIPS(越低越好) 0.044 SVG-EAR 0.093 / SVOO 0.095 感知误差减半以上(-52.7%)
Wan2.2-I2V-A14B 720p 图生视频 端到端加速 1.80×(328.70 PFLOPs) SVOO 1.61× / SVG-EAR 1.61× / SVG2 1.59× +11.8%(延迟 1650s→917s)
Cosmos-Predict2.5-14B 图生世界 PSNR (dB) 26.328 SVG-EAR 25.549 / SVOO 22.066 / SVG2 20.075 +0.78 dB,密度却是最低的 22.14%
Cosmos-Predict2.5-14B 图生世界 PBench Quality 77.75 稠密 77.76 几乎无损(-0.01),SubCons 0.976 持平
Cosmos3-Nano-16B 图生世界 PSNR (dB) 24.417 SVG2 22.458 / SVG-EAR 21.167 / SVOO 16.642 +1.96 dB,PBench 77.30 对稠密 77.31
Cosmos3-Nano-16B 单元误差消融 mean/p99 归一化注意力输出误差(ρ=22%) SparsePR 0.0822/0.4951 语义分区+硬丢弃 0.3590/3.3557 mean 降 77%、p99 降 85%

局限与改进

作者承认的边界:单元级注意力误差仍可观(Cosmos-Predict2.5 在 22% 密度下 mean 0.0954、p99 0.5769);方法依赖\u201cpost-softmax 残差可被稀疏输出的仿射函数局部近似\u201d这一经验事实,附录 A.1 明确这是逐调用的经验局部模型而非结构性保证;分区、探针与拟合全部逐调用在线进行,端到端收益取决于这些开销被稀疏收益覆盖的程度;评测限定 720p、固定步数与引导设置。我的观察:(1) 探针预算 $M=64$、秩 $r=16$、$\lambda=0.1$ 在不同分辨率与序列长度下未做敏感性分析;(2) 收益分布不均——HunyuanVideo 本身两层都稀疏、分区收益小,Cosmos 系收益大,存在模型依赖;(3) SVOO 未报告密度与 PFLOPs,横向比较部分依赖作者复现协议;(4) 每次调用重算分区与拟合,对蒸馏少步数模型或自回归 KV 缓存场景是否划算未验证;(5) 引言提到 conditioning 差异会诱导不同注意力结构,但正文未系统量化。

独立分析的弱点

弱点一:全在线、逐调用重算。分区、k-means、探针拟合与 SVD 每次调用都重来,虽测得探针拟合仅占 1.1% 延迟,但在蒸馏少步数模型(4–8 步采样)上稀疏收益缩水、在线开销占比上升。改进方向:相邻去噪步共享分区或热启动质心做增量更新,配响应几何漂移检测。弱点二:探针预算固定且不按需分配。$M=64$ 对所有模型、所有层固定,而 O2 显示 Cosmos3-Nano 残差长尾,误差大的组可能探针不足;可按组残差方差或组大小自适应分配。弱点三:cell 不可分导致实际密度偏离目标,22% 目标对应 21.92%–25.96% 实现;可允许部分组以池化 token 表示实现分数密度。弱点四:单卡单批评测,FlashInfer 变长块核在多批、多卡流水线下是否保持 1.5× 以上加速未验证,BF16 下 SVD 与岭回归的数值稳健性也只在 H100 上检验,移植到 FP8 存疑。弱点五:仿射模型是修正能力的上界,O3 中仍有正交能量未被解释,极端分布外 prompt 下残差可能欠拟合。

未来方向

作者方向:项目页承诺更多定性结果;附录 D 已把密度敏感性扩展到 12%–35% 全区间,建议继续刻画不同操作点下分区收益的稳健性。可延伸方向:(1) 把响应耦合分区推广到自回归视频世界模型的 KV 缓存压缩,与 Samuel et al. 的时序缓存压缩结合,需解决因果掩码下响应坐标的构造;(2) 把逐调用探针回归蒸馏为跨调用共享的轻量修正头(以层/头/步索引为条件),进一步压缩 1.1% 的在线开销;(3) 用 O3 的仿射可解释能量作为免训练的分区质量代理指标,在线自适应选择组数 $C_q$、$C_k$ 与密度目标,替代人工调参;(4) 系统量化 conditioning(文生视频/图生视频/物理预测)对注意力结构与稀疏收益的影响,为世界模型设计条件感知路由;(5) 与 FP8、FlashAttention-3 或 TensorRT 后端集成,检验谱嵌入与 SVD 在不同数值格式下的稳健性;(6) 把探针选择扩展为序贯主动采样,用更少探针达到同等拟合质量,并研究跨头共享探针的收益。

复现评估

开源情况:正文只给出 Project page 占位链接,未见官方代码仓库,算法需自行实现。可复现材料相当充分:表 C.1 完整给出全部超参($r_K=48$、$r_Q=64$、$M=64$、$r=16$、$\lambda=0.1$、目标密度 22%/26%),算法 A.1 给出探针选择伪代码,附录 A/B 推导完备(硬丢弃残差恒等式、加权岭回归闭式解、SVD 子空间投影),附录 C.1 还给出融合投影核延迟(1.6170ms→0.5064ms)与相对误差 3.68×10⁻⁴ 作正确性参照。评测依赖公开模型(HunyuanVideo、Wan2.2、Cosmos-Predict2.5、Cosmos3-Nano)与公开基准(VBench、VBench++、PBench)。算力:单张 H100(BF16)可复现计时实验,Wan2.2 单次全生成稠密 1650s、稀疏 917s,跑满 VBench 规模对比需数百 GPU 时;GPU k-means、融合核与 FlashInfer 集成需要工程能力。综合判断:算法路径清晰、无隐藏数据依赖,无官方代码前属中等偏高复现难度,开源后降为中等。