提出缓冲层解决测试时适应中的统计不稳问题
Buffer layers for Test-Time Adaptation
- 用可学习的缓冲层替代更新归一化参数,避免小批量干扰
- 在多个数据集上提升准确率,且显著减少灾难性遗忘
- 模块化设计,兼容主流测试时适应框架
近期测试时适应(TTA)方法主要依赖更新归一化层以适配测试域,但此类方法在小批量下统计不稳,且受限于预训练模型结构,难以应对显著域偏移。本文提出基于缓冲层的新范式,不修改主干网络参数,有效避免灾难性遗忘。实验表明,该方法在多种架构上均优于传统方法,显著提升鲁棒性与抗遗忘能力。缓冲层具备模块化特性,可无缝集成至几乎所有现有TTA框架中,带来一致性能提升。代码已开源。
原文摘要 · Abstract (English)
In recent advancements in Test Time Adaptation (TTA), most existing methodologies focus on updating normalization layers to adapt to the test domain. However, the reliance on normalization-based adaptation presents key challenges. First, normalization layers such as Batch Normalization (BN) are highly sensitive to small batch sizes, leading to unstable and inaccurate statistics. Moreover, normalization-based adaptation is inherently constrained by the structure of the pre-trained model, as it relies on training-time statistics that may not generalize well to unseen domains. These issues limit the effectiveness of normalization-based TTA approaches, especially under significant domain shift. In this paper, we introduce a novel paradigm based on the concept of a Buffer layer, which addresses the fundamental limitations of normalization layer updates. Unlike existing methods that modify the core parameters of the model, our approach preserves the integrity of the pre-trained backbone, inherently mitigating the risk of catastrophic forgetting during online adaptation. Through comprehensive experimentation, we demonstrate that our approach not only outperforms traditional methods in mitigating domain shift and enhancing model robustness, but also exhibits strong resilience to forgetting. Furthermore, our Buffer layer is modular and can be seamlessly integrated into nearly all existing TTA frameworks, resulting in consistent performance improvements across various architectures. These findings validate the effectiveness and versatility of the proposed solution in real-world domain adaptation scenarios. The code is available at https://github.com/hyeongyu-kim/Buffer_TTA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。