WBUN: an interpretable convolutional neural network with wavelet basis unit embedded for fault diagnosis

Abstract

Convolutional Neural Network (CNN) is extensively applied in mechanical system fault diagnosis. However, the absence of transparent decision mechanisms in CNNs hinders credibility. To address these challenges, this paper proposes an interpretable wavelet basis unit convolutional network (WBUN). This network incorporates meticulously designed wavelet basis unit (WBU) functions into convolutional layer, creating the interpretable wavelet basis unit convolutional (WBUConv) layer. Convolutional kernels with clear physical significance enable the WBUConv layer to extract fault-related features in both time and frequency domains, enhancing diagnostic performance, and interpreting the CNN’s attention frequency along with the convolutional kernel’s training outcomes. In this paper, three WBU functions are designed to construct the corresponding WBUNs, and their effectiveness and interpretability are verified through three sets of mechanical fault diagnosis experiments. Meanwhile, experimental results demonstrate the WBUConv layer’s remarkable advantages in noise robustness, convergence speed, and strong generalization ability.