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

2509.25804v1 cs.LG, cs.AI, cs.NI 2025-10-02
Авторы:

Vaskar Chakma, Ju Xiaolin, Heling Cao, Xue Feng, Ji Xiaodong, Pan Haiyan, Gao Zhan

Abstract

This study aims to develop and evaluate an ensemble machine learning-based framework for the automatic detection of Wide QRS Complex Tachycardia (WCT) from ECG signals, emphasizing diagnostic accuracy and interpretability using Explainable AI. The proposed system integrates ensemble learning techniques, i.e., an optimized Random Forest known as CardioForest, and models like XGBoost and LightGBM. The models were trained and tested on ECG data from the publicly available MIMIC-IV dataset. The testing was carried out with the assistance of accuracy, balanced accuracy, precision, recall, F1 score, ROC-AUC, and error rate (RMSE, MAE) measures. In addition, SHAP (SHapley Additive exPlanations) was used to ascertain model explainability and clinical relevance. The CardioForest model performed best on all metrics, achieving a test accuracy of 94.95%, a balanced accuracy of 88.31%, and high precision and recall metrics. SHAP analysis confirmed the model's ability to rank the most relevant ECG features, such as QRS duration, in accordance with clinical intuitions, thereby fostering trust and usability in clinical practice. The findings recognize CardioForest as an extremely dependable and interpretable WCT detection model. Being able to offer accurate predictions and transparency through explainability makes it a valuable tool to help cardiologists make timely and well-informed diagnoses, especially for high-stakes and emergency scenarios.

Ссылки и действия

Связанные статьи

Attention Beyond Neighborhoods: Reviving Transformer for Graph Clustering

## Контекст Графы широко применяются в различных областях, включая социальные сети, финансы и биоинформатику. Однако, п...

2025-09-20

Sample Efficient Experience Replay in Non-stationary Environments

#### Контекст Область исследования — reinforcement learning (RL) в нестационарных средах, где изменение динамики и возна...

2025-09-20

Multi-Agent Reinforcement Learning for Task Offloading in Wireless Edge Networks

## Контекст Edge computing системы представляют собой распределенные среды, где несколько устройств должны принимать ск...

2025-09-05