让视觉语言模型在测试时自动过滤噪声数据,提升鲁棒性。
Noisy Test-Time Adaptation in Vision-Language Models
- 分离分类器与噪声检测器,用冻结模型输出生成伪标签训练检测器。
- 在ImageNet上使零样本噪声测试适应的准确率提升8.32%,误报率降低9.40%。
- 无需重新训练分类器,计算高效,适合实际部署场景。
测试时自适应(TTA)旨在仅使用目标数据缓解源域与目标域之间的分布偏移。在开放世界场景中,模型常遇到非分布内(ID)标签范围外的噪声样本。利用预训练视觉语言模型(VLMs)的零样本能力,本文提出零样本噪声测试自适应(ZS-NTTA),在测试时零样本地处理含噪声的目标数据。我们发现现有TTA方法在ZS-NTTA下表现不佳,甚至落后于冻结模型。通过全面实验分析,揭示未过滤噪声数据的负面影响超过清洁数据的增益。同时,同时优化分类与噪声检测会损害两个任务。为此,我们提出解耦框架:保持分类器冻结,专注训练独立噪声检测器。技术上引入自适应噪声检测器(AdaND),利用冻结模型输出作为伪标签训练检测器,并在适应过程中注入高斯噪声以防止将清洁样本误判为噪声。此外,AdaND还能提升VLM的零样本分布外检测能力。实验表明,其在ZS-NTTA和ZS-OOD检测上均优于当前最优方法,在ImageNet上分别实现8.32%的调和平均准确率提升和9.40%的FPR95改进。该方法计算高效,接近冻结模型开销。代码已开源。
原文摘要 · Abstract (English)
Test-time adaptation (TTA) aims to address distribution shifts between source and target data by relying solely on target data during testing. In open-world scenarios, models often encounter noisy samples, i.e., samples outside the in-distribution (ID) label space. Leveraging the zero-shot capability of pre-trained vision-language models (VLMs), this paper introduces Zero-Shot Noisy TTA (ZS-NTTA), focusing on adapting the model to target data with noisy samples during test-time in a zero-shot manner. We find existing TTA methods underperform under ZS-NTTA, often lagging behind even the frozen model. We conduct comprehensive experiments to analyze this phenomenon, revealing that the negative impact of unfiltered noisy data outweighs the benefits of clean data during model updating. Also, adapting a classifier for ID classification and noise detection hampers both sub-tasks. Built on this, we propose a framework that decouples the classifier and detector, focusing on developing an individual detector while keeping the classifier frozen. Technically, we introduce the Adaptive Noise Detector (AdaND), which utilizes the frozen model's outputs as pseudo-labels to train a noise detector. To handle clean data streams, we further inject Gaussian noise during adaptation, preventing the detector from misclassifying clean samples as noisy. Beyond the ZS-NTTA, AdaND can also improve the zero-shot out-of-distribution (ZS-OOD) detection ability of VLMs. Experiments show that AdaND outperforms in both ZS-NTTA and ZS-OOD detection. On ImageNet, AdaND achieves a notable improvement of $8.32\%$ in harmonic mean accuracy ($\text{Acc}_\text{H}$) for ZS-NTTA and $9.40\%$ in FPR95 for ZS-OOD detection, compared to SOTA methods. Importantly, AdaND is computationally efficient and comparable to the model-frozen method. The code is publicly available at: https://github.com/tmlr-group/ZS-NTTA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。