Machine learning-based classifiers for obstructive sleep apnea diagnosis using ECG and EDR signals

Abstract
item.page.abstract.eng
Obstructive sleep apnea (OSA) is one of the most common sleep-related breathing disorders with serious cardiovascular and neurological implications. It is normally diagnosed in the context of polysomnography (PSG) which is very expensive, cumbersome, and invasive in the process. In this work, we applied machine learning and deep learning techniques to develop an automated ECG method for detecting and assessing the severity of OSA to make diagnosis more accessible and efficient. We used the PhysioNet Apnea-ECG database and used 10 classification methods: logistic regression, support vector machine with RBF kernel, random forest, CNN and RNN with raw ECG, CNN and RNN models with RR intervals and ECG-derived respiration (RR+EDR) and three ensemble strategies: soft voting, AUC-weighted voting, and stacking. Heart rate variability and EDR features were extracted from one-minute ECG windows with a ±1-minute temporal context. All the methods have subject-independent data partitioning to avoid data leakage and to generate more realistic models. The models were evaluated internally and independently using the official x01–x35 test set and 35 subjects and 12,600 one-minute windows. Performance was assessed based on precision, sensitivity, specificity, accuracy, F1 score, AUC, and Cohen's kappa coefficient. The patient-level hierarchical analysis was also conducted, where the minute-level predictions were collected into an Apnea Index and four categories of severity were established: normal, mild, moderate and severe. The results showed that classical machine learning and ensemble approaches provided the most consistent generalization. The Weighted-AUC set had the best overall external performance with an AUC of 0.939 and F1 score of 0.834. Patient severity estimation also showed promising results with a mean absolute error of the apnea index of 3.01 events per hour and a severity accuracy of around 80% in the external assessment.
item.page.descriptioneng
Citation
DOI