Feb, 2024

心电图诊断中的迁移学习:有效吗?

TL;DR深度学习在心电图诊断中的应用受限于现实场景中大规模标记数据集的稀缺,因此使用迁移学习来利用从更大数据集中学到的特征。然而,迁移学习能够始终优于从头开始训练的普遍假设尚未得到系统验证。本研究通过调查不同心电图数据集和深度神经网络的微调性能与从头开始训练的性能的比较,首次进行了关于迁移学习在多标签心电图分类中效果的大规模实证研究,结果显示对于小规模下游数据集,微调是一种较好的选择;然而,当数据集足够大时,从头开始训练也能够取得可比较的性能,尽管需要更长的训练时间来迎头赶上。此外,我们发现迁移学习与卷积神经网络相比较适用于时序心电图应用中最常见的两种结构:卷积神经网络和循环神经网络。我们的结果强调了迁移学习在心电图诊断中的重要性,但根据可用数据量,研究人员可能选择不使用迁移学习,考虑到与预训练相关的不可忽视的成本。