用原型动态聚合测试样本知识,让视觉语言模型更快更准适应新数据。
Prototype-Based Test-Time Adaptation of Vision-Language Models
- 用类别专属原型替代缓存,通过样本置信度自适应融合视觉特征。
- 在15个图像识别和4个点云分析基准上达到顶尖性能,跨域准确率提升3.74%。
- 无需缓存存储与检索,推理速度保持92%,远超传统方法的50%效率。
测试时适配(TTA)已成为弥合视觉语言模型预训练与测试数据分布差异的有力范式。现有无反向传播的TTA方法依赖缓存设计,存在两大缺陷:一是缓存随类别增加导致推理延迟上升,大场景下效率低下;二是缓存样本不足或错误时性能不佳。本文提出原型驱动的测试时适配(PTA),利用一组类别专属知识原型,从测试样本中累积知识。具体地,每个测试样本的零样本分类置信度用于自适应加权其视觉特征并融入对应原型。关键在于,历史测试样本的知识仅存储于原型中,避免了缓存构建与查询开销,显著提升效率。PTA在15个图像识别与4个鲁棒点云分析基准上均达领先性能:例如,在10个跨域基准上将CLIP准确率从65.64%提升至69.38%,同时在大规模ImageNet-1K上保持92%的推理速度;而基于缓存的TDA仅达67.97%准确率,且速度降至50%。
原文摘要 · Abstract (English)
Test-time adaptation (TTA) has emerged as a promising paradigm for vision-language models (VLMs) to bridge the distribution gap between pre-training and test data. Recent works have focused on backpropagation-free TTA methods that rely on cache-based designs, but these introduce two key limitations. First, inference latency increases as the cache grows with the number of classes, leading to inefficiencies in large-scale settings. Second, suboptimal performance occurs when the cache contains insufficient or incorrect samples. In this paper, we present Prototype-Based Test-Time Adaptation (PTA), an efficient and effective TTA paradigm that uses a set of class-specific knowledge prototypes to accumulate knowledge from test samples. Particularly, knowledge prototypes are adaptively weighted based on the zero-shot class confidence of each test sample, incorporating the sample's visual features into the corresponding class-specific prototype. It is worth highlighting that the knowledge from past test samples is integrated and utilized solely in the prototypes, eliminating the overhead of cache population and retrieval that hinders the efficiency of existing TTA methods. This endows PTA with extremely high efficiency while achieving state-of-the-art performance on 15 image recognition benchmarks and 4 robust point cloud analysis benchmarks. For example, PTA improves CLIP's accuracy from 65.64% to 69.38% on 10 cross-domain benchmarks, while retaining 92% of CLIP's inference speed on large-scale ImageNet-1K. In contrast, the cache-based TDA achieves a lower accuracy of 67.97% and operates at only 50% of CLIP's inference speed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。