A Topology Informed Random Forest Classifier for ECG Classification

Paul Samuel Ignacio, Jay-Anne Bulauan, John Rick Manzanares
University of the Philippines Baguio


Abstract

We leverage mathematically computable topological signatures of 12-lead electrocardiograms (ECGs) as proxy for features informed by medical expertise to train a random forest model in a 9-class classification task. As has been shown for detecting Atrial Fibrillation using single-lead ECGs, this approach verifies the existence and viability of signal in the topology of ECGs for diagnosing cardiac conditions. Upscaling this to the use of all 12 leads of a standard ECG to diagnose multiple heart conditions improves accessibility to automated diagnostics by reducing expert-dependent input in feature extraction.

We view ECGs as multivariate time series data and convert different segments and groupings of leads to point cloud embeddings. This stores both local and global structures of ECGs, and encodes periodic information as attractor cycles in highdimensional space. We then employ topological data analysis on these embeddings to extract topological features based on different summaries (persistence barcodes, landscapes, entropy) available in the literature. We supplement these features with demographic data and statistical moments of RR intervals based on the Pan-Tomkins algorithm for each lead to train the classifier.

We rank features by importance and narrow down a pool of 365 features to obtain an initial shortlist of 56 features of which 2 are demographic, 31 are RR-statistics, and 23 are topological. We use the shortlisted features to train a random forest classifier with 800 trees, and perform a stratified 10-fold cross validation using the Challenge metrics. A 95% confidence interval of the cross validated training performance for the Challenge metrics are: AUROC (0.876+/- 0.014), AUPRC (0.585+/-0.062), Accuracy (0.833+/-0.012), F1 score (0.473+/-0.032), F2 score (0.590+/-0.034), and G2 score (0.287+/-0.028).

Testing performed by the Challenge organizers on the trained classifier during the unofficial phase yields an F2 score of 0.577 and a G2 score of 0.274 for a geometric mean of 0.398.