通过调整视觉变压器的奇异值实现高效测试时自适应,显著提升模型泛化能力。
IMSE: Intrinsic Mixture of Spectral Experts Fine-tuning for Test-Time Adaptation
- 仅更新线性层奇异值,保持向量固定,减少参数量
- 提出多样性损失缓解特征坍缩,提升分类判别性
- 支持持续适应,复用历史域知识,参数量减少385倍
测试时自适应(TTA)被广泛研究以应对测试数据分布与训练分布不一致导致的性能下降问题。然而,如何在最小参数更新下充分利用大型预训练模型的丰富表示仍待探索。本文提出固有谱专家混合(IMSE),利用视觉变压器中内在的谱专家结构。通过奇异值分解(SVD)对每一层线性层进行分解,仅适应奇异值,而保持奇异向量固定。我们进一步发现熵最小化在TTA中的关键缺陷:常引发特征坍缩,使模型依赖领域特定特征而非类别判别特征。为此,我们提出基于专家-输入对齐的多样性最大化损失,鼓励适应过程中谱专家的多样化使用。在持续测试时自适应(CTTA)场景中,除了保留预训练知识外,还需保留并重用先前观察到的域的知识。我们引入域感知谱码检索机制,通过估计输入分布检测域偏移,并快速检索已适配的奇异值。结果表明,该方法在多种分布偏移基准上达到最先进性能;在CTTA和渐进式CTTA设置下,分别提升准确率3.4个百分点和2.4个百分点,同时仅需385倍更少的可训练参数。代码已开源于https://github.com/baek85/IMSE。
原文摘要 · Abstract (English)
Test-time adaptation (TTA) has been widely explored to prevent performance degradation when test data differ from the training distribution. However, fully leveraging the rich representations of large pretrained models with minimal parameter updates remains underexplored. In this paper, we propose Intrinsic Mixture of Spectral Experts (IMSE) that leverages the spectral experts inherently embedded in Vision Transformers. We decompose each linear layer via singular value decomposition (SVD) and adapt only the singular values, while keeping the singular vectors fixed. We further identify a key limitation of entropy minimization in TTA: it often induces feature collapse, causing the model to rely on domain-specific features rather than class-discriminative features. To address this, we propose a diversity maximization loss based on expert-input alignment, which encourages diverse utilization of spectral experts during adaptation. In the continual test-time adaptation (CTTA) scenario, beyond preserving pretrained knowledge, it is crucial to retain and reuse knowledge from previously observed domains. We introduce Domain-Aware Spectral Code Retrieval, which estimates input distributions to detect domain shifts, and retrieves adapted singular values for rapid adaptation. Consequently, our method achieves state-of-the-art performance on various distribution-shift benchmarks under the TTA setting. In CTTA and Gradual CTTA, it further improves accuracy by 3.4 percentage points (pp) and 2.4 pp, respectively, while requiring 385 times fewer trainable parameters. Our code is available at https://github.com/baek85/IMSE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。