【AI赋能医学】基于深度学习和HRV特征的多类别心电图分类

一、数据集简介

论文中使用了来自三类不同心电图记录的162条数据,这些数据来自三个公开的数据库:

MIT-BIH 心律失常数据库 (ARR)

96条记录,主要包含不同类型的心律失常样本。

MIT-BIH 正常窦性心律数据库 (NSR)

36条记录,包含健康人的正常心电图信号。

Beth Israel Deaconess Medical Center 心力衰竭数据库 (CHF)

30条记录,包含充血性心力衰竭患者的心电图。

所有心电图信号都来自导联II和VI,并且经过统一处理,将采样频率调整为128 Hz,消除了偏移效应。每条心电图信号都由心脏病专家手动标注,确保了数据的准确性。

再次说明:

该方法的有效性通过162条心电图(ECG)记录进行测试,这些记录分为三类:心律失常(ARR)、充血性心力衰竭(CHF)和正常窦性心律(NSR)。这些记录来自以下公共数据库:

MIT-BIH 心律失常数据库

MIT-BIH 正常窦性心律数据库

Beth Israel Deaconess Medical Center 心力衰竭数据库

数据集包含:

96条心律失常记录

30条心力衰竭记录

36条正常窦性心律记录

所有心电信号均来自导联II和导联VI,并且经过重新采样,统一为128 Hz的采样频率,以确保数据一致性并消除偏移效应。每条记录都由多位心脏病专家进行了手动分析和标注。

二、模型实现步骤

按照以下步骤进行模型实现:

  1. 数据预处理
    信号归一化:将心电图信号标准化,消除偏移效应。
    特征提取:
    提取基于心电图(ECG)的特征,如RR间期、P波、QRS波等。
    提取基于心率变异性(HRV)的特征,如时域和频域特征。
    重新采样:将所有信号重采样至128 Hz。
  2. 模型结构
    特征组合:将心电图特征与HRV特征结合,作为多类别分类模型的输入。
    深度学习模型:使用深度学习框架(如CNN、LSTM等)自动提取特征。论文中建议使用卷积神经网络(CNN)处理心电图信号,并结合心率变异性特征进行分类。
    分类模型:利用一个全连接层进行多类别分类,模型的输出对应ARR、CHF和NSR三类。
  3. 模型训练
    使用训练集对模型进行训练,优化损失函数(如交叉熵损失函数)。
    采用K折交叉验证来评估模型的性能。
  4. 模型评估
    使用准确率、灵敏度、特异性等评价指标评估模型性能。
    对ARR、CHF和NSR三类心电图数据进行多类别分类,检查模型的泛化能力。
  5. 实现步骤
    使用Python、TensorFlow、Keras等深度学习库实现模型。
    将数据集分为训练集和测试集,进行训练、评估。

参考

https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10542398/

https://www.mdpi.com/1424-8220/23/3/1697

相关推荐
从零开始学习人工智能1 小时前
GPUStack:开源GPU集群管理工具,解锁AI模型高效运行新可能
人工智能·开源
C嘎嘎嵌入式开发1 小时前
(六)机器学习之图卷积网络
人工智能·python·机器学习
Msshu1232 小时前
PD快充诱骗协议芯片XSP25支持PD+QC+FCP+SCP+AFC协议支持通过串口读取充电器功率信息
人工智能
一RTOS一4 小时前
东土科技连投三家核心企业 发力具身机器人领域
人工智能·科技·机器人·具身智能·鸿道实时操作系统·国产嵌入式操作系统选型
ACP广源盛139246256736 小时前
(ACP广源盛)GSV1175---- MIPI/LVDS 转 Type-C/DisplayPort 1.2 转换器产品说明及功能分享
人工智能·音视频
胡耀超6 小时前
隐私计算技术全景:从联邦学习到可信执行环境的实战指南—数据安全——隐私计算 联邦学习 多方安全计算 可信执行环境 差分隐私
人工智能·安全·数据安全·tee·联邦学习·差分隐私·隐私计算
停停的茶8 小时前
深度学习(目标检测)
人工智能·深度学习·目标检测
Y200309168 小时前
基于 CIFAR10 数据集的卷积神经网络(CNN)模型训练与集成学习
人工智能·cnn·集成学习
老兵发新帖8 小时前
主流神经网络快速应用指南
人工智能·深度学习·神经网络
AI量化投资实验室8 小时前
15年122倍,年化43.58%,回撤才20%,Optuna机器学习多目标调参backtrader,附python代码
人工智能·python·机器学习