A few-shot sample augmentation algorithm based on SCAM and DEPS for pump fault diagnosis

In recent years, pumps have become critical components in agriculture, industry, and the military, necessitating extensive development and implementation of the fault diagnosis method. In the majority of existing fault classification models, the connection between performance improvement and the amount of training data remains high, yet real-world samples are difficult to obtain. Combining domain migration theory and sample expansion method, this paper introduces a few-shot learning fault diagnosis method. Employing the T-SNE visualization algorithm, we examine the validity of the self-calibration attention mechanism (SCAM) and distribution edge prediction strategy (DEPS). The accomplishment demonstrated that the proposed algorithm could effectively map the expanded sample space within a separate interval, thereby avoiding the problem of feature aliasing caused by the overlap of sample features among similar categories and significantly enhancing the quality and quantity of training samples. The experimental analysis indicates that the proposed methodology can effectively increase the accuracy of few-shot tasks, especially in the 9way-15shot task, where it maintains a performance of 72 %, which leading the mean accuracy calculated from the others of about 30%. It is believed that much of the work has superior applicability to other few-shot diagnosis cases.