Few-shot condition diagnosis of rolling bearing using adversarial transfer network with class aggregation-guided

Abstract

For the challenge of fault identification under limited labeled data in engineering applications,a novel adversarial transfer network with class aggregation-guided (ATN-CA) is proposed for few-shot condition diagnosis of bearings. The ATN-CA can focus on the discrepancy features of bearings by the proposed local discrepancy feature representation, avoiding that the features extracted by a single neural network may omit important fault information. Moreover, the further proposed class aggregation-guided strategy uses the semantic information of signals to guide the dynamic adaptation of marginal and conditional distributions of source and target data, which shortens the distribution distance of the same category in different domains, thus completing the transfer diagnosis. By comparing with other current methods on the artificial and real bearing fault datasets,results show this proposed method has the highest test precision and the smallest accuracy deviation in bearing transfer diagnosis.