Abstract:As wheels are important components of train operation, diagnosing and predicting wheel faults are essential to ensure the reliability of rail transit. Currently, the existing studies always separately deal with two main types of wheel faults, namely wheel radius difference and wheel flat, even though they are both reflected by wheel radius changes. Moreover, traditional diagnostic methods, such as mechanical methods or a combination of data analysis methods, have limited abilities to efficiently extract data features. Deep learning models have become useful tools to automatically learn features from raw vibration signals. However, research on improving the feature-learning capabilities of models under noise interference to yield higher wheel diagnostic accuracies has not yet been conducted. In this paper, a unified training framework with the same model architecture and loss function is established for two homologous wheel faults. After selecting deep residual networks (ResNets) as the backbone network to build the model, we add the squeeze and excitation (SE) module based on a multichannel attention mechanism to the backbone network to learn the global relationships among feature channels. Then the influence of noise interference features is reduced while the extraction of useful information features is enhanced, leading to the improved feature-learning ability of ResNet. To further obtain effective feature representation using the model, we introduce supervised contrastive loss (SCL) on the basis of ResNet + SE to enlarge the feature distances of different fault classes through a comparison between positive and negative examples under label supervision to obtain a better class differentiation and higher diagnostic accuracy. We also complete a regression task to predict the fault degrees of wheel radius difference and wheel flat without changing the network architecture. The extensive experimental results show that the proposed model has a high accuracy in diagnosing and predicting two types of wheel faults.

RSG-Net: A Recurrent Similarity Network with Ghost Convolution for Wheelset Laser Stripe Image Inpainting

Image Inpainting Based on Interactive Separation Network and Progressive Reconstruction Algorithm

Free-Form Image Inpainting with Separable Gate Encoder-Decoder Network

The Improved Image Inpainting Algorithm Via Encoder and Similarity Constraint

URS: A Light-Weight Segmentation Model for Train Wheelset Monitoring

A Decoder-Free Reconstruction Method for Semi-Supervised Surface Defect Detection

RSDNet: A New Multiscale Rail Surface Defect Detection Model

A Double Feature Fusion Network with Progressive Learning for Sharper Inpainting

A Multichannel MN-GCN for Wheelset-Bearing System Fault Diagnosis

RNON: image inpainting via repair network and optimization network

Wheel fault diagnosis model based on multichannel attention and supervised contrastive learning

Semantic Image Inpainting with Multi-Stage Feature Reasoning Generative Adversarial Network

A subtle defect recognition method for catenary fastener in high-speed railroad using destruction and reconstruction learning

Blind Image Restoration Based on Cycle-Consistent Network

Laser Curve Extraction of a Train Wheelset Based on an Encoder-Decoder Network.

Improved YOLOv8-Based Target Precision Detection Algorithm for Train Wheel Tread Defects

Recurrent Feature Reasoning for Image Inpainting

Detection of Train Wheelset Tread Defects with Small Samples Based on Local Inference Constraint Network

Attention-Based Real Image Restoration

Image Inpainting Using Two-Stage Loss Function and Global and Local Markovian Discriminators

A novel multi-exposure fusion-induced stripe inpainting method for blade reflection-encoded images