Back to Source: Open-Set Continual Test-Time Adaptation via Domain Compensation
Yingkai Yang, Chaoqi Chen, Hui Huang
TLDR
DOCO introduces a novel framework for Open-set Continual Test-Time Adaptation (OCTTA), synergistically handling domain shifts and unknown classes.
Key contributions
- Tackles Open-set Continual Test-Time Adaptation (OCTTA), a challenging setting with evolving domains and unknown classes.
- Introduces DOCO, a lightweight framework for synergistic domain adaptation and out-of-distribution (OOD) detection.
- Employs dynamic sample splitting to separate in-distribution (ID) from OOD samples within a batch.
- Learns a domain compensation prompt from ID samples, then propagates it to OOD samples for reliable detection.
Why it matters
This paper addresses the critical and realistic problem of Open-set Continual Test-Time Adaptation (OCTTA), where models face both evolving domains and novel semantic classes. By proposing DOCO, it offers a robust solution that significantly improves classification and OOD detection, establishing a new state-of-the-art. This advances the practical applicability of TTA methods in dynamic environments.
Original Abstract
Test-Time Adaptation (TTA) aims to mitigate distributional shifts between training and test domains during inference time. However, existing TTA methods fall short in the realistic scenario where models face both continually changing domains and the simultaneous emergence of unknown semantic classes, a challenging setting we term Open-set Continual Test-Time Adaptation (OCTTA). The coupling of domain and semantic shifts often collapses the feature space, severely degrading both classification and out-of-distribution detection. To tackle this, we propose DOmain COmpensation (DOCO), a lightweight and effective framework that robustly performs domain adaptation and OOD detection in a synergistic, closed loop. DOCO first performs dynamic, adaptation-conditioned sample splitting to separate likely ID from OOD samples. Then, using only the ID samples, it learns a domain compensation prompt by aligning feature statistics with the source domain, guided by a structural preservation regularizer that prevents semantic distortion. This learned prompt is then propagated to the OOD samples within the same batch, effectively isolating their semantic novelty for more reliable detection. Extensive experiments on multiple challenging benchmarks demonstrate that DOCO outperforms prior CTTA and OSTTA methods, establishing a new state-of-the-art for the demanding OCTTA setting.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.