arXiv · 2407.15773
STAMP: Outlier-Aware Test-Time Adaptation with Stable Memory Replay
Abstract
Test-time adaptation (TTA) aims to address the distribution shift between the training and test data with only unlabeled data at test time. Existing TTA methods often focus on improving recognition performance specifically for test data associated with classes in the training set. However, during the open-world inference process, there are inevitably test data instances from unknown classes, commonly referred to as outliers. This paper pays attention to the problem that conducts both sample recognition and outlier rejection during inference while outliers exist. To address this problem, we propose a new approach called STAble Memory rePlay (STAMP), which performs optimization over a stable memory bank instead of the risky mini-batch. In particular, the memory bank is dynamically updated by selecting low-entropy and label-consistent samples in a class-balanced manner. In addition, we develop a self-weighted entropy minimization strategy that assigns higher weight to low-entropy samples. Extensive results demonstrate that STAMP outperforms existing TTA methods in terms of both recognition and outlier detection performance. The code is released at https://github.com/yuyongcan/STAMP.
Explore related subjects
Keep this discovery
Yongcan Yu, Lijun Sheng, Ran He, Jian Liang. 2024-07-22. STAMP: Outlier-Aware Test-Time Adaptation with Stable Memory Replay. https://arxiv.org/abs/2407.15773
Cite the original work for its findings. Save a collection to share your selection of sources.