We use cookies to improve your experience with our site.

融合双向LSTM与多任务时间注意力的心电图信号分类方法

Electrocardiogram Signal Classification Based on Bidirectional LSTM and Multi-Task Temporal Attention

  • 摘要:
    研究背景 心电图(ECG)信号的分析对于预测心律失常和预防心血管疾病具有重要意义。然而,心电图数据具有高噪声和高度时间相关的时间特征,这对机器学习算法在分类任务中提出了挑战。
    目的 为解决噪声和高度相关的心电图数据带来的分类挑战,我们提出了一种新的深度学习心电信号分类模型BiLMTA。
    方法 该文提出了基于双向长短期记忆(BiLSTM)网络和多任务时间注意力的心电图信号分类,命名为BiLMTA,它结合了时间序列编码模块和多任务解码模块。首先,将滤波后的心电图信号输入编码器模块,利用双向LSTM对前后时间步长之间的时间依赖关系进行建模。随后,多任务解码模块处理两个关键任务:一个负责主任务,即心电信号的分类任务;另一个为预测正常和异常心跳的辅助任务。为更好地捕获心电信号中的关键信息,提高对异常模式的敏感性,文章引入了一种时间注意力机制。此外,为了促进两个任务之间的交互学习,提出了一种新的关节损失函数。该函数综合考虑了主任务和辅助任务的交叉熵损失,以确保两个任务能够共同优化模型,提高分类结果的准确性。
    结果 通过在 MIT-BIH 心律失常数据集上的对比实验,我们将所提出的模型与近二十年来的多种传统机器学习算法和深度学习方法进行了性能比较。实验结果表明,本模型在测试精度上超过了大多数现有方法,有效验证了其优越性。在 PTB 诊断心电图数据集上的实验结果显示,本模型的测试准确率达到 99.92%,优于大多数已有方法,充分体现了模型良好的泛化能力。不同数据集的对比实验结果验证了 BiLMTA 模型在心电信号处理领域的有效性与优势,为未来在心律失常与心肌梗死的诊断和分类研究中提供了重要的参考与启示。然而,由于该数据集的规模相对有限,未来仍需在更多的数据集上进一步验证其可靠性。
    结论 本文提出了一种新的多任务心电图信号分类模型BiLMTA。为了解决现有机器学习算法在心电图信号分析中的局限性,该模型采用了时间编码模块和多任务解码模块。BiLMTA采用离散小波变换对心电信号进行去噪,并结合双向LSTM和时间注意力机制,提高模型对关键数据段的注意力,增强对异常模式的检测灵敏度。实验结果表明,该方法在心律失常和心肌梗死的分类方面优于传统机器学习和一些先进的深度学习方法。未来工作包括在实验中使用更多的心电图数据集来验证模型的泛化能力和可靠性,从而更好地将其应用于临床实践。

     

    Abstract: The analysis of electrocardiogram (ECG) signals is of great significance in predicting arrhythmias and preventing cardiovascular diseases. However, ECG data has high noise and highly temporally correlated temporal characteristics, which challenges machine learning algorithms in classification tasks. This paper proposes an ECG signal classification method based on the bidirectional long short-term memory (BiLSTM) network and multi-task temporal attention, named BiLMTA, which combines a time series encoding module and a multi-task decoding module. Specifically, the first step of the method involves feeding the filtered ECG signals into the encoder module, in which BiLSTM is utilized to model temporal dependencies between preceding and succeeding time steps. Subsequently, the multi-task decoding module handles two key tasks: one is responsible for the main task, the classification task of ECG signals; the other serves as an auxiliary task to predict normal and abnormal heartbeats. To better capture the key information in ECG signals and improve sensitivity to abnormal patterns, a temporal attention mechanism is introduced. In addition, in order to promote the interactive learning between two tasks, a new joint loss function is proposed. This loss function comprehensively considers the cross entropy loss of the main task and auxiliary task to ensure that the two tasks can jointly optimize the model and improve the accuracy of classification results. The experimental results show that the BiLMTA method performs well on two publicly available datasets, surpassing traditional methods and state-of-the-art deep learning methods.

     

/

返回文章
返回