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
    Figures & Tables(19)
    Reverse knowledge distillation network structure based on attention
    Structure of encoder bottleneck
    Structure of decoder bottleneck
    Multi-scale feature fusion
    Attention mechanism
    SE module structure
    SGSE module structure
    Wheel tread image. (a)‒(c) Normal wheel tread image; (d) peeling defect image; (e) scratch defect image; (f) injury defect image
    Segmentation model training loss curve
    Prediction diagram of wheel tread image segmentation
    Detection view on RD model before wheel tread image segmentation
    Detection view on RD model after wheel tread image segmentation
    Comparison of detection view of wheel tread by reverse knowledge distillation method and the proposed method
    Comparison of detection view of wheel tread by different anomaly detection methods and the proposed method
    • Table 1. Comparison of results of reverse knowledge distillation model before and after anomaly detection dataset segmentation

      View table

      Table 1. Comparison of results of reverse knowledge distillation model before and after anomaly detection dataset segmentation

      ModelSegmentationAUC /%P /%R /%Acc /%
      RDBefore66.755.597.958.4
      After93.682.194.086.3
    • Table 2. Comparison of ablation results

      View table

      Table 2. Comparison of ablation results

      MethodAUC /%P /%R /%Acc /%
      RD93.682.194.086.3
      RD+Attention(our)93.882.395.487.0
    • Table 3. Recall of different types of defects detected before and after model improvement

      View table

      Table 3. Recall of different types of defects detected before and after model improvement

      MethodRecall of peel /%Recall of scratch /%Recall of injury /%
      RD97.599.085.1
      RD+Attention(our)98.099.488.2
    • Table 4. Comparison of experimental results of different models

      View table

      Table 4. Comparison of experimental results of different models

      ModelAUC /%P /%R /%Acc /%
      GANomaly57.851.9100.052.2
      MKD84.373.088.977.3
      Our93.882.395.487.0
    • Table 5. Recall of different types of defects detected on different models

      View table

      Table 5. Recall of different types of defects detected on different models

      ModelRecall of peel /%Recall of scratch /%Recall of injury /%
      GANomaly100100100
      MKD95.691.782.1
      Our98.099.488.2
    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