论文标题
MLBF-NET:使用12 LEAD ECG的多级心律失常分类的多线分支融合网络
MLBF-Net: A Multi-Lead-Branch Fusion Network for Multi-Class Arrhythmia Classification Using 12-Lead ECG
论文作者
论文摘要
使用12铅心电图(ECG)信号自动心律失常检测在早期预防和诊断心血管疾病中起关键作用。在先前关于自动心律失常检测的研究中,大多数方法将ECG的12个导线串成矩阵,然后将矩阵输入到各种特征提取器或深神经网络中,以提取有用的信息。在这样的框架下,这些方法具有提取12铅ECG的全面特征(称为完整性)的能力,因为在训练过程中,每个铅的信息在训练过程中彼此相互作用。但是,在12个线索中忽略了12条线索中不同铅特异性特异性特征(称为多样性),从而导致12铅ECG的信息学习不足。为了最大程度地利用多铅心电图的信息学习,应考虑具有完整性和具有多样性的特定于铅特征的综合特征的信息融合。在本文中,我们提出了一个新型的多核分支融合网络(MLBF-NET)结构,以通过将多损失优化整合到共同学习多样性和多铅ECG的完整性中,以进行心律失常分类。 MLBF-NET由三个组成部分组成:1)多个铅特异性分支,用于学习多铅ECG的多样性; 2)通过串联所有分支的输出特征图来学习多元素ECG的完整性,跨铅特征融合; 3)所有单个分支和串联网络的多损失合作式化。我们展示了2018年中国生理信号挑战的MLBF-NET,该挑战是一个开放的12铅ECG数据集。实验结果表明,MLBF-NET的平均$ F_1 $得分为0.855,达到了心律失常最高的分类性能。提出的方法从信息融合的角度提供了多铅ECG分析的有希望的解决方案。
Automatic arrhythmia detection using 12-lead electrocardiogram (ECG) signal plays a critical role in early prevention and diagnosis of cardiovascular diseases. In the previous studies on automatic arrhythmia detection, most methods concatenated 12 leads of ECG into a matrix, and then input the matrix to a variety of feature extractors or deep neural networks for extracting useful information. Under such frameworks, these methods had the ability to extract comprehensive features (known as integrity) of 12-lead ECG since the information of each lead interacts with each other during training. However, the diverse lead-specific features (known as diversity) among 12 leads were neglected, causing inadequate information learning for 12-lead ECG. To maximize the information learning of multi-lead ECG, the information fusion of comprehensive features with integrity and lead-specific features with diversity should be taken into account. In this paper, we propose a novel Multi-Lead-Branch Fusion Network (MLBF-Net) architecture for arrhythmia classification by integrating multi-loss optimization to jointly learning diversity and integrity of multi-lead ECG. MLBF-Net is composed of three components: 1) multiple lead-specific branches for learning the diversity of multi-lead ECG; 2) cross-lead features fusion by concatenating the output feature maps of all branches for learning the integrity of multi-lead ECG; 3) multi-loss co-optimization for all the individual branches and the concatenated network. We demonstrate our MLBF-Net on China Physiological Signal Challenge 2018 which is an open 12-lead ECG dataset. The experimental results show that MLBF-Net obtains an average $F_1$ score of 0.855, reaching the highest arrhythmia classification performance. The proposed method provides a promising solution for multi-lead ECG analysis from an information fusion perspective.