Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2415002(2023)
Wheel Tread Anomaly Detection Based on Attentional Reverse Knowledge Distillation
Wheels are an essential part of railway trains; thus, defects on the wheel tread present serious risk regarding the safety of railway trains. Due to the limited samples of wheel tread defects in practice, the corresponding supervised detection model is insufficient. To solve this problem, an unsupervised knowledge distillation anomaly detection model is proposed to detect wheel tread anomalies. Accordingly, UNet is employed to segment the tread region and reduce the influence of non-tread regions on the anomaly detection model. An attention mechanism is then added after the multiscale feature fusion to improve the ability of the student network to reconstruct normal features in the reverse knowledge distillation structure, as well as enhance the reconstruction of normal features. From the experimental results, the improved model achieves the performance indexes of 93.8% area under receiver operating characteristic curve, 82.3% precision, 95.4% recall, and 87.0% accuracy considering the railway wheel tread dataset. Compared with the original model, the detection performance of the model is improved.
Get Citation
Copy Citation Text
Rongrong Qin, xiaorong Gao, Lin Luo, Jinlong Li. Wheel Tread Anomaly Detection Based on Attentional Reverse Knowledge Distillation[J]. Laser & Optoelectronics Progress, 2023, 60(24): 2415002
Category: Machine Vision
Received: Mar. 7, 2023
Accepted: May. 6, 2023
Published Online: Dec. 4, 2023
The Author Email: Gao xiaorong (gxrr@vip.163.com)