Test-time Alignment-Enhanced Adapter for Vision-Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tong, Baoshun, Song, Kaiyu, Lai, Hanjiang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929603673587712
author Tong, Baoshun
Song, Kaiyu
Lai, Hanjiang
author_facet Tong, Baoshun
Song, Kaiyu
Lai, Hanjiang
contents Test-time adaptation with pre-trained vision-language models (VLMs) has attracted increasing attention for tackling the issue of distribution shift during the test phase. While prior methods have shown effectiveness in addressing distribution shift by adjusting classification logits, they are not optimal due to keeping text features unchanged. To address this issue, we introduce a new approach called Test-time Alignment-Enhanced Adapter (TAEA), which trains an adapter with test samples to adjust text features during the test phase. We can enhance the text-to-image alignment prediction by utilizing an adapter to adapt text features. Furthermore, we also propose to adopt the negative cache from TDA as enhancement module, which further improves the performance of TAEA. Our approach outperforms the state-of-the-art TTA method of pre-trained VLMs by an average of 0.75% on the out-of-distribution benchmark and 2.5% on the cross-domain benchmark, with an acceptable training time. Code will be available at https://github.com/BaoshunWq/clip-TAEA.
format Preprint
id arxiv_https___arxiv_org_abs_2411_15735
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Test-time Alignment-Enhanced Adapter for Vision-Language Models
Tong, Baoshun
Song, Kaiyu
Lai, Hanjiang
Computer Vision and Pattern Recognition
Test-time adaptation with pre-trained vision-language models (VLMs) has attracted increasing attention for tackling the issue of distribution shift during the test phase. While prior methods have shown effectiveness in addressing distribution shift by adjusting classification logits, they are not optimal due to keeping text features unchanged. To address this issue, we introduce a new approach called Test-time Alignment-Enhanced Adapter (TAEA), which trains an adapter with test samples to adjust text features during the test phase. We can enhance the text-to-image alignment prediction by utilizing an adapter to adapt text features. Furthermore, we also propose to adopt the negative cache from TDA as enhancement module, which further improves the performance of TAEA. Our approach outperforms the state-of-the-art TTA method of pre-trained VLMs by an average of 0.75% on the out-of-distribution benchmark and 2.5% on the cross-domain benchmark, with an acceptable training time. Code will be available at https://github.com/BaoshunWq/clip-TAEA.
title Test-time Alignment-Enhanced Adapter for Vision-Language Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.15735