DELTA: DEBIASED FULLY TEST-TIME ADAPTATION

Bowen Zhao,Chen Chen,Shu-Tao Xia
2023-01-01
Abstract:Fully test-time adaptation aims at adapting a pre-trained model to the test stream during real-time inference, which is urgently required when the test distribution differs from the training distribution. Several efforts have been devoted to improving adaptation performance. However, we find that two unfavorable biases are concealed in the prevalent adaptation methodologies like test-time batch normalization (BN) and self-learning. First, we reveal that the normalization statistics in test-time BN are inherently biased in favor of currently received test samples. Second, we show that during test-time adaptation, the parameter update is biased towards some dominant classes. In addition to the extensively studied IID test stream, we further observe that the biases can be exacerbated in more complicated test environments, such as non-IID or class-imbalanced data. In this paper, we provide a plug-in solution called DELTA for debiased fully test-time adaptation, which consists of two components: (i) Test-time batch renormalization (TBR), introduced to alleviate the bias in normalization statistics. (ii) Dynamic online re-weighting (DOT), designed to address the class bias within optimization. We investigate various test-time adaptation methods on three commonly used datasets and four scenarios. Previous approaches only work well in certain scenarios while failing in others; DELTA can help them deal with all scenarios simultaneously, leading to new state-of-the-art test-time adaptation performance.
What problem does this paper attempt to address?