arXiv:2605.23892cs.CVcs.AI2026-05

通过精简注意力计算中的关键帧,大幅提升视觉几何变换器的效率。

Good Token Hunting: A Hitchhiker's Guide to Token Selection for Visual Geometry Transformers

论文配图:Good Token Hunting: A Hitchhiker's Guide to Token Selection for Visual Geometry Transformers
图 1 · 摘自论文原文
  • 分两阶段筛选关键帧与帧内冗余令牌,减少计算量。
  • 500张图像场景下提速超85%,精度不降反升。
  • 适合追求高效3D重建的科研与工程人员。

视觉几何变换器在多视角3D重建中表现强大,可一次性预测多个3D属性。但其计算开销随输入序列长度呈平方增长,受限于全局注意力层,影响扩展性与效率。本文提出一种简单通用的策略:限制每个查询在全局注意力中交互的关键/值令牌数量。设计两阶段框架:首先在帧级进行跨帧选择,保留有代表性的帧;其次在帧内进一步剔除冗余令牌。分析表明,跨帧选择采用多样性策略可实现场景广泛覆盖;帧内选择需分层感知,由全局注意力模式的熵引导。实验显示,该方法在500张图像场景下加速超85%,且性能保持或提升,凸显其在未来视觉几何变换器应用中的关键潜力。

原文摘要 · Abstract (English)

Visual geometry transformers have become powerful architectures for multi-view 3D reconstruction, enabling joint prediction of multiple 3D attributes in a feed-forward manner. However, their computational cost grows quadratically with the input sequence length due to the global attention layers inside these models. This limits both their scalability and efficiency. In this work, we address this challenge with a simple yet general strategy: restricting the number of key/value tokens that each query interacts with during global attention. To achieve effective token selection, we introduce a two-stage framework. First, an inter-frame selection step operates at the frame level to identify frames that should be preserved. Second, an intra-frame selection step further discards more redundant tokens within the selected frames. Our analysis highlights the advantage of a diversity-based strategy for inter-frame selection, which ensures broad coverage of the scene. For intra-frame selection, we show that layer-aware sparsification is necessary, with the selection process guided by the entropy of the global attention pattern. Our approach offers a superior speed-accuracy trade-off compared to existing solutions. Extensive experiments show that it accelerates visual geometry transformers by over 85% for scenes with 500 images while maintaining, or even improving, baseline performance, which hints that how our token selection strategy can play a crucial role in future applications of visual geometry transformers. Our project website is available at https://zsh2000.github.io/good-token-hunting.github.io.

视觉几何注意力机制3D重建稀疏化

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