Enhancing Test Time Adaptation with Few-shot Guidance

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Luo, Siqi, Xin, Yi, Du, Yuntao, Tan, Tao, Zhai, Guangtao, Liu, Xiaohong
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911305529556992
author Luo, Siqi
Xin, Yi
Du, Yuntao
Tan, Tao
Zhai, Guangtao
Liu, Xiaohong
author_facet Luo, Siqi
Xin, Yi
Du, Yuntao
Tan, Tao
Zhai, Guangtao
Liu, Xiaohong
contents Deep neural networks often encounter significant performance drops while facing with domain shifts between training (source) and test (target) data. To address this issue, Test Time Adaptation (TTA) methods have been proposed to adapt pre-trained source model to handle out-of-distribution streaming target data. Although these methods offer some relief, they lack a reliable mechanism for domain shift correction, which can often be erratic in real-world applications. In response, we develop Few-Shot Test Time Adaptation (FS-TTA), a novel and practical setting that utilizes a few-shot support set on top of TTA. Adhering to the principle of few inputs, big gains, FS-TTA reduces blind exploration in unseen target domains. Furthermore, we propose a two-stage framework to tackle FS-TTA, including (i) fine-tuning the pre-trained source model with few-shot support set, along with using feature diversity augmentation module to avoid overfitting, (ii) implementing test time adaptation based on prototype memory bank guidance to produce high quality pseudo-label for model adaptation. Through extensive experiments on three cross-domain classification benchmarks, we demonstrate the superior performance and reliability of our FS-TTA and framework.
format Preprint
id arxiv_https___arxiv_org_abs_2409_01341
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Enhancing Test Time Adaptation with Few-shot Guidance
Luo, Siqi
Xin, Yi
Du, Yuntao
Tan, Tao
Zhai, Guangtao
Liu, Xiaohong
Computer Vision and Pattern Recognition
Deep neural networks often encounter significant performance drops while facing with domain shifts between training (source) and test (target) data. To address this issue, Test Time Adaptation (TTA) methods have been proposed to adapt pre-trained source model to handle out-of-distribution streaming target data. Although these methods offer some relief, they lack a reliable mechanism for domain shift correction, which can often be erratic in real-world applications. In response, we develop Few-Shot Test Time Adaptation (FS-TTA), a novel and practical setting that utilizes a few-shot support set on top of TTA. Adhering to the principle of few inputs, big gains, FS-TTA reduces blind exploration in unseen target domains. Furthermore, we propose a two-stage framework to tackle FS-TTA, including (i) fine-tuning the pre-trained source model with few-shot support set, along with using feature diversity augmentation module to avoid overfitting, (ii) implementing test time adaptation based on prototype memory bank guidance to produce high quality pseudo-label for model adaptation. Through extensive experiments on three cross-domain classification benchmarks, we demonstrate the superior performance and reliability of our FS-TTA and framework.
title Enhancing Test Time Adaptation with Few-shot Guidance
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.01341