arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2309.14949cs.LGcs.CV

面向真实世界测试时自适应:三网络自训练与平衡归一化

Towards Real-World Test-Time Adaptation: Tri-Net Self-Training with Balanced Normalization

South China University of Technology(华南理工大学);Institute for Infocomm Research, A*STAR(新加坡科技研究局信息通信研究所);School of Data Science, The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳)数据科学学院)

Yongyi Su, Xun Xu, Kui Jia

AI总结:

针对真实世界测试时自适应中类别不平衡与持续域偏移的挑战,提出TRIBE模型,采用三网络架构与平衡批归一化层,并利用锚定损失改进自训练,在四个数据集上达到最先进性能。

摘要:

测试时自适应旨在推理阶段将源域模型适应于测试数据,在适应未知损坏方面已展示出成功。然而,在更具挑战性的真实世界场景下,这些尝试可能会失败。现有工作主要考虑非独立同分布数据流和持续域偏移下的真实世界测试时自适应。在本工作中,我们首先用全局类别不平衡的测试集补充现有的真实世界TTA协议。我们证明将所有设置组合在一起给现有方法带来了新的挑战。我们认为最先进方法的失败首先是由于不加区分地将归一化层适应于不平衡的测试数据。为弥补这一缺陷,我们提出一种平衡批归一化层,在推理阶段替换常规批归一化。新的批归一化层能够适应而不偏向多数类。我们进一步受到自训练在从无标签数据中学习方面成功的启发,并将自训练适应于测试时自适应。然而,自训练本身容易过度适应,这导致在持续域偏移下性能不佳。因此,我们提出通过使用锚定损失正则化模型更新来改进持续域偏移下的自训练。最终的TTA模型,称为TRIBE,基于具有平衡批归一化层的三网络架构构建。我们在代表真实世界TTA设置的四个数据集上评估TRIBE。TRIBE在多个评估协议上持续达到最先进的性能。代码可在https://github.com/Gorilla-Lab-SCUT/TRIBE获取。

英文摘要:

Test-Time Adaptation aims to adapt source domain model to testing data at inference stage with success demonstrated in adapting to unseen corruptions. However, these attempts may fail under more challenging real-world scenarios. Existing works mainly consider real-world test-time adaptation under non-i.i.d. data stream and continual domain shift. In this work, we first complement the existing real-world TTA protocol with a globally class imbalanced testing set. We demonstrate that combining all settings together poses new challenges to existing methods. We argue the failure of state-of-the-art methods is first caused by indiscriminately adapting normalization layers to imbalanced testing data. To remedy this shortcoming, we propose a balanced batchnorm layer to swap out the regular batchnorm at inference stage. The new batchnorm layer is capable of adapting without biasing towards majority classes. We are further inspired by the success of self-training (ST) in learning from unlabeled data and adapt ST for test-time adaptation. However, ST alone is prone to over adaption which is responsible for the poor performance under continual domain shift. Hence, we propose to improve self-training under continual domain shift by regularizing model updates with an anchored loss. The final TTA model, termed as TRIBE, is built upon a tri-net architecture with balanced batchnorm layers. We evaluate TRIBE on four datasets representing real-world TTA settings. TRIBE consistently achieves the state-of-the-art performance across multiple evaluation protocols. The code is available at https://github.com/Gorilla-Lab-SCUT/TRIBE.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑