Le-LWTNet: A Learnable Lifting Wavelet Convolutional Neural Network for Heart Sound Abnormality Detection

Abstract

Automatic heart sound abnormality detection plays a vital role in the preliminary diagnosis of cardiovascular diseases. Many handcraft-designed or learning-based methods have been proposed in recent years. However, due to the influence of environment, the divergence of different stethoscopes, and data collection protocol, the pattern of heart sound signals are so complex that fixed pattern feature extraction or learning features directly from the signal can not enough lead to final accurate classification. For this issue, a learnable lifting wavelet transform block (Le-LWT), which embeds the trainable convolutional neural network (CNN) into the lifting wavelet transform, is proposed in this paper. Le-LWT can utilize the non-linear learning ability of CNN while maintaining the multi-resolution time-frequency analysis ability of wavelet transform, as well as more interpretation than the deep networks designed as black boxes. Based on Le-LWT module, we propound an end-to-end Le-LWTNet that has stronger non-linear characterization capabilities and few parameters for automatic abnormality detection of the heart sound. Experimental evaluations are performed on a 10-fold cross-validation task using the 2016 PhysioNet/CinC Challenge dataset and the new publicly available pediatric heart sound dataset we collected. Results demonstrate that the proposed method excels the state-of-the-art models both in abnormality detection and parameter consumption. Moreover, the proposed method can give an interpretation of the current classification results, which will help doctors do a second review.