Few-shot Bearing Fault Diagnosis using Adaptive Detail Convolution and Global KAN-Transformer with Mahalanobis Distance
摘要
In bearing failure diagnosis, accurate fault detection is essential for ensuring the safe operation of machinery. However, acquiring enough labeled fault data for training is expensive and challenging.
ObjectiveThis paper aims to address key challenges in few-shot fault diagnosis by proposing a novel framework incorporating an Adaptive Detail Convolution and a dual-branch architecture.
MethodsThe proposed framework consists of three main components: an Adaptive Detail Convolution module for enhanced feature extraction, a Global KAN-Transformer learning branch to model long-range dependencies between global features, and an adaptive-regularized Mahalanobis distance module to measure the similarity of local features between support and query samples.
ResultsExperimental results on the CWRU dataset show that the proposed framework significantly improves classification performance in terms of accuracy, robustness, and efficiency.
ConclusionThe proposed solution offers an effective approach for few-shot bearing fault diagnosis and outperforms existing methods in both accuracy and computational efficiency.