Automated detection of heart valve disorders with time-frequency and deep features on PCG signals

Abstract

Heart valve diseases (HVDs) can cause cardiac arrhythmias, heart attacks, and sudden cardiac death if not diagnosed early. Therefore, the detection of HVDs is critical in order to avoid heart-related mortality. The focus of this research is to establish an efficient computer-aided diagnosis approach that detects HVDs using phono-cardiogram (PCG) signals. The proposed approach uses traditional time-frequency and deep features with ma-chine learning models. The time-frequency features are extracted from non-linear measurements using discrete wavelet transform (DWT), wavelet packet transform (WPT), perceptual wavelet packet transform (PWPT) and empirical mode decomposition (EMD) methods. Deep features are extracted from VGG16, ResNet50 and MobileNetV2 pre-trained CNN models, and multilayer extreme learning machine (ML-ELM) model using scalo-gram images of PCG signals. Recursive feature elimination (RFE) algorithm is applied to all features and the most distinctive features are selected. Experimental results show that the PWPT + EMD features selected by RFE and the random forest (RF) classification model achieve the highest performance with accuracy of 99.4%, Matthews correlation coefficient (MCC) and G-mean of 99.3%. In another proposed approach, ML-ELM deep features selected by RFE algorithm and RF classification model provide accuracy and G-mean of 98.9%, and MCC values of 98.6%. It was observed that the time-frequency features have outperformed compared to deep features for the detection of HVDs. The proposed approach is compared with the existing studies and it has obtained higher performance values than the approaches using the same database. The proposed approach can be considered as an easily integrated system on the embedded platform.