Researchers from a collaborative team including Hua Zhang, Jing Li, and Haiyan Wang have developed a deep learning algorithm that stratifies cardiac arrhythmias into three distinct risk levels: normal, not life-threatening, and life-threatening. Published in the journal BioData Mining in July 2026, the study introduces a model named PaliConvLSTM, which utilizes a palindromic structure of Convolutional Neural Networks (CNN) and Long Short-Term Memory (LSTM) layers to analyze electrocardiogram (ECG) signals. The primary objective of this research is to address the clinical challenge of limited medical resources by providing a refined triage system that allows for optimized allocation of care, ensuring that patients with critical arrhythmias receive immediate intervention while those with milder conditions are monitored appropriately.
Cardiovascular diseases are characterized by sudden onset, high mortality, and high recurrence rates, making early screening and timely intervention essential for patient survival. In clinical practice, ECGs are a vital tool for diagnosing and monitoring these conditions, but the sheer volume of data creates a significant burden on physicians. Due to insufficient medical resources, real-time and refined monitoring of all patients is difficult to implement, leading to potential delays in responding to critical cases. Existing artificial intelligence methods often focus on detecting specific arrhythmia types or use a simple two-level classification of critical versus non-critical, which fails to capture the full spectrum of severity ranging from mild rhythm abnormalities to life-threatening ventricular arrhythmias.
To overcome these limitations, the study proposes a three-level risk stratification scheme that aligns with clinical urgency. The first category, labeled as normal, includes normal sinus rhythm and requires no specific monitoring. The second category, labeled as not life-threatening, encompasses common arrhythmias such as atrial bigeminy, ventricular bigeminy, and second-degree heart block, which do not endanger life in the short term but require close attention. The third category, labeled as life-threatening, includes malignant ventricular arrhythmias such as ventricular flutter, ventricular tachycardia, and ventricular fibrillation, which can cause severe hemodynamic changes and sudden cardiac death, necessitating immediate response.
The core of the algorithm is the PaliConvLSTM model, which adopts a multi-layer fusion architecture with a palindromic sequence of CNN-LSTM-CNN-LSTM-CNN. This structure is designed to comprehensively capture the complex spatiotemporal features of ECG signals. CNNs are effective at extracting local spatial features and detailed morphological information through convolutional and pooling layers, while LSTMs excel at capturing temporal correlations and long-term dependencies within time-series data. By alternating these layers, the model hierarchically couples multi-scale spatial features with temporal rhythm information, allowing for a more accurate representation of the ECG signal than single-layer fusion architectures.
The study utilized three publicly available ECG databases for experimentation: the MIT-BIH Malignant Ventricular Arrhythmia Database (VFDB), the MIT-BIH Arrhythmia Database (MITDB), and the Creighton University Ventricular Tachyarrhythmia Database (CUDB). Data preprocessing involved extracting complete rhythm segments based on annotations, classifying them into the three risk categories, and segmenting them into fixed lengths of 2, 3, 5, 8, and 10 seconds. To verify the generalization performance of the algorithm, the VFDB and CUDB were also merged to form a combined dataset, with the VFDB data down-sampled to match the sampling rate of the CUDB. One-third of the data in each dataset was randomly partitioned as the test set, with the remainder used for training.
Experimental results demonstrated that the PaliConvLSTM algorithm exhibits favorable generalization performance across multiple datasets and segment lengths. On the 2-second ECG segments from the MITDB dataset, the model achieved an accuracy of 99.68% and a specificity of 99.65%. Specificity, which denotes the probability of accurately detecting all abnormal arrhythmia samples, is a critical metric for clinical screening as it reflects the model’s capacity to identify pathological cases. The 10-fold cross-validation on the MITDB dataset yielded an accuracy of 99.62% with a standard deviation of 0.09% for 3-second segments, indicating high stability and precision in the overall classification task.
When compared with standalone CNN and LSTM models, as well as other hybrid architectures like CNN-CNN-LSTM-LSTM and LSTM-LSTM-CNN-CNN, the proposed PaliConvLSTM model outperformed them in accuracy, sensitivity, and specificity. The model also surpassed classical machine learning algorithms, including Decision Trees, Random Forest, Multilayer Perceptron, K-Nearest Neighbors, and Support Vector Machines. While classical algorithms exhibited substantial variability and instability in performance across different datasets and segment lengths, PaliConvLSTM maintained consistent high performance, with accuracy ranges between 89.95% and 99.69% and specificity ranges between 93.06% and 99.65% across all tested conditions.
Validation based on patient-level record partitioning further confirmed the model’s robustness in cross-patient scenarios. The model accuracy remained stably above 95% across different ECG segment lengths, with the 10-second segment achieving an optimal accuracy of 96.50%. Sensitivity, which assesses the model’s ability to identify normal rhythm samples, consistently stayed above 99% at all lengths. This stability suggests that the model can reliably distinguish between normal and abnormal heartbeats even when applied to new patient data, reducing the risk of false positives for normal ECG signals.
The study acknowledges several limitations, primarily the class imbalance present in the experimental datasets. The proportion of life-threatening arrhythmias is relatively low compared to normal and not life-threatening cases, which can affect the model’s learning performance for minority classes. No data augmentation strategies were adopted to optimize the sample distribution, which may limit the model’s ability to generalize to rare arrhythmia types. Additionally, the performance on the CUDB dataset was lower than on the other databases, with an accuracy of 90.56% on 2-second segments, likely due to differences in sampling frequency and signal characteristics.
Despite these limitations, the proposed three-level risk stratification system and PaliConvLSTM model offer a significant technical solution for intelligent ECG diagnosis. The model’s ability to process raw ECG inputs without the need for manual feature extraction or signal filtering makes it suitable for integration into portable ECG devices, dynamic monitoring systems, or hospital ECG analysis platforms. Future research will focus on addressing data imbalance through augmentation techniques and incorporating multimodal medical data, such as ambulatory blood pressure and blood oxygen saturation, to further enhance the model’s adaptability to complex clinical scenarios.
The development of this algorithm represents a step forward in the application of artificial intelligence for cardiovascular disease management. By providing a refined classification system that aligns with clinical urgency, the model supports the rational scheduling of medical resources and the timely treatment of cardiovascular emergencies. The high accuracy and specificity achieved on standard ECG databases suggest that the method has strong potential for real-world clinical deployment, where it can assist physicians in rapid ECG waveform interpretation and arrhythmia screening.
The study highlights the importance of integrating both spatial and temporal features in ECG analysis. The palindromic structure of the PaliConvLSTM model allows for a comprehensive extraction of these features, addressing the limitations of previous models that relied on single-layer fusion or incomplete feature representation. This approach not only improves detection accuracy but also provides a systematic standard for clinical risk assessment, ensuring that patients receive appropriate attention based on the severity of their condition.
As the demand for efficient and accurate cardiovascular diagnostics continues to grow, the development of such intelligent tools becomes increasingly vital. The PaliConvLSTM model offers a reliable technical support for three-level arrhythmia risk stratification, with implications for improving early screening capabilities and optimizing medical resource allocation. The findings of this study contribute to the broader field of AI-assisted diagnosis, demonstrating the potential of deep learning architectures to handle the complex spatiotemporal dynamics of physiological signals.
Subject of Research: Development of a CNN-LSTM palindromic network for three-level arrhythmia risk stratification in ECG signals
Article Title: A three-level risk stratification algorithm for arrhythmia based on CNN-LSTM palindromic structure
Article References: Zhang, H., Li, J., Wang, M., Xuan, X., Xia, K., Tan, Y., Li, L., Chen, J., & Wang, H. (2026). A three-level risk stratification algorithm for arrhythmia based on CNN-LSTM palindromic structure. BioData Mining, 19(1), Article 80. https://doi.org/10.1186/s13040-026-00585-7
Image Credits: AI Generated
DOI: 10.1186/s13040-026-00585-7
Keywords: arrhythmia, deep learning, ECG, risk stratification, CNN-LSTM, cardiovascular disease, medical AI, clinical diagnosis, three-level, risk, stratification, algorithm
