Imbalanced Heart Sound Signal Classification Based on Two-Stage Trained DsaNet

Guangyang Tian,Cheng Lian,Zhigang Zeng,Bingrong Xu,Yixin Su,Junbin Zang,Zhidong Zhang,Chenyang Xue
DOI: https://doi.org/10.1007/s12559-022-10009-3
IF: 4.89
2022-03-28
Cognitive Computation
Abstract:Many deaths are caused by heart disease. A phonocardiogram (PCG) reflects the general rule of heart movement, so the analysis of heart sound signals is particularly important. In this paper, we propose a new deep neural network termed DsaNet, which is mainly constructed by depthwise separable convolution and the attention module. DsaNet can directly classify PCG signals without complicated feature engineering processes. To address the long-tail distribution problem in the PCG dataset, we adopt a novel imbalanced learning approach (two-stage training) to train our DsaNet. Specifically, we propose a random cropping operation to increase the amount and diversity of the data in the training stage. We also combine random cropping with the idea of integration to improve test accuracy in the testing stage. Moreover, we study the effectiveness of several attention modules and data balancing methods for improving the performance of DsaNet. To verify the performance of DsaNet, we compare our proposed DsaNet with 7 different baseline models on the public 2016 PhysioNet/CinC Challenge dataset. The experimental results show that the proposed DsaNet can achieve competitive performance for imbalanced PCG signal classification with relatively few parameters and computations. Results obtained prove that our model is effective and efficient. In addition, two-stage training significantly improved the generalization performance of DsaNet.
computer science, artificial intelligence,neurosciences
What problem does this paper attempt to address?