A "two-step classification" machine learning method for non-invasive localization of premature ventricular contraction origins based on 12-lead ECG

BackgroundPremature ventricular contraction (PVC) is a type of cardiac arrhythmia that originates from ectopic pacemaker in the ventricles. The localization of the origin of PVC is essential for successful catheter ablation. However, most studies on non-invasive PVC localization focus on elaborate localization in specific regions of the ventricle. This study aims to propose a machine learning algorithm based on 12-lead electrocardiogram (ECG) data that can improve the accuracy of PVC localization in the whole ventricle.MethodsWe collected 12-lead ECG data from 249 patients with spontaneous or pacing-induced PVCs. The ventricle was divided into 11 segments. In this paper, we propose a machine learning method consisting of two consecutive classification steps. In the first classification step, each PVC beat was labeled to one of the 11 ventricular segments using six features, including a newly proposed morphological feature called "Peak_index." Four machine learning methods were tested for comparative multi-classification performance and the best classifier result was kept to the next step. In the second classification step, a binary classifier was trained using a smaller combination of features to further differentiate segments that are easily confused.ResultsThe Peak_index as a proposed new classification feature combined with other features is suitable for whole ventricle classification by machine learning methods. The test accuracy of the first classification reached 75.87%. It is shown that a second classification for confusable categories can improve the classification results. After the second classification, the test accuracy reached 76.84%, and when a sample classified into adjacent segments was considered correct, the test "rank accuracy" was improved to 93.49%. The binary classification corrected 10% of the confused samples.ConclusionThis paper proposes a "two-step classification" method to localize the origin of PVC beats into the 11 regions of the ventricle using non-invasive 12-lead ECG. It is expected to be a promising technique to be used in clinical settings to help guide ablation procedures.