Implementation of a novel algorithm of wheelset and axle box concurrent fault identification based on an efficient neural network with the attention mechanism

With the rapid development of urban rail transit in recent years, it becomes necessary to ensure the operation safety of train wheelset axle boxes. Aiming at the problems of large model size and long diagnosis time in traditional fault diagnosis methods, this paper proposed a novel model to identify concurrent faults in wheelset axle boxes based on an efficient neural network and the attention mechanism. The model was developed based on an improved Ghost bottleneck module sequence to achieve an efficient model. Furthermore, the use of the convolutional block attention module adaptively refined the feature map to improve the generalization of the model. Feature pyramid network was used to fuse the shallow and deep features in the network to improve the extraction ability of various size features. Experiments were carried out on wheelset axle box concurrent fault datasets. Model size, diagnosis speed and accuracy were used as evaluation indexes to compare the efficiencies of existing fault diagnosis methods and our proposed model. Experimental results revealed that the proposed algorithm effectively diagnosed wheelset axle box concurrent faults.