Revisiting TENT for Test-Time Adaption Semantic Segmentation and Classification Head Adjustment.

Xuanpu Zhao,Qi Chu ,Changtao Miao,Bin Liu ,Nenghai Yu
DOI: https://doi.org/10.1007/978-3-031-46311-2_6
2023-01-01
Abstract:Test-time adaption is very effective at solving the domain shift problem where the training data and testing data are sampled from different domains. However, most test-time adaption methods made their success on classification tasks while object detection and segmentation tasks usually have more applications in the real world. Meanwhile, methods that update the model at test-time which is a main branch in test-time adaption (e.g., TENT [ 1 ], a typical method of this branch) only update the backbone, and the classification head remains unchanged. Though the classification head trained by the training data behaves well on the source domain, it is not guaranteed to be effective for a new domain and a new backbone. In our work, we re-weight the entropy of pixels in an image and adopt SAR [ 2 ] to overcome the instability in online adaption. Experiment results show that the segmentation method in TENT becomes more efficient and stable thanks to these improvements. For the classification task, we propose to use T3A [ 3 ] to update the backbone and finetune the classification head in the meantime based on TENT, which boosts the classification accuracy by a large margin.
What problem does this paper attempt to address?