Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2415002(2023)

Wheel Tread Anomaly Detection Based on Attentional Reverse Knowledge Distillation

Rongrong Qin, xiaorong Gao*, Lin Luo, and Jinlong Li
Author Affiliations
  • School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610000, Sichuan, China
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    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)

    DOI:10.3788/LOP230787

    Topics