Source-Free Test-Time Adaptation For Online Surface-Defect Detection
This addresses the problem of costly and time-consuming data collection and annotation for industrial defect detection, though it appears incremental as it builds on existing test-time adaptation methods.
The paper tackles the challenge of adapting pre-trained models for surface defect detection to new domains and classes during inference without requiring source data or retraining, achieving state-of-the-art performance in real-time.
Surface defect detection is significant in industrial production. However, detecting defects with varying textures and anomaly classes during the test time is challenging. This arises due to the differences in data distributions between source and target domains. Collecting and annotating new data from the target domain and retraining the model is time-consuming and costly. In this paper, we propose a novel test-time adaptation surface-defect detection approach that adapts pre-trained models to new domains and classes during inference. Our approach involves two core ideas. Firstly, we introduce a supervisor to filter samples and select only those with high confidence to update the model. This ensures that the model is not excessively biased by incorrect data. Secondly, we propose the augmented mean prediction to generate robust pseudo labels and a dynamically-balancing loss to facilitate the model in effectively integrating classification and segmentation results to improve surface-defect detection accuracy. Our approach is real-time and does not require additional offline retraining. Experiments demonstrate it outperforms state-of-the-art techniques.