Data-driven fault diagnosis methods for rotating machinery have developed rapidly with the help of deep learning methods. However, traditional intelligent fault diagnosis methods still have some limitations in fault feature extraction and the latest object detection theory has not been applied in fault diagnosis. To this end, a fault diagnosis method based on a sparse short-term Fourier transform (SSTFT) and object detection theory is developed in this paper. First, a sparse constraint is introduced in time-frequency analysis to improve the time-frequency resolution of the model without cross-term interference and proximal gradient descent (PGD) is adopted to quickly and effectively optimize the model to obtain a high-quality time-frequency representation (TFR). Second, a fault diagnosis model based on a region-based convolutional neural network (RCNN) is built; the model can extract multiple regions that can characterize fault features from the TFR. This process avoids the interference of irrelevant vibration components and improves the interpretability of the fault diagnosis model. Finally, multicategory rolling bearing fault identification is realized. The effectiveness of the proposed method is validated by simulation signals and bearing experiments. The results indicate that the proposed method is more effective than existing methods.
Keywords: RCNN; fault diagnosis; proximal gradient descent; rolling bearing.