Magnetocardiography-Based Ischemic Heart Disease Detection and Localization Using Machine Learning Methods

Abstract

Objective: This study focused on developing a fast and accurate automatic ischemic heart disease detection/localization methodology. Methods: Twavewas segmented from averaged Magnetocardiography (MCG) recordings and 164 features were subsequently extracted. These features were categorized into three groups: time domain features, frequency domain features, and informa-tion theory features. Next, we compared different machine learning classifiers including: k-nearest neighbor, decision tree, support vector machine (SVM), and XGBoost. To identify ischemia heart disease (IHD) case, we selected three classifiers with best performance and applied model ensemble to average results. All 164 features were used in this stage. To localize ischemia, we classified IHD group according to stenosis locations, including left anterior descending (LAD), left circumflex artery (LCX), and right coronary artery (RCA). For this task, we used XGBoost classifier and 18 time domain features. Results: For IHD detection, the SVM-XGBoost model achieved best results with accuracy = 94.03%, precision = 86.56%, recall = 97.78%, F-score = 92.79%, AUC = 0.98, and average precision = 0.98. For ischemia localization, XGBoost model achieved accuracy = 0.74, 0.68, and 0.65, for LAD, LCX, and RCA, respectively. Conclusion: we have developed an automatic IHD detection and localization system. We find that 1. T wave repolarization synchronicity is an important factor to distinguish IHD from normal subjects 2. Magnetic field pattern is associated with stenosis location. Significance: The proposed machine learning method provides the clinicians a fast and accurate diagnosis tool to interpret MCG data, boosting its acceptance into clinics. Furthermore, the magnetic pole characteristics revealed by the method shows to be related to ischemia location, presenting the opportunity to noninvasively locate ischemia.