Multi-sensor data fusion-enabled lightweight convolutional double regularization contrast transformer for aerospace bearing small samples fault diagnosis

Abstract

Aiming at the problems of low information utilization and lack of feature mining capability in multi-sensor fusion networks, this study presents a multi-sensor data fusion-enabled lightweight convolutional double regularization contrast transformer for aerospace bearing small samples fault diagnosis. Firstly, a metric termed integrated cliff entropy is devised to assign weights to vibration signals from diverse sensor channels. It aims to enhance the cyclic impulse characteristics within the fused signals, thereby facilitating more precise fault identification. Secondly, a lightweight Diwaveformer architecture is constructed as the backbone of contrast learning. It enables the global and local features of faulty signals to be comprehensively extracted with less computational effort. Finally, a double contrast loss is constructed to optimize the distribution of intra-class and inter-class features to improve the fault identification ability of the network with small samples. Additionally, a discard regularization method is designed to remove the projection head during the contrast learning process, further advancing the model lightweight. Our method achieved accuracies of 95.54% and 92.56% on two aerospace bearing datasets with extremely sparse training samples, which proved its superior performance.