Entropy is not Enough for Test-Time Adaptation: From the Perspective of Disentangled Factors
Jonghyun Lee, Dahuin Jung, Saehyung Lee, Junsung Park, Juhyeon Shin, Uiwon Hwang, Sungroh Yoon
TL;DR
The paper demonstrates that entropy alone is insufficient for reliable test-time adaptation in the presence of disentangled factors that differentially correlate with labels. It introduces PLPD, a confidence metric derived from object-shape distortions, and a TTA method DeYO that jointly uses entropy and PLPD for sample selection and weighting, emphasizing CPR factors. Across mild and wild distribution shifts on ImageNet-C, Waterbirds, ColoredMNIST, ImageNet-R, and VisDA-2021, DeYO consistently outperforms state-of-the-art baselines, with notable gains in hard scenarios and even surpassing random chance on ColoredMNIST. The approach offers practical robustness with modest computational overhead and provides insights into leveraging factor-aware signals for online adaptation.
Abstract
Test-time adaptation (TTA) fine-tunes pre-trained deep neural networks for unseen test data. The primary challenge of TTA is limited access to the entire test dataset during online updates, causing error accumulation. To mitigate it, TTA methods have utilized the model output's entropy as a confidence metric that aims to determine which samples have a lower likelihood of causing error. Through experimental studies, however, we observed the unreliability of entropy as a confidence metric for TTA under biased scenarios and theoretically revealed that it stems from the neglect of the influence of latent disentangled factors of data on predictions. Building upon these findings, we introduce a novel TTA method named Destroy Your Object (DeYO), which leverages a newly proposed confidence metric named Pseudo-Label Probability Difference (PLPD). PLPD quantifies the influence of the shape of an object on prediction by measuring the difference between predictions before and after applying an object-destructive transformation. DeYO consists of sample selection and sample weighting, which employ entropy and PLPD concurrently. For robust adaptation, DeYO prioritizes samples that dominantly incorporate shape information when making predictions. Our extensive experiments demonstrate the consistent superiority of DeYO over baseline methods across various scenarios, including biased and wild. Project page is publicly available at https://whitesnowdrop.github.io/DeYO/.
