详情

原文传递 Fault Diagnosis for Rolling Bearings of a Freight Train under Limited Fault Data: Few-Shot Learning Method

题名：	Fault Diagnosis for Rolling Bearings of a Freight Train under Limited Fault Data: Few-Shot Learning Method
正文语种：	eng
作者：	Chenzhong Li;Kanghua Yang;Haichuan Tang;Ping Wang;Jiebo Li;Qing He
作者单位：	School of Civil Engineering Southwest Jiaotong Univ. Chengdu 610031 China Key Laboratory Ministry of Education of High-Speed Railway Engineering Southwest Jiaotong Univ. Chengdu 610031 China;School of Civil Engineering Southwest Jiaotong Univ. Chengdu 610031 China Graduate Student Key Laboratory Ministry of Education of High-Speed Railway Engineering Southwest Jiaotong Univ. Chengdu 610031 China;Dept. of Technical Research China Railway Rolling Stock Corporation Academy Fengtai Beijing 100071 China;Key Laboratory Ministry of Education of High-Speed Rail-way Engineering School of Civil Engineering Southwest Jiaotong Univ. Chengdu 610031 China;Locomotive and Car Research Institute China Academy of Railway Sciences Corporation Limited Haidian Beijing 100081 China;Key Laboratory Ministry of Education of High-Speed Rail-way Engineering School of Civil Engineering Southwest Jiaotong Univ. Chengdu 610031 China
关键词：	Freight train bearing; Fault diagnosis; Few-shot learning; Convolutional neural network (CNN); Conditional Wasserstein generative adversarial network (C-WGAN)
摘要：	In recent years, many machine learning-based methods have emerged to detect faulty bearings. However, most of these methods may not be practical due to the need to collect a large number of fault samples for training. This paper developed a novel few-shot learning framework for the fault diagnosis of freight train rolling bearings. The proposed method has the capability to transfer the learning outcome from one bearing fault diagnosis model to another different but related task for which very limited training data are available. The authors established a single-wheelset platform to collect acceleration signals of different types of bearing faults. The authors preprocessed the data through data segmentation and frequency domain transformation, and divided the data into training and test sets according to a certain ratio. A one-dimensional convolutional neural network (ID-CNN) was established to automatically extract the features of the bearing vibration signals and classify the fault types. The authors implemented two few-shot learning methods through parameter fine-tuning and a conditional Wasserstein generative adversarial network (C-WGAN). A case study demonstrated the classification performance of the proposed models. The results showed that the diagnosis capability of the ID-CNN in the frequency domain is significantly superior to that in the time domain. However, when the amount of data is small, the ID-CNN model does not work. In contrast, the few-shot learning of bearing faults works well for both the fine-tuned CNN and C-WGAN models. Furthermore, the classification performance of the C-WGAN is better than that of the fine-tuned CNN when the training data are extremely limited.
出版年：	2021
期刊名称：	Journal of Transportation Engineering
卷：	147
期：	8