Test-Time Adaptation with Binary Feedback
By: Taeckyung Lee , Sorn Chottananurak , Junsu Kim and more
Potential Business Impact:
Helps computers learn better with less feedback.
Deep learning models perform poorly when domain shifts exist between training and test data. Test-time adaptation (TTA) is a paradigm to mitigate this issue by adapting pre-trained models using only unlabeled test samples. However, existing TTA methods can fail under severe domain shifts, while recent active TTA approaches requiring full-class labels are impractical due to high labeling costs. To address this issue, we introduce a new setting of TTA with binary feedback. This setting uses a few binary feedback inputs from annotators to indicate whether model predictions are correct, thereby significantly reducing the labeling burden of annotators. Under the setting, we propose BiTTA, a novel dual-path optimization framework that leverages reinforcement learning to balance binary feedback-guided adaptation on uncertain samples with agreement-based self-adaptation on confident predictions. Experiments show BiTTA achieves 13.3%p accuracy improvements over state-of-the-art baselines, demonstrating its effectiveness in handling severe distribution shifts with minimal labeling effort. The source code is available at https://github.com/taeckyung/BiTTA.
Similar Papers
Monitoring Risks in Test-Time Adaptation
Machine Learning (CS)
Warns when AI models stop working correctly.
BoTTA: Benchmarking on-device Test Time Adaptation
Machine Learning (CS)
Makes AI work better on phones and small devices.
Effortless Active Labeling for Long-Term Test-Time Adaptation
CV and Pattern Recognition
Helps computers learn from messy data with less work.