Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation
Sleep-stage classification is a critical aspect of understanding sleep patterns in sleep research and healthcare. However, challenges arise when dealing with a limited number of labeled samples in the target domain. Traditional methods in Deep Learning (DL) and Domain Adaptation (DA) globally compar...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2024-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10597569/ |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850113996312543232 |
|---|---|
| author | Roya Ghasemigarjan Mohammad Mikaeili Seyed Kamaledin Setarehdan |
| author_facet | Roya Ghasemigarjan Mohammad Mikaeili Seyed Kamaledin Setarehdan |
| author_sort | Roya Ghasemigarjan |
| collection | DOAJ |
| description | Sleep-stage classification is a critical aspect of understanding sleep patterns in sleep research and healthcare. However, challenges arise when dealing with a limited number of labeled samples in the target domain. Traditional methods in Deep Learning (DL) and Domain Adaptation (DA) globally compare feature distributions, often overlooking intricate decision boundaries between sleep-stage classes. This results in ambiguous features near class boundaries, diminishing classification accuracy. The conventional two-step process of using a pre-trained classifier for predictions and assessing uncertainty fails to effectively incorporate unlabeled data in classifier training, neglecting the complexities of the target domain. To address these challenges, we propose Adversarial Deep Learning Joint Domain Adaptation (ADLJDA). This innovative approach integrates an adversarial model and deploys two distinct sleep-stage classifiers as discriminators, allowing for a nuanced consideration of class boundaries during feature distribution alignment. ADLJDA also incorporates an entropy measure with cross-entropy loss during training to harness information from unlabeled data in the target domain. Experimental results on three benchmark EEG datasets highlight the efficacy of ADLJDA. The approach consistently demonstrates the ability to generate robust and transferable features, mitigating the impact of ambiguous features near original class boundaries. Importantly, ADLJDA shows a significant improvement in classification accuracy compared to existing state-of-the-art DA methods, even in datasets with intricate patterns and complexities. This research contributes to advancing sleep-stage classification methodologies, offering a promising solution for enhanced accuracy in real-world applications and furthering our understanding of sleep-related phenomena. |
| format | Article |
| id | doaj-art-25d611a407c041eeafa944c117f5259b |
| institution | OA Journals |
| issn | 2169-3536 |
| language | English |
| publishDate | 2024-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj-art-25d611a407c041eeafa944c117f5259b2025-08-20T02:37:01ZengIEEEIEEE Access2169-35362024-01-011218663918665710.1109/ACCESS.2024.342843510597569Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain AdaptationRoya Ghasemigarjan0https://orcid.org/0009-0000-0218-8711Mohammad Mikaeili1https://orcid.org/0000-0002-7245-885XSeyed Kamaledin Setarehdan2Department of Engineering, Shahed University, Tehran, IranDepartment of Engineering, Shahed University, Tehran, IranControl and Intelligent Processing Centre of Excellence, School of Electrical and Computer Engineering, College of Engineering, University of Tehran, Tehran, IranSleep-stage classification is a critical aspect of understanding sleep patterns in sleep research and healthcare. However, challenges arise when dealing with a limited number of labeled samples in the target domain. Traditional methods in Deep Learning (DL) and Domain Adaptation (DA) globally compare feature distributions, often overlooking intricate decision boundaries between sleep-stage classes. This results in ambiguous features near class boundaries, diminishing classification accuracy. The conventional two-step process of using a pre-trained classifier for predictions and assessing uncertainty fails to effectively incorporate unlabeled data in classifier training, neglecting the complexities of the target domain. To address these challenges, we propose Adversarial Deep Learning Joint Domain Adaptation (ADLJDA). This innovative approach integrates an adversarial model and deploys two distinct sleep-stage classifiers as discriminators, allowing for a nuanced consideration of class boundaries during feature distribution alignment. ADLJDA also incorporates an entropy measure with cross-entropy loss during training to harness information from unlabeled data in the target domain. Experimental results on three benchmark EEG datasets highlight the efficacy of ADLJDA. The approach consistently demonstrates the ability to generate robust and transferable features, mitigating the impact of ambiguous features near original class boundaries. Importantly, ADLJDA shows a significant improvement in classification accuracy compared to existing state-of-the-art DA methods, even in datasets with intricate patterns and complexities. This research contributes to advancing sleep-stage classification methodologies, offering a promising solution for enhanced accuracy in real-world applications and furthering our understanding of sleep-related phenomena.https://ieeexplore.ieee.org/document/10597569/Adversarial learningchannel optimizationclassification accuracydeep learningdomain adaptationEEG signal |
| spellingShingle | Roya Ghasemigarjan Mohammad Mikaeili Seyed Kamaledin Setarehdan Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation IEEE Access Adversarial learning channel optimization classification accuracy deep learning domain adaptation EEG signal |
| title | Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation |
| title_full | Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation |
| title_fullStr | Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation |
| title_full_unstemmed | Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation |
| title_short | Optimizing EEG-Based Sleep Staging: Adversarial Deep Learning Joint Domain Adaptation |
| title_sort | optimizing eeg based sleep staging adversarial deep learning joint domain adaptation |
| topic | Adversarial learning channel optimization classification accuracy deep learning domain adaptation EEG signal |
| url | https://ieeexplore.ieee.org/document/10597569/ |
| work_keys_str_mv | AT royaghasemigarjan optimizingeegbasedsleepstagingadversarialdeeplearningjointdomainadaptation AT mohammadmikaeili optimizingeegbasedsleepstagingadversarialdeeplearningjointdomainadaptation AT seyedkamaledinsetarehdan optimizingeegbasedsleepstagingadversarialdeeplearningjointdomainadaptation |