用Gabor字典与弹性网正则化提升心音信号分类精度,达98.95%。
Elastic Net Regularization and Gabor Dictionary for Classification of Heart Sound Signals using Deep Learning
- 结合Gabor字典与弹性网正则化生成高分辨时频特征矩阵
- 最优模型下分类准确率达98.95%,优于传统方法
- 适合心脏病智能诊断、医疗信号处理研究者参考
本文通过优化时频原子分辨率与拟合模型正则化,提升心音信号表征能力。基于新生成的时频特征矩阵,评估深度学习网络对五类心脏瓣膜病的分类性能。采用包含一维卷积神经网络(1D CNN)与长短期记忆(LSTM)层的双架构,分别以随机梯度下降带动量(SGDM)和自适应矩估计(ADAM)训练。实验使用包含五种心瓣膜病变的心音数据库,最佳结果为:在2D CNN + LSTM架构上使用ADAM优化,配合高时间低频率分辨率的Gabor原子字典及稀疏约束模型,实现98.95%的分类准确率。
原文摘要 · Abstract (English)
In this article, we propose the optimization of the resolution of time-frequency atoms and the regularization of fitting models to obtain better representations of heart sound signals. This is done by evaluating the classification performance of deep learning (DL) networks in discriminating five heart valvular conditions based on a new class of time-frequency feature matrices derived from the fitting models. We inspect several combinations of resolution and regularization, and the optimal one is that provides the highest performance. To this end, a fitting model is obtained based on a heart sound signal and an overcomplete dictionary of Gabor atoms using elastic net regularization of linear models. We consider two different DL architectures, the first mainly consisting of a 1D convolutional neural network (CNN) layer and a long short-term memory (LSTM) layer, while the second is composed of 1D and 2D CNN layers followed by an LSTM layer. The networks are trained with two algorithms, namely stochastic gradient descent with momentum (SGDM) and adaptive moment (ADAM). Extensive experimentation has been conducted using a database containing heart sound signals of five heart valvular conditions. The best classification accuracy of $98.95\%$ is achieved with the second architecture when trained with ADAM and feature matrices derived from optimal models obtained with a Gabor dictionary consisting of atoms with high-time low-frequency resolution and imposing sparsity on the models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。