AdaMerge通过关注重要特征,让视觉Transformer在不训练的情况下更高效压缩。
AdaMerge: Salience-Aware Adaptive Token Merging for Training-Free Acceleration of Vision Transformers

- 根据注意力重要性加权匹配相似度,保护关键信息
- 按输入内容动态调整每层压缩量,减少冗余损失
- 在相同算力下比现有方法准确率更高,适合部署加速
视觉Transformer(ViT)中自注意力的二次复杂度是实际部署的主要瓶颈,推动了大量无训练的令牌压缩研究。其中,令牌合并(ToMe)作为一种简洁的解决方案脱颖而出,但其设计基于令牌等价的隐含假设,违背了自注意力非均匀性的已知规律,导致在高强度压缩下高显著性令牌的信息丢失。为此,我们提出AdaMerge,一个基于双重互补机制的令牌合并框架:第一,显著性加权相似度利用列向特征亲和中心性作为令牌重要性代理,并将显著性得分融入二分匹配评分,确保关键令牌对合并表示贡献更大;第二,自适应合并强度利用预计算的逐层相似性统计,根据输入特定冗余动态调节每层的缩减数量。在ImageNet-1k上使用ViT-B/16,AdaMerge在所有与FLOPs匹配的场景中持续优于ToMe、PiToMe和DSM。随着压缩加剧,性能差距单调扩大:在13.4G FLOPs时,AdaMerge仅造成-1.06%的Top-1准确率下降,而PiToMe为-1.45%,DSM为-4.62%。据我们所知,AdaMerge是首个将显著性加权相似度与自适应逐层缩减结合的无训练令牌合并框架,显著提升了ViT加速的精度-FLOPs帕累托前沿。
原文摘要 · Abstract (English)
The quadratic cost of self-attention in Vision Transformers (ViTs) constitutes a fundamental bottleneck for practical deployment, motivating a vibrant line of research on token reduction. Among existing approaches, token merging (ToMe) has emerged as an elegant training-free solution; yet its design rests on an unspoken premise of token equality, which contravenes the well-documented non-uniformity of self-attention and leads to information loss in high-salience tokens under aggressive compression. We address this limitation with AdaMerge, a token-merging framework based on two complementary mechanisms. First, salience-weighted similarity leverages column-wise feature-affinity centrality as a token-importance proxy and incorporates the resulting salience scores into the bipartite matching score, ensuring that pivotal tokens contribute more strongly to the merged representation. Second, adaptive merging intensity uses pre-computed layer-wise similarity statistics to dynamically modulate the per-layer reduction count in accordance with input-specific redundancy. On ImageNet-1k with ViT-B/16, AdaMerge consistently outperforms ToMe, PiToMe, and DSM across all FLOPs-matched regimes. The accuracy gap widens monotonically with compression: at the 13.4G FLOPs operating point, AdaMerge sustains a Top-1 degradation of only -1.06%, compared to -1.45% for PiToMe and -4.62% for DSM. To our knowledge, AdaMerge is the first to combine salience-weighted similarity and adaptive per-layer reduction into a single training-free token merging framework, advancing the accuracy-FLOPs Pareto frontier of ViT acceleration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。