论文标题
METAVA:深层神经网络的课程元学习和预先调整,用于基于ECG检测心室心律不齐
MetaVA: Curriculum Meta-learning and Pre-fine-tuning of Deep Neural Networks for Detecting Ventricular Arrhythmias based on ECGs
论文作者
论文摘要
心室心律不齐(VA)是心脏突然死亡的主要原因。开发用于根据心电图(ECG)检测VA的机器学习方法可以帮助挽救人们的生命。但是,为ECG开发这样的机器学习模型是具有挑战性的,因为以下内容:1)来自不同受试者的群体级多样性; 2)单个主题不同时刻的个人水平多样性。在这项研究中,我们旨在在预训练和微调阶段解决这些问题。对于训练阶段,我们提出了一种新型模型不可知的元学习(MAML),其课程学习(CL)方法可以解决群体级别的多样性。预计MAML将更好地将知识从大型数据集转移,并仅使用几个录音来快速将模型调整给新人。 CL应该通过轻松到困难的任务来进一步改善MAML。对于微调阶段,我们提出改进的预先调整,以解决个体级别的多样性。我们使用三个公开可用的心电图数据集的组合进行实验。结果表明,根据所有评估指标,我们的方法优于比较方法。消融研究表明,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.