基于对数谱图的深度学习心音分类

本文涉及的产品
实时计算 Flink 版,1000CU*H 3个月
实时数仓Hologres,5000CU*H 100GB 3个月
智能开放搜索 OpenSearch行业算法版,1GB 20LCU 1个月
简介: 这是一篇很有意思的论文,他基于心音信号的对数谱图,提出了两种心率音分类模型,我们都知道:频谱图在语音识别上是广泛应用的,这篇论文将心音信号作为语音信号处理,并且得到了很好的效果。

对心音信号进行一致长度的分帧,提取其对数谱图特征,论文提出了长短期记忆(LSTM)和卷积神经网络(CNN)两种深度学习模型,根据提取的特征对心跳声进行分类。

心音数据集

影像学诊断包括心脏核磁共振成像(MRI)、CT扫描、心肌灌注成像。这些技术的缺点也很明显对现代机械、专业人员的要求高,诊断时间长。

论文使用的是公共数据集,由1000个。wav格式的信号样本组成,采样频率为8 kHz。数据集分为5类,包括1个正常类(N)和4个异常类:主动脉瓣狭窄(AS)、二尖瓣反流(MR)、二尖瓣狭窄(MS)和二尖瓣脱垂(MVP)。

主动脉瓣狭窄(AS)是指主动脉瓣太小、狭窄或僵硬。主动脉瓣狭窄的典型杂音是高音调的“菱形”杂音。

二尖瓣返流(MR)是指心脏的二尖瓣没有正常关闭,导致血液回流到心脏而不是被泵出。听诊胎儿心脏时,S1可能很低(有时很响)。直到S2,杂音的音量增加。由于S3后二尖瓣急流,可听到短而隆隆声的舒张中期杂音。

二尖瓣狭窄(MS)是指二尖瓣受损不能完全打开。心音听诊显示二尖瓣狭窄早期S1加重,严重二尖瓣狭窄时S1软。随着肺动脉高压的发展,S2音将被强调。纯多发性硬化症患者几乎没有左室S3。

二尖瓣脱垂(MVP)是指在心脏收缩期二尖瓣小叶脱垂至左心房。MVP通常是良性的,但并发症包括二尖瓣反流、心内膜炎和脊索断裂。体征包括收缩期中期的咔嗒声和收缩期晚期的杂音(如果存在反流)。

预处理与特征提取

声音信号有不同的长度。所以需要固定每个记录文件的采样率。长度被裁剪后使声音信号包含至少一个完整的心脏周期。成年人每分钟心跳65-75次,心跳周期约为0.8秒,所以信号样本被裁剪为2.0-s, 1.5-s和1.0-s段。

基于离散傅里叶变换(DFT),将心音信号的原始波形转换为对数谱图。声音信号的DFT y(k)为Eq.(1),对数谱图s定义为Eq.(2)。

式中,N为向量x的长度,ε = 10^(- 6)是一个小偏移量。部分心音样本的波形和对数谱图如下:

深度学习模型

1、LSTM

LSTM模型设计为2层直接连接,然后是3层完全连接。第三个完全连接的层输入softmax分类器。

2、CNN模型

如上图所示,前两个卷积层之后是重叠的最大池化层。第三个卷积层直接连接到第一个全连接层。第二个完全连接的层提供给具有五个类标签的softmax分类器。在每个卷积层之后使用BN和ReLU。

3、训练细节

结果

训练集包含整个数据集的70%,测试集包含其余部分。

当CNN模型片段持续时间为2.0 s时,准确率最高为0.9967;分割时间为1.0 s的LSTM准确率最低为0.9300。

CNN模型的整体准确率分别为0.9967、0.9933和0.9900,片段持续时间分别为2.0 s、1.5 s和1.0 s,而LSTM模型的这三个数字分别为0.9500、0.9700和0.9300。

CNN模型比LSTM模型在各时段的预测精度更高。

混淆矩阵如下:

N类(Normal)的预测正确率最高,在5个案例中达到60个,而MVP类在所有案例中预测正确率最低。

LSTM模型输入时间长度为2.0 s,最长预测时间为9.8631 ms。分类时间为1.0 s的CNN模型预测时间最短,为4.2686 ms。

与其他SOTA比较,一些研究的准确率非常高,但这些研究只进行了两类(正常和异常),而本研究分为五类。

与使用相同数据集的其他研究相比(0.9700),论文研究有了显著提高,最高准确率为0.9967。

论文地址:

Heart Sound Classification Using Deep Learning Techniques Based on Log-mel Spectrogram

https://avoid.overfit.cn/post/c8f5ca920d5c4d7c819b2678eaaf946a

作者:Sik-Ho Tsang

目录
相关文章
|
机器学习/深度学习 编解码 人工智能
人脸表情[七种表情]数据集(15500张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
本数据集包含15,500张已划分、已标注的人脸表情图像,覆盖惊讶、恐惧、厌恶、高兴、悲伤、愤怒和中性七类表情,适用于YOLO系列等深度学习模型的分类与检测任务。数据集结构清晰,分为训练集与测试集,支持多种标注格式转换,适用于人机交互、心理健康、驾驶监测等多个领域。
|
2月前
|
机器学习/深度学习 人工智能 监控
河道塑料瓶识别标准数据集 | 科研与项目必备(图片已划分、已标注)| 适用于YOLO系列深度学习分类检测任务【数据集分享】
随着城市化进程加快和塑料制品使用量增加,河道中的塑料垃圾问题日益严重。塑料瓶作为河道漂浮垃圾的主要类型,不仅破坏水体景观,还威胁水生生态系统的健康。传统的人工巡查方式效率低、成本高,难以满足实时监控与治理的需求。
|
2月前
|
机器学习/深度学习 传感器 人工智能
火灾火焰识别数据集(2200张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
在人工智能和计算机视觉的快速发展中,火灾检测与火焰识别逐渐成为智慧城市、公共安全和智能监控的重要研究方向。一个高质量的数据集往往是推动相关研究的核心基础。本文将详细介绍一个火灾火焰识别数据集,该数据集共包含 2200 张图片,并已按照 训练集(train)、验证集(val)、测试集(test) 划分,同时配有对应的标注文件,方便研究者快速上手模型训练与评估。
火灾火焰识别数据集(2200张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
|
2月前
|
机器学习/深度学习 人工智能 自动驾驶
7种交通场景数据集(千张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
在智能交通与自动驾驶技术快速发展的今天,如何高效、准确地感知道路环境已经成为研究与应用的核心问题。车辆、行人和交通信号灯作为城市交通系统的关键元素,对道路安全与交通效率具有直接影响。然而,真实道路场景往往伴随 复杂光照、遮挡、多目标混杂以及交通信号状态多样化 等挑战,使得视觉识别与检测任务难度显著增加。
|
2月前
|
机器学习/深度学习 人工智能 监控
坐姿标准好坏姿态数据集(图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
坐姿标准好坏姿态数据集的发布,填补了计算机视觉领域在“细分健康行为识别”上的空白。它不仅具有研究价值,更在实际应用层面具备广阔前景。从青少年的健康教育,到办公室的智能提醒,再到驾驶员的安全监控和康复训练,本数据集都能发挥巨大的作用。
坐姿标准好坏姿态数据集(图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
|
2月前
|
机器学习/深度学习 编解码 人工智能
102类农业害虫数据集(20000张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
在现代农业发展中,病虫害监测与防治 始终是保障粮食安全和提高农作物产量的关键环节。传统的害虫识别主要依赖人工观察与统计,不仅效率低下,而且容易受到主观经验、环境条件等因素的影响,导致识别准确率不足。
|
26天前
|
机器学习/深度学习 数据采集 编解码
基于深度学习分类的时相关MIMO信道的递归CSI量化(Matlab代码实现)
基于深度学习分类的时相关MIMO信道的递归CSI量化(Matlab代码实现)
|
机器学习/深度学习 人工智能 文字识别
中药材图像识别数据集(100类,9200张)|适用于YOLO系列深度学习分类检测任务
本数据集包含9200张中药材图像,覆盖100类常见中药材,适用于YOLO系列及主流深度学习模型的图像分类与目标检测任务。数据已划分为训练集(8000张)与验证集(1200张),采用标准文件夹结构和简体中文命名,适配PyTorch、TensorFlow等框架,可用于中药识别系统开发、医学辅助诊断、移动端图像识别App研发及AI科研训练,具备较强的实用性与拓展性。
571 45
|
2月前
|
机器学习/深度学习 自动驾驶 算法
道路表面缺陷数据集(裂缝/井盖/坑洼)(6000张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
随着城市化与交通运输业的快速发展,道路基础设施的健康状况直接关系到出行安全与城市运行效率。长期高强度的使用、气候变化以及施工质量差异,都会导致道路表面出现裂缝、坑洼、井盖下沉及修补不良等缺陷。这些问题不仅影响驾驶舒适度,还可能引发交通事故,增加道路养护成本。
道路表面缺陷数据集(裂缝/井盖/坑洼)(6000张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
|
机器学习/深度学习 人工智能 编解码
AI虫子种类识别数据集(近3000张图片已划分、已标注)|适用于YOLO系列深度学习分类检测任务【数据集分享】
本数据集包含近3000张已划分、标注的虫子图像,适用于YOLO系列模型的目标检测与分类任务。涵盖7类常见虫子,标注采用YOLO格式,结构清晰,适合农业智能化、小样本学习及边缘部署研究。数据来源多样,标注精准,助力AI虫害识别落地应用。