Research on Sleep Staging Based on Support Vector Machine and Extreme Gradient Boosting Algorithm
Yiwen Wang,Shuming Ye,Zhi Xu,Yonghua Chu,Jiarong Zhang,Wenke Yu
DOI: https://doi.org/10.2147/nss.s467111
2024-11-29
Nature and Science of Sleep
Abstract:Yiwen Wang, 1 Shuming Ye, 2 Zhi Xu, 3 Yonghua Chu, 1 Jiarong Zhang, 4 Wenke Yu 5 1 Clinical Medical Engineering Department, The Second Affiliated Hospital, Zhejiang University School of Medicine, HangZhou, ZheJiang, People's Republic of China; 2 Department of Biomedical Engineering, Zhejiang University, HangZhou, ZheJiang, People's Republic of China; 3 China Astronaut Research and Training Center, BeiJing, People's Republic of China; 4 Baidu Inc, BeiJing, People's Republic of China; 5 Radiology Department, ZheJiang Province Qing Chun Hospital, HangZhou, ZheJiang, People's Republic of China Correspondence: Yiwen Wang; Shuming Ye, Email ; Purpose: To develop a sleep-staging algorithm based on support vector machine (SVM) and extreme gradient boosting model (XB Boost) and evaluate its performance. Methods: In this study, data features were extracted based on physiological significance, feature dimension reduction was performed through appropriate methods, and XG Boost classifier and SVM were used for classification. One hundred and twenty training sets and 80 test sets were randomly composed of the first 200 groups of data from the SHH1 database. The polysomnography (PSG) data of 20 real individuals in the clinic were selected as the experimental data. The C3 electroencephalogram (EEG), left and right electrooculogram (EOG), electromyogram (EMG), and other signals were analyzed. Finally, the stages were adjusted based on human sleep laws. The standard staging of the database and the doctor's diagnosis staging was used as the standard. Results: The SHHS1 database test results were as follows: the average accuracy was 83.24%, the precision and recall of Stage Wake and Stage 2 NREM sleep (N2) were over 80%, and the precision, F1-Score and recall of Stage 3 NREM sleep (N3) and Rapid Eye Movement (REM) were more than 70%. The clinical data test results were as follows: the average accuracy rate was 76.37%; for Wake and N3, the precision reached 85%; for Wake, N2, and REM, the recall rate reached over 70%; for Wake, the F-1 Score reached over 90%. Conclusion: This study shows that the sleep staging results of the algorithm for the database and clinical data were similar. The staging results meet the requirements at the medical level. Keywords: sleep staging, physiological significance, feature dimension reduction, databases and clinical trials, confusion matrix In 2007, the American Academy of Sleep Medicine (AASM) developed new sleep interpretation guidelines - The AASM Guidelines. The AASM guidelines define sleep stages as Wake, non-rapid eye movement (NREM), and rapid eye movement (REM). NREM is divided into NREMI stage (N1), NREMII stage (N2) and NREMIII stage (N3). When falling asleep, the sleep stage of normal people first enters the NREM stage, gradually shifts from N1 stage to N3 stage, and then enters the REM stage at N2 stage or N3 stage. The sleep stage passes from NREM to REM, representing a complete sleep cycle 1–5 . Sleep staging results can be linked to certain diseases. Rapid eye movement sleep behavior disorder (RBD), as an early biological marker for predicting neurodegenerative diseases, such as Parkinson's disease, is of great significance for the prevention and treatment of the latent stage of neurodegenerative diseases. RBD diagnosis requires accurate REM staging. 2 Depressed patients had low sleep efficiency and poor sleep quality. Accurate and detailed sleep staging results are required for both diagnosis and pharmacological intervention, particularly N3. 3,4 The proportion of REM sleep time is reduced in adolescents with bipolar depressive disorder (BDD). The duration of N2 was shorter in adult BDD patients than in normal adults. 5 Although the diagnosis of sleep apnea syndrome (SAS) requires respiratory indications, and blood oxygen indications, the results of sleep staging also play an important role in the determination and research of SAS of the central and obstructive types. 6 Clinically, polysomnography (PSG) is generally used for sleep-stage classification in patients. 7 It takes a doctor two hours to determine a patient's sleep stage using PSG (particularly electroencephalogram (EEG)). The workload is large and it depends on the experience of the doctor. Patients are required to wear EEG electrodes overnight. The measurement method is inconvenient and subject to greater subjective influence. 8,9 The lack of medical resources and low diagnostic efficiency have resulted in some shortcomings in professional diagnosis of related diseases and insufficient popularization of sleep quality preliminary screening. Therefore, an eff -Abstract Truncated-
neurosciences,clinical neurology