An anti-noise algorithm based on locally linear embedding and weighted XGBoost for fault diagnosis of T/R module

Abstract

Due to the noise accompanied with fault signals, it is challenging to identify the discriminant information and the local geometric feature from the complex fault data for enhancing fault diagnosis accuracy. To address this challenge, this work proposed an anti-noise algorithm based on locally linear embedding integrated with diffusion distance and maximum correntropy criterion (DMLLE). In DMLLE, diffusion distance was adopted instead of the Euclidean distance for neighborhood construction. Meanwhile, the optimal weights were updated to reveal local geometry information based on the loss function of the maximum correntropy criterion. Subsequently, DMLLE is eventually developed to further restrain noise embedding into raw signals and obtain low dimensional features. Furthermore, weighted extreme gradient boosting is used to map the low dimensional features to the types of faults, which easily implements fault pattern recognition. Finally, two synthetic manifold datasets and fault data acquired from the transmit/receive (T/R) module are used to validate the performance of the proposed diagnosis methodology. Compared with the existing methods, the proposed diagnosis methodology generates a smoother flow structure by preserving the local neighborhood of the dataset with noises and realizes a higher accuracy of 94.41% on the T/R module dataset, which outperforms 3%–9% better than other classification models. Therefore, it can be concluded that the proposed diagnosis methodology can effectively extract intrinsic fault features by weakening the influence aroused by noise and achieve superior accuracy in fault diagnosis by addressing the problem of small sample size.