Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification

Understanding sleep stages is crucial for diagnosing sleep disorders, developing treatments, and studying sleep’s impact on overall health. With the growing availability of affordable brain monitoring devices, the volume of collected brain data has increased significantly. However, analyzing these d...

Full description

Saved in:
Bibliographic Details
Main Authors: Masoud Nateghi, Mahdi Rahbar Alam, Hossein Amiri, Samaneh Nasiri, Reza Sameni
Format: Article
Language:English
Published: MDPI AG 2024-12-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/24/24/7881
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850085173773729792
author Masoud Nateghi
Mahdi Rahbar Alam
Hossein Amiri
Samaneh Nasiri
Reza Sameni
author_facet Masoud Nateghi
Mahdi Rahbar Alam
Hossein Amiri
Samaneh Nasiri
Reza Sameni
author_sort Masoud Nateghi
collection DOAJ
description Understanding sleep stages is crucial for diagnosing sleep disorders, developing treatments, and studying sleep’s impact on overall health. With the growing availability of affordable brain monitoring devices, the volume of collected brain data has increased significantly. However, analyzing these data, particularly when using the gold standard multi-lead electroencephalogram (EEG), remains resource-intensive and time-consuming. To address this challenge, automated brain monitoring has emerged as a crucial solution for cost-effective and efficient EEG data analysis. A critical component of sleep analysis is detecting transitions between wakefulness and sleep states. These transitions offer valuable insights into sleep quality and quantity, essential for diagnosing sleep disorders, designing effective interventions, enhancing overall health and well-being, and studying sleep’s effects on cognitive function, mood, and physical performance. This study presents a novel EEG feature extraction pipeline for the accurate classification of various wake and sleep stages. We propose a noise-robust model-based Kalman filtering (KF) approach to track changes in a time-varying auto-regressive model (TVAR) applied to EEG data during different wake and sleep stages. Our approach involves extracting features, including instantaneous frequency and instantaneous power from EEG, and implementing a two-step classifier for sleep staging. The first step classifies data into wake, REM, and non-REM categories, while the second step further classifies non-REM data into N1, N2, and N3 stages. Evaluation on the extended Sleep-EDF dataset (Sleep-EDFx), with 153 EEG recordings from 78 subjects, demonstrated compelling results with classifiers including Logistic Regression, Support Vector Machines, Extreme Gradient Boosting (XGBoost), and Light Gradient Boosting Machine (LGBM). The best performance was achieved with the LGBM and XGBoost classifiers, yielding an overall accuracy of over 77%, a macro-averaged F1 score of 0.69, and a Cohen’s kappa of 0.68, highlighting the efficacy of the proposed method with a remarkably compact and interpretable feature set.
format Article
id doaj-art-d5fb27a4cbc5460dbff70b3a827a61eb
institution DOAJ
issn 1424-8220
language English
publishDate 2024-12-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj-art-d5fb27a4cbc5460dbff70b3a827a61eb2025-08-20T02:43:47ZengMDPI AGSensors1424-82202024-12-012424788110.3390/s24247881Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage ClassificationMasoud Nateghi0Mahdi Rahbar Alam1Hossein Amiri2Samaneh Nasiri3Reza Sameni4Department of Biomedical Informatics, School of Medicine, Emory University, Atlanta, GA 30322, USAIndependent Researcher, Shiraz 7197688711, IranDepartment of Biomedical Informatics, School of Medicine, Emory University, Atlanta, GA 30322, USADepartment of Biomedical Informatics, School of Medicine, Emory University, Atlanta, GA 30322, USADepartment of Biomedical Informatics, School of Medicine, Emory University, Atlanta, GA 30322, USAUnderstanding sleep stages is crucial for diagnosing sleep disorders, developing treatments, and studying sleep’s impact on overall health. With the growing availability of affordable brain monitoring devices, the volume of collected brain data has increased significantly. However, analyzing these data, particularly when using the gold standard multi-lead electroencephalogram (EEG), remains resource-intensive and time-consuming. To address this challenge, automated brain monitoring has emerged as a crucial solution for cost-effective and efficient EEG data analysis. A critical component of sleep analysis is detecting transitions between wakefulness and sleep states. These transitions offer valuable insights into sleep quality and quantity, essential for diagnosing sleep disorders, designing effective interventions, enhancing overall health and well-being, and studying sleep’s effects on cognitive function, mood, and physical performance. This study presents a novel EEG feature extraction pipeline for the accurate classification of various wake and sleep stages. We propose a noise-robust model-based Kalman filtering (KF) approach to track changes in a time-varying auto-regressive model (TVAR) applied to EEG data during different wake and sleep stages. Our approach involves extracting features, including instantaneous frequency and instantaneous power from EEG, and implementing a two-step classifier for sleep staging. The first step classifies data into wake, REM, and non-REM categories, while the second step further classifies non-REM data into N1, N2, and N3 stages. Evaluation on the extended Sleep-EDF dataset (Sleep-EDFx), with 153 EEG recordings from 78 subjects, demonstrated compelling results with classifiers including Logistic Regression, Support Vector Machines, Extreme Gradient Boosting (XGBoost), and Light Gradient Boosting Machine (LGBM). The best performance was achieved with the LGBM and XGBoost classifiers, yielding an overall accuracy of over 77%, a macro-averaged F1 score of 0.69, and a Cohen’s kappa of 0.68, highlighting the efficacy of the proposed method with a remarkably compact and interpretable feature set.https://www.mdpi.com/1424-8220/24/24/7881automatic sleep stagingelectroencephalograminstantaneous frequency trackingKalman filter
spellingShingle Masoud Nateghi
Mahdi Rahbar Alam
Hossein Amiri
Samaneh Nasiri
Reza Sameni
Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification
Sensors
automatic sleep staging
electroencephalogram
instantaneous frequency tracking
Kalman filter
title Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification
title_full Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification
title_fullStr Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification
title_full_unstemmed Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification
title_short Model-Based Electroencephalogram Instantaneous Frequency Tracking: Application in Automated Sleep–Wake Stage Classification
title_sort model based electroencephalogram instantaneous frequency tracking application in automated sleep wake stage classification
topic automatic sleep staging
electroencephalogram
instantaneous frequency tracking
Kalman filter
url https://www.mdpi.com/1424-8220/24/24/7881
work_keys_str_mv AT masoudnateghi modelbasedelectroencephalograminstantaneousfrequencytrackingapplicationinautomatedsleepwakestageclassification
AT mahdirahbaralam modelbasedelectroencephalograminstantaneousfrequencytrackingapplicationinautomatedsleepwakestageclassification
AT hosseinamiri modelbasedelectroencephalograminstantaneousfrequencytrackingapplicationinautomatedsleepwakestageclassification
AT samanehnasiri modelbasedelectroencephalograminstantaneousfrequencytrackingapplicationinautomatedsleepwakestageclassification
AT rezasameni modelbasedelectroencephalograminstantaneousfrequencytrackingapplicationinautomatedsleepwakestageclassification