CardioForest: An Explainable Ensemble Learning Model for Automatic Wide QRS Complex Tachycardia Diagnosis from ECG

Wide QRS Complex Tachycardia (WCT) is a life-threatening cardiac arrhythmia requiring rapid and accurate diagnosis. Traditional manual ECG interpretation is time-consuming and subject to inter-observer variability, while existing AI models often lack the clinical interpretability necessary for trusted deployment in emergency settings. We developed CardioForest, an optimized Random Forest ensemble model, for automated WCT detection from 12-lead ECG signals. The model was trained, tested, and validated using 10-fold cross-validation on 800,000 ten-second-long 12-lead Electrocardiogram (ECG) recordings from the MIMIC-IV dataset (15.46% WCT prevalence), with comparative evaluation against XGBoost, LightGBM, and Gradient Boosting models. Performance was assessed using accuracy, balanced accuracy, precision, recall, F1-score, ROC-AUC, RMSE, and MAE. SHAP (SHapley Additive exPlanations) analysis provided feature-level interpretability to ensure clinical validity. CardioForest achieved superior and consistent performance across all metrics: test accuracy 95.19% (±0.33%), balanced accuracy 88.76% (±0.79%), precision 95.26%, recall 78.42%, F1-score 86.02%, and ROC-AUC 0.8886, with the lowest error rates (RMSE: 0.2532, MAE: 0.1944). Statistical significance testing confirmed CardioForest’s advantages over XGBoost (accuracy +6.75%, p < 0.001), LightGBM (accuracy +10.85%, p < 0.001), and Gradient Boosting (accuracy +2.69%, p = 0.012). Notably, CardioForest demonstrated exceptional stability (coefficient of variation: 0.35%) compared to competing models. SHAP analysis revealed that QRS duration—the primary clinical diagnostic criterion—dominated model predictions (mean SHAP value: 0.45), with additional contributions from QRS morphology and axis measurements, perfectly aligning with established cardiological knowledge. CardioForest represents a clinically validated, interpretable AI solution for WCT detection that balances diagnostic accuracy with transparent decision-making. With inference times under 10 milliseconds and comprehensive explainability through SHAP visualizations, the model is deployment-ready for real-time emergency department screening. By providing cardiologists with both accurate predictions and clinically interpretable feature attributions, CardioForest addresses the critical gap between AI performance and clinical trust, offering a practical tool for timely, evidence-based cardiac diagnosis in high-stakes scenarios.

Authors

Institutions

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-01-22
DOI
https://doi.org/10.5281/zenodo.18937173
Primary Topic
ECG Monitoring and Analysis
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

CardioForest: An Explainable Ensemble Learning Model for Automatic Wide QRS Complex Tachycardia Diagnosis from ECG

Xiaolin Ju, Ji X, Vaskar Chakma, Pan Haiyan et al.
Zenodo (CERN European Organization for Nuclear Research)
ECG Monitoring and Analysis
article

CardioForest: An Explainable Ensemble Learning Model for Automatic Wide QRS Complex Tachycardia Diagnosis from ECG

Xiaolin Ju, Ji X, Vaskar Chakma, Pan Haiyan, Heling Cao, Zhan Gao, Feng Xue
article en

Abstract

Wide QRS Complex Tachycardia (WCT) is a life-threatening cardiac arrhythmia requiring rapid and accurate diagnosis. Traditional manual ECG interpretation is time-consuming and subject to inter-observer variability, while existing AI models often lack the clinical interpretability necessary for trusted deployment in emergency settings. We developed CardioForest, an optimized Random Forest ensemble model, for automated WCT detection from 12-lead ECG signals. The model was trained, tested, and validated using 10-fold cross-validation on 800,000 ten-second-long 12-lead Electrocardiogram (ECG) recordings from the MIMIC-IV dataset (15.46% WCT prevalence), with comparative evaluation against XGBoost, LightGBM, and Gradient Boosting models. Performance was assessed using accuracy, balanced accuracy, precision, recall, F1-score, ROC-AUC, RMSE, and MAE. SHAP (SHapley Additive exPlanations) analysis provided feature-level interpretability to ensure clinical validity. CardioForest achieved superior and consistent performance across all metrics: test accuracy 95.19% (±0.33%), balanced accuracy 88.76% (±0.79%), precision 95.26%, recall 78.42%, F1-score 86.02%, and ROC-AUC 0.8886, with the lowest error rates (RMSE: 0.2532, MAE: 0.1944). Statistical significance testing confirmed CardioForest’s advantages over XGBoost (accuracy +6.75%, p < 0.001), LightGBM (accuracy +10.85%, p < 0.001), and Gradient Boosting (accuracy +2.69%, p = 0.012). Notably, CardioForest demonstrated exceptional stability (coefficient of variation: 0.35%) compared to competing models. SHAP analysis revealed that QRS duration—the primary clinical diagnostic criterion—dominated model predictions (mean SHAP value: 0.45), with additional contributions from QRS morphology and axis measurements, perfectly aligning with established cardiological knowledge. CardioForest represents a clinically validated, interpretable AI solution for WCT detection that balances diagnostic accuracy with transparent decision-making. With inference times under 10 milliseconds and comprehensive explainability through SHAP visualizations, the model is deployment-ready for real-time emergency department screening. By providing cardiologists with both accurate predictions and clinically interpretable feature attributions, CardioForest addresses the critical gap between AI performance and clinical trust, offering a practical tool for timely, evidence-based cardiac diagnosis in high-stakes scenarios.

Zenodo (CERN European Organization for Nuclear Research)
Nantong University (CN), Henan University of Technology (CN), Affiliated Hospital of Nantong University (CN)
Life in Land
Openalex Percentile: Top 36%
ECG Monitoring and Analysis
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.