A Novel Fault Diagnosis Method of High-Speed Train Based on Few-Shot Learning

Yunpu Wu,Jianhua Chen,Xia Lei,Weidong Jin
DOI: https://doi.org/10.3390/e26050428
IF: 2.738
2024-05-17
Entropy
Abstract:Ensuring the safe and stable operation of high-speed trains necessitates real-time monitoring and diagnostics of their suspension systems. While machine learning technology is widely employed for industrial equipment fault diagnosis, its effective application relies on the availability of a large dataset with annotated fault data for model training. However, in practice, the availability of informational data samples is often insufficient, with most of them being unlabeled. The challenge arises when traditional machine learning methods encounter a scarcity of training data, leading to overfitting due to limited information. To address this issue, this paper proposes a novel few-shot learning method for high-speed train fault diagnosis, incorporating sensor-perturbation injection and meta-confidence learning to improve detection accuracy. Experimental results demonstrate the superior performance of the proposed method, which introduces perturbations, compared to existing methods. The impact of perturbation effects and class numbers on fault detection is analyzed, confirming the effectiveness of our learning strategy.
physics, multidisciplinary
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper aims to address the issue of data scarcity in high-speed train fault diagnosis. Specifically: 1. **Data Scarcity Issue**: In practical applications, there is often a lack of labeled fault data samples for training machine learning models, with most data being unlabeled. This leads to traditional machine learning methods easily overfitting when faced with limited training data, thereby affecting the accuracy of fault detection. 2. **Real-time and Accurate Fault Diagnosis**: Ensuring the safe and stable operation of high-speed trains requires real-time monitoring and diagnosis of their suspension systems. Traditional fault diagnosis methods rely on a large amount of labeled data, which is difficult to obtain in practical applications. Therefore, a new method is needed to improve the accuracy and real-time performance of fault detection, especially in the case of data scarcity. ### Solution To address the above challenges, the paper proposes a novel fault diagnosis method based on Few-Shot Learning, combining Sensor-Perturbation Injection and Meta-Confidence Learning. The main contributions of this method include: 1. **Sensor-Perturbation Injection**: By introducing sensor-level perturbations during training, the model's adaptability to different data distributions and features is enhanced, thereby improving the accuracy of fault classification. 2. **Meta-Confidence Learning**: Utilizing meta-learning techniques, the model updates class prototypes and calculates confidence scores to improve the reliability of classifying unknown samples. 3. **Experimental Validation**: The effectiveness of the proposed method is validated through experimental results on a high-speed train fault diagnosis dataset, and the impact of perturbation effects and the number of classes on fault detection is analyzed. ### Summary The paper effectively addresses the issue of data scarcity in high-speed train fault diagnosis by proposing a Few-Shot Learning method that combines Sensor-Perturbation Injection and Meta-Confidence Learning. This method improves the accuracy and reliability of fault detection, providing a new solution for fault diagnosis in practical applications.