arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.18592eess.SPcs.LG

面向12导联心电图分类的深度神经网络泛化与微调

Deep Neural Networks Generalization and Fine-Tuning for 12-lead ECG Classification

Ivannikov Institute for System Programming of the Russian Academy of Sciences(俄罗斯科学院伊万尼科夫系统编程研究所);Russian-Armenian University(俄罗斯-亚美尼亚大学);Novgorod State University(诺夫哥罗德国立大学)

Aram Avetisyan, Shahane Tigranyan, Ariana Asatryan, Olga Mashkova, Sergey Skorik, Vladislav Ananev, Yury Markin

AI总结:

本文提出先在大规模多源心电图数据集上预训练深度神经网络,再针对特定小数据集微调的方法,以提升12导联心电图心脏疾病分类的跨数据集泛化性能,并在TIS和PTB-XL数据集上验证了其优于单一小数据集训练的效果。

摘要:

大量研究致力于基于12导联心电图(ECG)记录并利用深度学习方法诊断心脏疾病。这些研究通常使用在规模和参数上各不相同的特定数据集,例如患者元数据、标注心电图的医生数量、心电图记录设备类型、数据预处理技术等。众所周知,在一个心电图数据集上训练的高质量深度神经网络并不一定能在另一个数据集或临床环境中表现良好。在本文中,我们提出了一种方法,通过在各种数据集上训练神经网络,并针对特定数据集进行进一步微调,从而无论使用何种数据集都能提高心脏疾病预测的质量。为了展示其适用性,我们分别在包含来自多家医院的各种心电图记录的大型私有数据集TIS和相对较小的公开数据集PTB-XL上训练了不同的神经网络。我们证明,在大型数据集上训练网络并在来自另一来源的小型数据集上进行微调,其性能优于仅在单个小型数据集上训练的网络。我们还展示了深度神经网络的泛化能力如何有助于提高更多疾病的分类质量。

英文摘要:

Numerous studies are aimed at diagnosing heart diseases based on 12-lead electrocardiographic (ECG) records using deep learning methods. These studies usually use specific datasets that differ in size and parameters, such as patient metadata, number of doctors annotating ECGs, types of devices for ECG recording, data preprocessing techniques, etc. It is well-known that high-quality deep neural networks trained on one ECG dataset do not necessarily perform well on another dataset or clinical settings. In this paper, we propose a methodology to improve the quality of heart disease prediction regardless of the dataset by training neural networks on a variety of datasets with further fine-tuning for the specific dataset. To show its applicability, we train different neural networks on a large private dataset TIS containing various ECG records from multiple hospitals and on a relatively small public dataset PTB-XL. We demonstrate that training the networks on a large dataset and fine-tuning it on a small dataset from another source outperforms the networks trained only on one small dataset. We also show how the ability of a deep neural networks to generalize allows to improve classification quality of more diseases.

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

↑