Diagnostic Accuracy of Different Machine Learning Algorithms for Obstructive Sleep Apnea
- Author:
Hyun-Woo KIM
1
;
Euihwan PARK
;
Dae Jin KIM
;
Sue Jean MUN
;
Jiyoung KIM
;
Gha-Hyun LEE
;
Jae Wook CHO
Author Information
- Publication Type:1
- From:Journal of Sleep Medicine 2020;17(2):128-137
- CountryRepublic of Korea
- Language:Korean
-
Abstract:
Objectives:The objective of this study was to develop models for predicting obstructive sleep apnea (OSA) based on easily obtainable clinical information of patients using various machine learning techniques.
Methods:We used a data set that included the records of 1,368 patients, in which 1,074 patients were male (78.5 %), and 294 patients were female (21.5 %). We randomly divided the data into a training set (1,000) and test set (368). Five machine learning methods, i.e., support vector machine model, lasso logit model, naïve bayes, discriminant analysis, and K-nearest neighbor (KNN), with a 10-cross fold technique were used with the proposed model to predict OSA. We evaluated the accuracy, sensitivity, specificity, and precision of each model for three thresholds [Apnea-Hypopnea Index (AHI)≥5, AHI≥15, and AHI≥30].
Results:Among the machine learning techniques, KNN showed the best results compared to the other techniques. The accuracy, sensitivity, specificity, and precision of OSA prediction were 87.0%, 91.0%, 74.4%, and 91.9%, respectively, based on AHI≥5. When the threshold of OSA was AHI≥15 or AHI≥30, KNN provided lower accuracy (79.6% each) and precision (79.0% and 68.7%), which were still higher than those of the other techniques.
Conclusions:The model derived from the KNN technique exhibited the best performance based on its highest level of accuracy. We demonstrate that this model is a useful tool for predicting OSA.
