A rolling bearing fault diagnosis method based on interactive generative feature space oversampling-based autoencoder under imbalanced data

作者：F Huang, Kai Zhang, Zhixuan Li, Qing Zheng, Guofu Ding, Minghang Zhao, Yuehong Zhang · 发表于：Structural Health Monitoring · 年份：2024 · DOI：10.1177/14759217241248209 · 被引用次数：28 · 研究领域：Machine Fault Diagnosis Techniques、Engineering Diagnostics and Reliability、Gear and Bearing Dynamics Analysis

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...