arXiv:2603.25336cs.CV2026-03中稿 · CVPR被引 2

针对视觉模型注意力头稀疏化不均导致精度下降的问题,提出按敏感度重分配计算资源的新方法。

HeSS: Head Sensitivity Score for Sparsity Redistribution in VGGT

  • 基于头敏感度评分(HeSS)量化各注意力头对稀疏化的敏感程度
  • 在高稀疏度下仍保持性能稳定,相比均匀稀疏化提升显著
  • 适合追求高效3D视觉模型推理的开发者和研究者

视觉几何基础变压器(VGGT)推动了3D视觉发展,但其全局注意力层存在二次计算开销,限制可扩展性。现有稀疏化加速方法常导致显著精度下降。我们假设这是由于注意力头在稀疏化敏感性上存在异质性,而现有方法对所有头采用统一稀疏模式所致。为此,我们提出两阶段稀疏化流程:第一阶段通过新指标“头敏感度评分”(HeSS),在小校准集上近似关于两个不同误差项的海森矩阵,量化每个头的敏感度;第二阶段在推理时执行HeSS引导稀疏化,利用预计算的HeSS将总注意力预算重新分配——对敏感头分配更密集关注,对鲁棒头分配更稀疏关注。实验表明,HeSS能有效捕捉头间敏感性差异,且全局注意力层中的注意力头确实表现出异质敏感性。大量实验进一步验证,在高稀疏度下本方法显著缓解性能退化,跨不同稀疏水平均具强鲁棒性。代码已公开于 https://github.com/libary753/HeSS。

原文摘要 · Abstract (English)

Visual Geometry Grounded Transformer (VGGT) has advanced 3D vision, yet its global attention layers suffer from quadratic computational costs that hinder scalability. Several sparsification-based acceleration techniques have been proposed to alleviate this issue, but they often suffer from substantial accuracy degradation. We hypothesize that the accuracy degradation stems from the heterogeneity in head-wise sparsification sensitivity, as the existing methods apply a uniform sparsity pattern across all heads. Motivated by this hypothesis, we present a two-stage sparsification pipeline that effectively quantifies and exploits headwise sparsification sensitivity. In the first stage, we measure head-wise sparsification sensitivity using a novel metric, the Head Sensitivity Score (HeSS), which approximates the Hessian with respect to two distinct error terms on a small calibration set. In the inference stage, we perform HeSS-Guided Sparsification, leveraging the pre-computed HeSS to reallocate the total attention budget-assigning denser attention to sensitive heads and sparser attention to more robust ones. We demonstrate that HeSS effectively captures head-wise sparsification sensitivity and empirically confirm that attention heads in the global attention layers exhibit heterogeneous sensitivity characteristics. Extensive experiments further show that our method effectively mitigates performance degradation under high sparsity, demonstrating strong robustness across varying sparsification levels. Code is available at https://github.com/libary753/HeSS.

稀疏化注意力机制3D视觉模型压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。