Semi-Supervised Bolt Anomaly Detection Based on Local Feature Reconstruction.
Yun Peng,Chuangwei Liu,Yi Yan,Nachuan Ma,Deming Wang,Chengju Liu,Qijun Chen
DOI: https://doi.org/10.1109/tim.2023.3278293
IF: 5.6
2023-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:During the process of high-speed railway online maintenance, anomaly detection of targets in grease injection facilities is extremely challenging. Environmental uncertainty (noise disturbance, various illumination conditions, changeable camera perspective, etc.) limits the performance of existing unsupervised anomaly detection methods, which tend to struggle in such complex backgrounds without prior knowledge of anomalies. Therefore, we propose a novel semi-supervised anomaly detection algorithm that only learns from anomaly-free samples and annotations while can recognize both anomaly-free and anomaly targets (key areas) and detecting anomalous targets simultaneously. We first propose a key-area recognition module (KRM) to segment anomaly-free targets and predict coordinates of both anomaly-free and anomalous key areas' center points (key points). By using the predicted coordinates, we generated a circular mask to erase features within the key areas. Based on the erased features, a local feature reconstruction and segmentation module (LFRSM) is proposed to restore the latent embeddings within the key areas and segment both anomaly-free and anomalous key areas. The final anomaly results are obtained by analyzing the original segmentation results of KRM and the reconstructed segmentation results of LFRSM. To validate the effectiveness of the proposed algorithm, we develop a dataset, called BOLT-DET, for bolt anomaly detection in complex scenes. The proposed algorithm outperforms other existing state-of-the-art (SOTA) anomaly detection algorithms (SPADE, DFR, CFLOW, Student-Teacher, PatchCore, and DRAEM) with mean intersection over union (mIoU) of 0.66, area under the receiver operating characteristic curve (AUROC) of 0.92, and F1-Measure of 0.69 on BOLT-DET. It can accurately detect anomalies and meet the requirement of practical industrial applications.