Towards a Fault Diagnosis Method for Rolling Bearings with Time-Frequency Region-Based Convolutional Neural Network

Abstract

An artificial-intelligence (AI)-based method for fault diagnosis is a strong candidate for industrial applications in the health management of rolling bearings. However, traditional fault diagnosis methods fail to improve the detection accuracy because they only extract a single feature and have limitations in feature representation. In addition, advanced object detection frameworks such as region-based convolutional neural networks have not yet been applied in fault diagnosis. To this end, a fault diagnosis model using a Time-Frequency Region-Based Convolutional Neural Network (TF-RCNN) is proposed in this paper. This method was mainly adopted to extract multiple regions that can characterize fault features from the Time-Frequency Representation (TFR). Specifically, an attention module was introduced so the model could focus on representative features. The existing classification strategy was also enhanced to perform multiple types of fault classification. Finally, an end-to-end rolling bearing fault diagnosis framework based on the TF-RCNN was developed with the aforementioned improvements. The effectiveness of this method was proven experimentally on artificial faults and real faults. The superiority of the proposed method is demonstrated using a comparison with the typical object detection method and an advanced fault diagnosis method.