Abstract
With the rapid development of railroads and the yearly increase in the scale of operation, the safe operation and maintenance of rail trains have become particularly important. Among them, deep learning-based bearing fault diagnosis methods have attracted more and more attention in rail train operation and maintenance. However, rail trains usually operate normally. Collecting complete fault data for deep learning model training is often difficult. Such scenarios with a large difference between the number of normal data and fault data usually affect the performance of fault diagnosis models. Here, an interactive generative feature space oversampling-based autoencoder (IGFSO-AE) is proposed to realize fault sample generation under imbalanced data. First, the original vibration signal is converted into a semantically stable amplitude–frequency signal by fast Fourier transform and input into the autoencoder; second, the order of the hidden layer space features of the autoencoder is randomly exchanged, and the interactive sample generation learning strategy trains the autoencoder; then, interpolation oversampling is used to interpolate samples in the hidden layer space where the Euclidean distance between samples is large, and is input into the decoder, the generated samples are mixed with the original samples to form a new training set, which is used to train the intelligent fault diagnosis model and output the diagnosis results. Finally, the performance of the proposed method is evaluated using the publicly available bearing dataset and the bogie-bearing fault simulation bench in our lab. The experimental results show that IGFSO-AE can generate diverse samples with incremental information and exhibits robustness and superiority in different imbalanced proportions of data.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.