A Multi-Task Group Bi-LSTM Networks Application on Electrocardiogram Classification

Qiu-Jie Lv1, Hsin-Yi Chen1, Wei-Bin Zhong1

  • 11Artificial Intelligence Medical Center, School of Intelligent Systems EngineeringSun Yat-sen UniversityShenzhen510275China.

Insights

A new multi-task group bidirectional long short-term memory (MTGBi-LSTM) framework effectively recognizes cardiovascular diseases (CVD) from electrocardiogram (ECG) signals. This AI tool offers reliable computer-aided diagnosis for CVD, improving patient outcomes.

Area of Science:

  • Artificial Intelligence
  • Biomedical Engineering
  • Cardiology

Background:

  • Cardiovascular diseases (CVD) represent a significant global health burden, being the primary cause of mortality worldwide.
  • Electrocardiogram (ECG) analysis is a crucial, efficient method for assessing various CVDs.
  • Accurate and timely diagnosis of CVDs remains a critical challenge in clinical practice.

Purpose of the Study:

  • To introduce a novel Multi-Task Group Bidirectional Long Short-Term Memory (MTGBi-LSTM) framework for intelligent recognition of multiple CVDs.
  • To leverage multi-lead ECG signals for enhanced CVD detection and diagnosis.
  • To develop an effective tool for computer-aided diagnosis (CAD) of cardiovascular diseases.

Main Methods:

  • The proposed MTGBi-LSTM framework integrates Group Bi-LSTM (GBi-LSTM) and Residual Group Convolutional Neural Network (Res-GCNN) for dual feature extraction from ECG spatial and time-series data.
  • GBi-LSTM incorporates Global and Intra-Group Bi-LSTM components to analyze individual ECG lead features and inter-lead relationships.
  • An attention mechanism integrates multi-lead ECG information, enhancing feature discriminability, while multi-task learning and a dynamic weighted loss function address disease associations and class imbalance.

Main Results:

  • The MTGBi-LSTM model was evaluated on over 170,000 clinical 12-lead ECG analyses.
  • The framework achieved high performance metrics: 88.86% accuracy, 90.67% precision, 94.19% recall, and 92.39% F1-score.
  • These results demonstrate the reliable ECG analysis capabilities of the MTGBi-LSTM method.

Conclusions:

  • The MTGBi-LSTM framework provides a robust and effective approach for the computer-aided diagnosis of cardiovascular diseases using ECG signals.
  • The study highlights the potential of deep learning models in improving the efficiency and accuracy of CVD detection.
  • The developed tool offers a promising solution for enhancing clinical decision-making in cardiology.
Abstract