A Drift Detection Method for Industrial Images Based on a Defect Segmentation Model

Weifeng Li,Bin Li,Zhenrong Wang,Chaochao Qiu,Shuanlong Niu,Xin Tan,Tongzhi Niu
DOI: https://doi.org/10.1016/j.knosys.2024.112320
IF: 8.139
2024-01-01
Knowledge-Based Systems
Abstract:In the widespread application of industrial defect detection supported by neural networks, changes in the characteristics of industrial site data affect the model performance. To achieve timely model adjustment, it is necessary to effectively detect concept drift. For industrial images, drift detection is affected by the difference between defective and normal images and the insensitivity of single-feature indicators to gradual drift. This article proposes a label-free concept drift detection method to characterize and detect concept drift in industrial data based on a segmentation model. We prioritize analyzing intermediate feature images over prediction results to reduce the impact of the differences between defective and normal images on drift detection. Moreover, to address the limitations of a single indicator, a multi-dimensional feature representation method is developed, which integrates grayscale and texture information. The cosine similarity and Relief algorithm is used to select key indicators, and considering the correlation between feature maps, Mahalanobis distance is used to quantify differences in the representation results of different images. The outliers are defined according to the distribution of the Mahalanobis distance within the training dataset, and an outlier ratio-driven drift detection method is designed. The experimental results demonstrate the outstanding performance of our method in both self-built and publicly available datasets, especially in dealing with gradual drift, indicating an important contribution to the monitoring of conceptual drift in industrial images. To promote progress in the field, our dataset will be open to the public for sharing: DDM4II-Based-on-a-Defect-Segmentation-Model.
What problem does this paper attempt to address?