提升视觉Transformer测试时适应效率,降低推理延迟同时保持性能。
Neutralizing Token Aggregation via Information Augmentation for Efficient Test-Time Adaptation
- 通过增强[CLS] token信息缓解令牌聚合导致的损失
- 在多个分布外基准上性能提升超2.5%,推理延迟降20%以上
- 适合资源受限场景下的高效模型部署
测试时适配(TTA)已成为在无需额外训练数据的情况下适应视觉变换器(ViT)应对分布偏移的有效方法。然而,现有TTA方法通常带来显著计算开销,限制了其在资源受限的实际场景中的应用。为降低推理成本,即插即用的令牌聚合方法通过合并冗余令牌来减少总处理令牌数。尽管高效,但直接与现有TTA方法结合时会导致性能显著下降。本文将此问题形式化为高效测试时适配(ETTA),旨在保留TTA适应能力的同时减少推理延迟。我们从新的互信息视角进行理论分析,表明令牌聚合会引发信息丢失,且传统基于范数调整的TTA方法无法完全弥补。受此启发,提出NAVIA(Neutralizing Token Aggregation via Information Augmentation):直接增强[CLS] token嵌入,并在浅层ViT中引入自适应偏置。理论上证明,通过熵最小化优化这些增强可恢复因令牌聚合损失的信息。大量实验表明,NAVIA在多个分布外基准上显著优于最先进方法(提升超2.5%),同时实现超过20%的推理延迟降低,有效解决ETTA挑战。
原文摘要 · Abstract (English)
Test-Time Adaptation (TTA) has emerged as an effective solution for adapting Vision Transformers (ViT) to distribution shifts without additional training data. However, existing TTA methods often incur substantial computational overhead, limiting their applicability in resource-constrained real-world scenarios. To reduce inference cost, plug-and-play token aggregation methods merge redundant tokens in ViTs to reduce total processed tokens. Albeit efficient, it suffers from significant performance degradation when directly integrated with existing TTA methods. We formalize this problem as Efficient Test-Time Adaptation (ETTA), seeking to preserve the adaptation capability of TTA while reducing inference latency. In this paper, we first provide a theoretical analysis from a novel mutual information perspective, showing that token aggregation inherently leads to information loss, which cannot be fully mitigated by conventional norm-tuning-based TTA methods. Guided by this insight, we propose to \textbf{N}eutralize Token \textbf{A}ggregation \textbf{v}ia \textbf{I}nformation \textbf{A}ugmentation (\textbf{NAVIA}). Specifically, we directly augment the [CLS] token embedding and incorporate adaptive biases into the [CLS] token in shallow layers of ViTs. We theoretically demonstrate that these augmentations, when optimized via entropy minimization, recover the information lost due to token aggregation. Extensive experiments across various out-of-distribution benchmarks demonstrate that NAVIA significantly outperforms state-of-the-art methods by over 2.5\%, while achieving an inference latency reduction of more than 20\%, effectively addressing the ETTA challenge.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。