WaveletKernelNet: An Interpretable Deep Neural Network for Industrial Intelligent Diagnosis

Abstract

Convolutional neural network (CNN), with ability of feature learning and nonlinear mapping, has demonstrated its effectiveness in prognostics and health management (PHM). However, explanation on the physical meaning of a CNN architecture has rarely been studied. In this paper, a novel wavelet driven deep neural network termed as WaveletKernelNet (WKN) is presented, where a continuous wavelet convolutional (CWConv) layer is designed to replace the first convolutional layer of the standard CNN. This enables the first CWConv layer to discover more meaningful filters. Furthermore, only the scale parameter and translation parameter are directly learned from raw data at this CWConv layer. This provides a very effective way to obtain a customized filter bank, specifically tuned for extracting defect-related impact component embedded in the vibration signal. In addition, three experimental verification using data from laboratory environment are carried out to verify effectiveness of the proposed method for mechanical fault diagnosis. The results show the importance of the designed CWConv layer and the output of CWConv layer is interpretable. Besides, it is found that WKN has fewer parameters, higher fault classification accuracy and faster convergence speed than standard CNN.

References in corpus (5)

• Improving neural networks by preventing co-adaptation of feature detectors
• Understanding Neural Networks Through Deep Visualization
• Deep Learning Algorithms for Rotating Machinery Intelligent Diagnosis: An Open Source Benchmark Study
• Unsupervised Deep Slow Feature Analysis for Change Detection in Multi-Temporal Remote Sensing Images
• The Mexican Hat Wavelet Family. Application to point source detection in CMB maps

Cited by in corpus (8)

• TFN: An Interpretable Neural Network with Time-Frequency Transform Embedded for Intelligent Fault Diagnosis
• Fully Learnable Deep Wavelet Transform for Unsupervised Monitoring of High-Frequency Time Series
• Filter-informed Spectral Graph Wavelet Networks for Multiscale Feature Extraction and Intelligent Fault Diagnosis
• Iterative Low-rank Network for Hyperspectral Image Denoising
• SigWavNet: Learning Multiresolution Signal Wavelet Network for Speech Emotion Recognition
• Rational Gaussian wavelets and corresponding model driven neural networks
• WaveletInception Networks for on-board Vibration-Based Infrastructure Health Monitoring
• Hard Class Rectification for Domain Adaptation