Unsupervised Feature-Preserving CycleGAN for Fault Diagnosis of Rolling Bearings Using Unbalanced Infrared Thermal Imaging Sample

Guo, Lujiale and Chuah, Joon Huang and Raymond, Wong Jee Keen and Gu, Xiaohui and Yao, Jie and Chang, Xiangqian (2024) Unsupervised Feature-Preserving CycleGAN for Fault Diagnosis of Rolling Bearings Using Unbalanced Infrared Thermal Imaging Sample. IEEE Access, 12. pp. 28449-28461. ISSN 2169-3536, DOI https://doi.org/10.1109/ACCESS.2024.3365551.

Full text not available from this repository.
Official URL: https://doi.org/10.1109/ACCESS.2024.3365551

Abstract

The fault diagnosis of rolling bearing is of great significance in industrial safety. The method of infrared thermal image combined with neural network can diagnose the fault of rolling bearing in a non-contact manner, however its data in different scenes are often unbalanced and difficult to obtain. The generative adversarial networks can solve this problem by generating data with the required features. In this paper, an unsupervised learning framework named Feature-Preserving Cycle-Consistent Generative Adversarial Networks (FP-CycleGAN) is designed for defect detection in unbalanced rolling bearing infrared thermography sample. Since the classical Cycle-Consistent Generative Adversarial Networks (CycleGAN) often must balance the weights between generation, discrimination and consistency loss when doing the feature conversion from source domain to target domain, and the process often results in pattern collapse or feature loss. To avoid this problem, a new discriminator is designed to identify whether the generated image A and B belong to two different classes, and a new class loss are proposed. In order to better extract fault features and perform features migration, the new generator is reconstructed based on the U-Network structure, the convtraspose method of the up-sampling network is replaced by Bicubic Interpolation to effectively avoid the checkerboard effect of the generated images. The defect detection of the expanded dataset was performed using Residual Network and compared with the pre-expansion data to demonstrate the usability of the generated data and the superiority of the proposed FP-CycleGAN method for rolling bearing defect detection in small sample of infrared thermal images.

Item Type: Article
Funders: UNSPECIFIED
Uncontrolled Keywords: Fault diagnosis; rolling bearing; infrared thermal imaging; unbalanced data; generative adversarial networks
Subjects: T Technology > TK Electrical engineering. Electronics Nuclear engineering
Divisions: Faculty of Engineering > Department of Electrical Engineering
Depositing User: Ms. Juhaida Abd Rahim
Date Deposited: 13 Nov 2024 04:08
Last Modified: 13 Nov 2024 04:08
URI: http://eprints.um.edu.my/id/eprint/45856

Actions (login required)

View Item View Item