Few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization
Dengming Zhang,Kai Zheng,Dengke Yao,Dewei Yang,Shaowang Wang,Bai Yin
DOI: https://doi.org/10.1088/1361-6501/ac8303
IF: 2.398
2022-07-22
Measurement Science and Technology
Abstract:In practical industrial applications, the collected fault data are usually insufficient due to the sudden occurrence of faults. However, the current deep learning based fault diagnosis methods often rely on a large number of samples to achieve satisfactory performance. Moreover, the heavy background noise and the variability of working conditions also degraded the performance of existing fault diagnostic approaches. Facing these challenges, a new fault diagnosis method for few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization (MLDSO) is proposed in this research. To begin with, the fault feature of the rolling bearing is extracted through the tailored networks. Then, the feature extractor is optimized by the discriminant space loss proposed in this paper, so as to promote the clustering of the extracted fault features of the same category and the distinguishment among different types of fault features. After that, the feature extractor and discriminant space optimizer are constructed to optimize the feature discriminant space, thus a high fault-tolerant discriminant space is obtained for meta-learning. Eventually, the faults in the new task can be accurately classified with the assistance of previously learned meta-knowledge and a few known samples when dealing with new tasks under different working conditions. The effectiveness and superiority of the proposed MLDSO method are verified through the datasets collected from the self-designed experimental platform and the Case Western Reserve University test platform. The experimental results show superior performance over the advanced methods. It indicates that the proposed method is a promising approach under small samples, heavy noise, and variable working conditions.
engineering, multidisciplinary,instruments & instrumentation