Classification of heart sounds based on the combination of the modified frequency wavelet transform and convolutional neural network.
Yongchao ChenShoushui WeiYatao ZhangPublished in: Medical & biological engineering & computing (2020)
We purpose a novel method that combines modified frequency slice wavelet transform (MFSWT) and convolutional neural network (CNN) for classifying normal and abnormal heart sounds. A hidden Markov model is used to find the position of each cardiac cycle in the heart sound signal and determine the exact position of the four periods of S1, S2, systole, and diastole. Then the one-dimensional cardiac cycle signal was converted into a two-dimensional time-frequency picture using the MFSWT. Finally, two CNN models are trained using the aforementioned pictures. We combine two CNN models using sample entropy (SampEn) to determine which model is used to classify the heart sound signal. We evaluated our model on the heart sound public dataset provided by the PhysioNet Computing in Cardiology Challenge 2016. Experimental classification performance from a 10-fold cross-validation indicated that sensitivity (Se), specificity (Sp) and mean accuracy (MAcc) were 0.95, 0.93, and 0.94, respectively. The results showed the proposed method can classify normal and abnormal heart sounds with efficiency and high accuracy. Graphical abstract Block diagram of heart sound classification.