Rolling Bearing Fault Diagnosis Method Based On Dual Invariant Feature Domain Generalization

Yining Xie,Jiangtao Shi,Cong Gao,Guojun Yang,Zhichao Zhao,Guohui Guan,Deyun Chen
DOI: https://doi.org/10.1109/tim.2024.3364263
IF: 5.6
2024-02-23
IEEE Transactions on Instrumentation and Measurement
Abstract:In industrial production, complex working conditions often result in poor generalization of fault diagnosis models. Consequently, we propose a method called dual invariant feature domain generalization (DIFDG) for fault diagnosis of rolling bearings. This method is built upon the knowledge distillation framework, wherein phase information is extracted from 1-D signals using the Fourier transform. The phase information remains insensitive to domain changes and serves as the training data for the teacher network, enabling it to learn internally invariant features. Simultaneously, employing the knowledge distillation technique, the student network learns internally invariant features from 1-D time-series signals. To prevent information loss, 2-D time–frequency diagram data is incorporated into the student network training. The student network is trained to acquire mutually invariant features using a class-related loss function for data from various domains. This method trains a student network to extract dual invariant features from the data, enhancing the generalization to applications involving unknown data. Experimental results demonstrate that this approach outperforms other representative methods in both the Case Western Reserve University (CWRU) and NEFU_PHM datasets.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?