arXiv:2410.14729cs.CVcs.AI2024-10ICCV被引 8

用精简视觉令牌提升大模型跨数据集表现,零训练低成本高效

Is Less More? Exploring Token Condensation as Training-free Test-time Adaptation

  • 通过保留关键视觉令牌并修正输出,实现无训练测试时适配
  • 在跨数据集任务上最高提升21.4%,计算量降低12.2%~48.9%
  • 适用于需快速适配新场景的视觉语言模型部署

对比语言-图像预训练(CLIP)在通用图像表征学习上表现优异,但在某些下游数据集的零样本推理中仍存在性能瓶颈。测试时适配(TTA)虽能缓解此问题,但通常需要大规模批处理和复杂增强,导致高计算开销。本文探索是否可通过高效、无需训练的方法改善视觉语言模型(VLM)在特定测试场景下的表现。我们研究了原本用于提升视觉变换器效率的令牌压缩(TC)技术,发现信息丰富的令牌可增强CLIP等模型在未见数据集上的视觉-文本对齐能力。然而,现有TC方法在减少令牌数时往往损害分布内性能。为此,我们提出令牌压缩作为适配(TCA),一种无需训练的适配方法:不被动丢弃令牌,而是引入基于水库的域锚定令牌,实现信息保留的令牌压缩与输出对齐修正。TCA在跨数据集基准和CIFAR-100-Corrupted数据集上,相比最强基线最高提升21.4%,同时将GFLOPs降低12.2%至48.9%,且对CLIP和SigLIP系列模型均表现出极低超参数依赖性。

原文摘要 · Abstract (English)

Contrastive Language-Image Pretraining (CLIP) excels at learning generalizable image representations but often falls short in zero-shot inference on certain downstream datasets. Test-time adaptation (TTA) mitigates this issue by adjusting components like normalization layers or context prompts, yet it typically requires large batch sizes and extensive augmentations, leading to high computational costs. This raises a key question: Can VLMs' performance drop in specific test cases be mitigated through efficient, training-free approaches? To explore the solution, we investigate token condensation (TC) techniques, originally designed to enhance vision transformer efficiency by refining token usage during inference. We observe that informative tokens improve visual-text alignment in VLMs like CLIP on unseen datasets. However, existing TC methods often fail to maintain in-distribution performance when reducing tokens, prompting us to ask: How can we transform TC into an effective ``free-lunch'' adaptation strategy for VLMs? To address this, we propose Token Condensation as Adaptation (TCA), a training-free adaptation method that takes a step beyond standard TC. Rather than passively discarding tokens, TCA condenses token representation by introducing reservoir-based domain anchor tokens for information-preserving token reduction and logits correction. TCA achieves up to a 21.4% performance improvement over the strongest baseline on cross-dataset benchmark and the CIFAR-100-Corrupted dataset while reducing GFLOPs by 12.2% to 48.9%, with minimal hyperparameter dependency on both CLIP and SigLIP series.

视觉语言模型测试时适配令牌压缩零训练

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