小模型能指导大模型,跨模型协同提升测试时自适应性能
When Small Guides Large: Cross-Model Co-Learning for Test-Time Adaptation
- 多个模型在线协作,互相补充知识减少偏差
- 小模型指导大模型,使ViT-Base在ImageNet-C上准确率从51.7%升至64.5%
- 可插拔模块,适用于ResNet、ViT、Mobile-ViT等各类模型
测试时自适应(TTA)通过在线无监督学习将模型适配到测试域数据,表现优异。但现有方法多聚焦单模型适配。本文探究跨模型知识对TTA的影响,发现即使模型规模差异显著(如10.6M参数的MobileViT与86.6M参数的ViT-Base),小模型也能提供互补且可靠的指导信息。为此提出COCA框架,包含两项核心策略:1)协同适应,动态融合其他模型的互补知识以降低个体偏差;2)自适应,通过无监督学习强化各模型独特优势,实现多样化域适应。大量实验表明,COCA作为即插即用模块,显著提升多种模型的最先进性能——例如,在MobileViT引导下,ViT-Base在ImageNet-C上的平均适应准确率从51.7%提升至64.5%。代码已公开于https://github.com/ycarobot/COCA。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。