Fault Diagnosis Method for Railway Signal Equipment Based on Data Enhancement and an Improved Attention Mechanism

Ni Yang,Youpeng Zhang,Jing Zuo,Bin Zhao
DOI: https://doi.org/10.3390/machines12050334
IF: 2.899
2024-05-14
Machines
Abstract:Railway signals' fault text data contain a substantial amount of expert maintenance experience. Extracting valuable information from these fault text data can enhance the efficiency of fault diagnosis for signal equipment, thereby contributing to the advancement of intelligent railway operations and maintenance technology. Considering that the characteristics of different signal equipment in actual operation can easily lead to a lack of fault data, a fault diagnosis method for railway signal equipment based on data augmentation and an improved attention mechanism (DEIAM) is proposed in this paper. Firstly, the original fault dataset is preprocessed based on data augmentation technology and retained noun and verb operations. Then, the neural network is constructed by integrating a bidirectional long short-term memory (BiLSTM) model with an attention mechanism and a convolutional neural network (CNN) model enhanced with a channel attention mechanism. The DEIAM method can more effectively capture the important text features and sequence features in fault text data, thereby facilitating the diagnosis and classification of such data. Consequently, it enhances onsite fault maintenance experience by providing more precise insights. An empirical study was conducted on a 10-year fault dataset of signal equipment produced by a railway bureau. The experimental results demonstrate that in comparison with the benchmark model, the DEIAM model exhibits enhanced performance in terms of accuracy, precision, recall, and F1.
engineering, mechanical, electrical & electronic
What problem does this paper attempt to address?
### What problem does this paper attempt to solve? This paper aims to address several key issues in the fault diagnosis of railway signal equipment: 1. **Data Imbalance**: Due to differences in the structure and usage frequency of different equipment, there is a significant imbalance in the amount of fault data generated during actual operation. This affects the quality of the training dataset and leads to classification results that are biased towards categories with more samples, thereby impacting the accuracy of fault diagnosis. 2. **Unstructured Nature of Text Data**: Railway signal fault text data is unstructured and contains a large amount of expert maintenance experience. However, this data is not conducive to computer analysis and processing, resulting in a waste of resources. To solve these problems, the authors propose a fault diagnosis method based on Data Augmentation and Improved Attention Mechanism (DEIAM). Specifically, this method includes the following steps: - Using data augmentation techniques (such as EDA and back-translation) to expand the original fault dataset and address the issue of sample imbalance. - Constructing a neural network model that combines Bidirectional Long Short-Term Memory (BiLSTM) and Convolutional Neural Network (CNN), and introducing an Improved Channel Attention Mechanism (ICWA) to capture important text features and sequence features, thereby improving the accuracy of fault diagnosis. Through the above methods, the model can more effectively extract important features from fault text data and classify them, ultimately improving the experience and efficiency of on-site fault maintenance. Experimental results show that compared to benchmark models, the DEIAM model performs excellently in terms of accuracy, precision, recall, and F1 score.