When Small Guides Large: Cross-Model Co-Learning for Test-Time Adaptation

TL;DR

This paper introduces COCA, a cross-model co-learning framework for test-time adaptation that leverages complementary knowledge between models of different sizes to improve domain adaptation performance.

Contribution

It proposes a novel cross-model co-learning approach for TTA, enabling models of varying sizes to mutually enhance each other's adaptation capabilities.

Findings

Small models can effectively guide larger models in TTA.

COCA significantly improves adaptation accuracy across various model architectures.

Cross-model co-learning boosts performance on domain shift benchmarks.

Abstract

Test-time Adaptation (TTA) adapts a given model to testing domain data with potential domain shifts through online unsupervised learning, yielding impressive performance. However, to date, existing TTA methods primarily focus on single-model adaptation. In this work, we investigate an intriguing question: how does cross-model knowledge influence the TTA process? Our findings reveal that, in TTA's unsupervised online setting, each model can provide complementary, confident knowledge to the others, even when there are substantial differences in model size. For instance, a smaller model like MobileViT (10.6M parameters) can effectively guide a larger model like ViT-Base (86.6M parameters). In light of this, we propose COCA, a Cross-Model Co-Learning framework for TTA, which mainly consists of two main strategies. 1) Co-adaptation adaptively integrates complementary knowledge from other models throughout the TTA process, reducing individual model biases. 2) Self-adaptation enhances each model's unique strengths via unsupervised learning, enabling diverse adaptation to the target domain. Extensive experiments show that COCA, which can also serve as a plug-and-play module, significantly boosts existing SOTAs, on models with various sizes--including ResNets, ViTs, and Mobile-ViTs--via cross-model co-learned TTA. For example, with Mobile-ViT's guidance, COCA raises ViT-Base's average adaptation accuracy on ImageNet-C from 51.7% to 64.5%. The code is publicly available at https://github.com/ycarobot/COCA.