Contrastive Learning Combining Frequency-Domain and Time-Domain Information for Intelligent Fault Diagnosis of Rotating Machinery

Yang Liu,Weigang Wen,Yihao Bai,Qingzhou Meng
DOI: https://doi.org/10.2139/ssrn.4147550
2022-01-01
SSRN Electronic Journal
Abstract:Data-driven intelligent fault diagnosis requires a large amount of data. However, it is usually difficult for us to acquire adequate labeled data in the field. In addition, with mechanical failure being a small probability event, the problem of insufficient fault data is often presented, which makes it hard for intelligent diagnosis methods. In this paper, a contrastive learning model combining frequency-domain and time-domain information (CLFT) is proposed for rotating machinery fault diagnosis. This model constrains the time-domain and frequency-domain information of vibration signals at the feature level, in order to obtain more refined features for the model encoder. Based on CLFT, a fault diagnosis method applicable to the rotating machinery has been constructed, which uses the pre-trained encoder to fine-tune the downstream task. Experiments indicate that CLFT has strong cross-domain learning capabilities, and the fault diagnosis methods outperform other state-of-the-art methods and perform well in small labeled sample situations.
What problem does this paper attempt to address?