A High Precision Machine Learning-Enabled System for Predicting Idiopathic Ventricular Arrhythmia Origins

Radiofrequency catheter ablation (CA) is an efficient antiarrhythmic treatment with a class I indication for idiopathic ventricular arrhythmia (IVA), only when drugs are ineffective or have unacceptable side effects. The accurate prediction of the origins of IVA can significantly increase the operation success rate, reduce operation duration and decrease the risk of complications. The present work proposes an artificial intelligence-enabled ECG analysis algorithm to estimate possible origins of idiopathic ventricular arrhythmia at a clinical-grade level accuracy.

A total of 18,612 ECG recordings extracted from 545 patients who underwent successful CA to treat IVA were proportionally sampled into training, validation and testing cohorts. We designed four classification schemes responding to different hierarchical levels of the possible IVA origins. For every classification scheme, we compared 98 distinct machine learning models with optimized hyperparameter values obtained through extensive grid search and reported an optimal algorithm with the highest accuracy scores attained on the testing cohorts.

For classification scheme 4, our pioneering study designs and implements a machine learning-based ECG algorithm to predict 21 possible sites of IVA origin with an accuracy of 98.24% on a testing cohort. The accuracy and F1-score for the left three schemes surpassed 99%.

In this work, we developed an algorithm that precisely predicts the correct origins of IVA (out of 21 possible sites) and outperforms the accuracy of all prior studies and human experts.