医学文献 >>
  • 检索发现
  • 增强检索
知识库 >>
  • 临床诊疗知识库
  • 中医药知识库
评价分析 >>
  • 机构
  • 作者
默认
×
热搜词:
换一批
论文 期刊
取消
高级检索

检索历史 清除

基于知识蒸馏和标签相关性的多标签心电图分类

Multi-label ECG Classification Based on Knowledge Distillation and Label Relevance

摘要心电图信号分类中,患者往往同时患有多种疾病,多标签分类更具挑战性和实际意义.深度学习中,训练集标签数目差异大,标签样本不均衡,导致模型易忽视样本数目少的类别.标签相关性有利于解决标签类别不均衡问题,本研究利用Student Model压缩Teacher Model,Student Model的损失函数基于真实标签及Teacher Model各层输出.基于知识蒸馏模型,提出了一种标签相关性矩阵构建方法,利用余弦相似度寻找标签间关联,对预测概率进行修正.合肥-天池数据集包含18 232条原始心电记录,其中,包含38条左束支传导阻滞数据.PTB_XL数据集包含21 836条原始心电数据,包含51条左束支传导阻滞原始心电数据.新的模型改善了标签类别不均衡问题,对于合肥-天池数据集,左束支传导阻滞的F1分数由0.799提升至0.837;起搏心律,由0.862提升至0.876;一度房室传导阻滞,F1分数由0.861提升至0.879,高于ECGNet的0.535和Acharya卷积神经网络的0.810.对于PTB_XL数据集,左束支传导阻滞的F1分数由0.679提升到0.857;右心室肥厚,由0.698提升至0.745;完全性左束支传导阻滞,由0.649提升至0.887.同时,合肥-天池数据集上Student Model的F1分数为0.816,Teacher Model的 F1 分数为 0.824.PTB_XL 上 Student Model 的 F1 分数为 0.876,Teacher Model 的 F1 分数为 0.873.Student Model在模型大小上仅为Teacher Model的0.62倍.模型改善了模型改善了长尾分布问题,提升了对少样本标签的预测能力,同时简化参数量,节约计算资源.

更多
广告
栏目名称
DOI 10.3969/j.issn.0258-8021.2025.06.003
发布时间 2026-02-03(万方平台首次上网日期,不代表论文的发表时间)
基金项目
国家自然科学基金(62171123) 国家自然科学基金(62101129) 国家自然科学基金(62211530112)
  • 浏览4
  • 下载0
中国生物医学工程学报

加载中!

相似文献

  • 中文期刊
  • 外文期刊
  • 学位论文
  • 会议论文

加载中!

加载中!

加载中!

加载中!

法律状态公告日 法律状态 法律状态信息

特别提示:本网站仅提供医学学术资源服务,不销售任何药品和器械,有关药品和器械的销售信息,请查阅其他网站。

  • 客服热线:4000-115-888 转3 (周一至周五:8:00至17:00)

  • |
  • 客服邮箱:yiyao@wanfangdata.com.cn

  • 违法和不良信息举报电话:4000-115-888,举报邮箱:problem@wanfangdata.com.cn,举报专区

官方微信
万方医学小程序
new医文AI 翻译 充值 订阅 收藏 移动端

官方微信

万方医学小程序

使用
帮助
Alternate Text
调查问卷