Tree-Based Machine Learning Techniques for Automated Human Sleep Stage Classification


ARSLAN R. S., ULUTAŞ H., KÖKSAL A. S., BAKIR M., Çiftçi B.

Traitement du Signal, cilt.40, sa.4, ss.1385-1400, 2023 (SCI-Expanded) identifier

  • Yayın Türü: Makale / Tam Makale
  • Cilt numarası: 40 Sayı: 4
  • Basım Tarihi: 2023
  • Doi Numarası: 10.18280/ts.400408
  • Dergi Adı: Traitement du Signal
  • Derginin Tarandığı İndeksler: Science Citation Index Expanded (SCI-EXPANDED), Scopus, PASCAL, Business Source Elite, Business Source Premier, Compendex, zbMATH
  • Sayfa Sayıları: ss.1385-1400
  • Anahtar Kelimeler: machine learning, multi-channel data, polysomnography (PSG), sleep stage scoring
  • Yozgat Bozok Üniversitesi Adresli: Evet

Özet

Background: Sleep disorders pose significant health risks, necessitating accurate diagnostics. The analysis of polysomnographic data and subsequent sleep stage classification by medical professionals are crucial in diagnosing these disorders. The application of artificial intelligence (AI)-based systems for automated sleep stage classification has gained significant momentum recently. Methodology: In this study, we introduce a machine learning model designed for high-accuracy, automated sleep stage classification. We utilized a dataset consisting of polysomnographic data from 50 individuals, obtained from the Yozgat Bozok University Sleep Center. A variety of classifiers, including Extra Tree, Decision Tree, Random Forest, Ada Boost, and Gradient Boost, were tested. Sleep stages were classified into three categories: Wakefulness (WK), Rapid Eye Movement (REM), and Non-Rapid Eye Movement (N-REM). Results: The overall classification accuracies were 95.4%, 95%, and 92% for three distinct classifiers, respectively, with the highest accuracy reaching 98.8%. Comparison with Existing Methods: This study distinguishes itself from comparable sleep stage-scoring research by utilizing a unique dataset, and by incorporating data from 16 channels, which contributes to the achieved accuracy. Conclusion: The machine learning model trained with a unique dataset demonstrated high classification success in the automated scoring of sleep stages. This research underscores the potential of machine learning techniques in improving sleep disorder diagnostics.