arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.12450cs.LGcs.AI
收藏

MetaVA:用于基于ECG检测室性心律失常的深度神经网络的课程元学习和预微调

MetaVA: Curriculum Meta-learning and Pre-fine-tuning of Deep Neural Networks for Detecting Ventricular Arrhythmias based on ECGs

National University of Singapore(新加坡国立大学);HeartVoice Medical Technology(心声医疗科技);University of Science and Technology of China(中国科学技术大学);Sir Run Run Shaw Hospital, Zhejiang University School of Medicine(浙江大学医学院附属邵逸夫医院);Peking University(北京大学);Institute of Medical Technology, Health Science Center of Peking University(北京大学医学部医学技术研究院)

Wenrui Zhang, Shijia Geng, Zhaoji Fu, Linlin Zheng, Chenyang Jiang, Shenda Hong

摘要:

室性心律失常是心脏性猝死的主要原因。开发基于心电图(ECG)检测VA的机器学习方法可以帮助挽救人们的生命。然而,开发用于ECG的这种机器学习模型是具有挑战性的,因为以下原因:1)来自不同受试者的组水平多样性和2)来自单个受试者的不同时刻的个体水平多样性。在本研究中,我们的目标是解决这些问题的预训练和微调阶段。在预训练阶段,我们提出了一种新的模型不可知的元学习(MAML)与课程学习(CL)的方法来解决组级的多样性。MAML有望更好地从大型数据集中转移知识,并仅使用少量记录来快速调整模型以适应新的人。CL应该通过从简单到困难的任务进行元学习来进一步改进MAML。对于微调阶段,我们提出了改进的预微调,以解决个人水平的多样性。我们使用三个公开的ECG数据集的组合进行实验。结果表明,我们的方法优于比较方法的所有评价指标。消融研究表明,MAML和CL可以帮助更均匀地执行,并且预微调可以更好地将模型拟合到训练数据。

英文摘要:

Ventricular arrhythmias (VA) are the main causes of sudden cardiac death. Developing machine learning methods for detecting VA based on electrocardiograms (ECGs) can help save people's lives. However, developing such machine learning models for ECGs is challenging because of the following: 1) group-level diversity from different subjects and 2) individual-level diversity from different moments of a single subject. In this study, we aim to solve these problems in the pre-training and fine-tuning stages. For the pre-training stage, we propose a novel model agnostic meta-learning (MAML) with curriculum learning (CL) method to solve group-level diversity. MAML is expected to better transfer the knowledge from a large dataset and use only a few recordings to quickly adapt the model to a new person. CL is supposed to further improve MAML by meta-learning from easy to difficult tasks. For the fine-tuning stage, we propose improved pre-fine-tuning to solve individual-level diversity. We conduct experiments using a combination of three publicly available ECG datasets. The results show that our method outperforms the compared methods in terms of all evaluation metrics. Ablation studies show that MAML and CL could help perform more evenly, and pre-fine-tuning could better fit the model to training data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑