Source-Free Test-Time Adaptation For Online Surface-Defect Detection

TL;DR

This paper introduces a real-time, source-free test-time adaptation method for surface-defect detection that dynamically updates models during inference, improving accuracy without retraining.

Contribution

It proposes a novel test-time adaptation framework with confidence-based sample filtering and robust pseudo-label generation for improved defect detection.

Findings

Outperforms state-of-the-art methods in accuracy.

Operates in real-time without offline retraining.

Effectively adapts to new domains and defect classes.

Abstract

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.