Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Reshaping the Online Data Buffering and Organizing Mechanism for Continual Test-Time Adaptation

About

Continual Test-Time Adaptation (CTTA) involves adapting a pre-trained source model to continually changing unsupervised target domains. In this paper, we systematically analyze the challenges of this task: online environment, unsupervised nature, and the risks of error accumulation and catastrophic forgetting under continual domain shifts. To address these challenges, we reshape the online data buffering and organizing mechanism for CTTA. We propose an uncertainty-aware buffering approach to identify and aggregate significant samples with high certainty from the unsupervised, single-pass data stream. Based on this, we propose a graph-based class relation preservation constraint to overcome catastrophic forgetting. Furthermore, a pseudo-target replay objective is used to mitigate error accumulation. Extensive experiments demonstrate the superiority of our method in both segmentation and classification CTTA tasks. Code is available at https://github.com/z1358/OBAO.

Zhilin Zhu, Xiaopeng Hong, Zhiheng Ma, Weijun Zhuang, Yaohui Ma, Yong Dai, Yaowei Wang• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet-C Severity 5 (test)
Mean Error Rate (Severity 5)59
216
Image ClassificationCIFAR-100-C--
137
Semantic segmentationCityscapes to ACDC (test)
mIoU61
85
Image ClassificationCIFAR10-C--
52
Online Continual Test-Time AdaptationImageNet-C Severity 5 (test)
Accuracy (Gaussian Noise, ImageNet-C S5)51.4
47
Image ClassificationImageNet-C
Gauss Error78.5
36
Online Continual Test-Time AdaptationCIFAR-10-C severity 5 (test)
Gaussian Noise Accuracy (Severity 5)33.1
24
Semantic segmentationACDC Round 2
mIoU (Fog)72.6
19
Semantic segmentationACDC Round 3
mIoU (Fog)72.8
19
Continual Test-Time AdaptationCIFAR100C Severity 5 (test)
Accuracy (Gaussian Corruption)56.3
10
Showing 10 of 11 rows

Other info

Follow for update