Deep transfer learning with limited data for machinery fault diagnosis

• Published in: Applied soft computing Vol. 103; p. 107150
• Main Authors: Han, Te, Liu, Chao, Wu, Rui, Jiang, Dongxiang
• Format: Journal Article
• Language: English
• Published: Elsevier B.V 01.05.2021
• Subjects: Adversarial domain adaptation; Fault diagnosis; Limited data; Rotating machinery; Transfer learning; Fault diagnosis; Rotating machinery; Transfer learning; Limited data; Adversarial domain adaptation
• ISSN: 1568-4946; 1872-9681
• DOI: 10.1016/j.asoc.2021.107150

Investigation of deep transfer learning on machinery fault diagnosis is helpful to overcome the limitations of a large volume of training data, and accelerate the practical applications of diagnostic algorithms. However, previous reported methods, mainly including parameter transfer and domain adaptation, still require a few labeled or massive unlabeled fault samples, which are not always available. In general, only extremely limited fault data, namely sparse data (single or several samples), can be obtained, and the labeling is also easy to be processed. This paper presents a novel framework for disposing the problem of transfer diagnosis with sparse target data. In consideration of the unclear data distribution described by the sparse data, the main idea is to pair the source and target data with the same machine condition and conduct individual domain adaptation so as to alleviate the lack of target data, diminish the distribution discrepancy as well as avoid negative transfer. More impressive, the issue of label space mismatching can be appropriately addressed in our network. The extensive experiments on two case studies are used to verify the proposed method. Comprehensive transfer scenarios, i.e., diverse working conditions and diverse machines, are considered. The thorough evaluation shows that the proposed method presents superior performance with respect to traditional transfer learning methods. •A novel framework for disposing the problem of transfer diagnosis with limited data is presented.•A novel adversarial training network with multiclass domain discriminators is utilized.•The challenges from unclear data distribution described by the sparse data and negative transfer are considered.•The issue of label space mismatching can be appropriately addressed in the proposed method.•The extensive experimental results demonstrate that the proposed method is superior to traditional transfer learning methods.