Cardiac arrhythmias on the MIT-BIH database
Recognise heartbeat types automatically, then measure what a premature beat changes in the beats that follow.
- Context
- Individual project, UQTR
- Supervisor
- Prof. Usef Faghihi
- Data
- MIT-BIH Arrhythmia Database
- Status
- Methodology fixes in progress
- Question
Can heartbeats be classified reliably, and does a premature beat really change the RR intervals that follow?
- Intervention
Classification with MiniRocket, XGBoost and a 1D CNN + LSTM network. Rhythm modelling, then causal effect estimation with Double ML and CausalForestDML.
- Effect
XGBoost reaches 0.74 accuracy. The average treatment effect (ATE) of a premature beat on the following RR intervals is significant.
These results are preliminary. My supervisor flagged nine issues, including data leakage between patients and post-treatment variables used as confounders. I am redoing the analysis with those fixed.
- Python
- Jupyter
- MiniRocket
- XGBoost
- 1D CNN + LSTM
- Double ML
- CausalForestDML






