Automatic sleep stage scoring using time-frequency analysis and stacked sparse autoencoders
File(s)art%3A10.1007%2Fs10439-015-1444-y.pdf (650.48 KB)
Published version
Author(s)
Tsinalis, O
Matthews, PM
Guo, Y
Type
Journal Article
Abstract
We developed a machine learning methodology for automatic sleep stage scoring. Our time-frequency analysis-based feature extraction is fine-tuned to capture sleep stage-specific signal features as described in the American Academy of Sleep Medicine manual that the human experts follow. We used ensemble learning with an ensemble of stacked sparse autoencoders for classifying the sleep stages. We used class-balanced random sampling across sleep stages for each model in the ensemble to avoid skewed performance in favor of the most represented sleep stages, and addressed the problem of misclassification errors due to class imbalance while significantly improving worst-stage classification. We used an openly available dataset from 20 healthy young adults for evaluation. We used a single channel of EEG from this dataset, which makes our method a suitable candidate for longitudinal monitoring using wearable EEG in real-world settings. Our method has both high overall accuracy (78%, range 75–80%), and high mean \(F_1\)-score (84%, range 82–86%) and mean accuracy across individual sleep stages (86%, range 84–88%) over all subjects. The performance of our method appears to be uncorrelated with the sleep efficiency and percentage of transitional epochs in each recording.
Date Issued
2015-10-13
Date Acceptance
2015-09-01
Citation
Annals of Biomedical Engineering, 2015, 44 (5), pp.1587-1597
ISSN
1573-9686
Publisher
Springer
Start Page
1587
End Page
1597
Journal / Book Title
Annals of Biomedical Engineering
Volume
44
Issue
5
Copyright Statement
© 2015 The Author(s). This article is published with open access at Springerlink.com. This article is distributed under the terms of the
Creative Commons Attribution 4.0 International License
(http://creativecommons.org/licenses/by/4.0/),
which permits unrestricted use, distribution, and reproduction
in any medium, provided you give appropriate
credit to the original author(s) and the source,
provide a link to the Creative Commons license, and
indicate if changes were made.
Creative Commons Attribution 4.0 International License
(http://creativecommons.org/licenses/by/4.0/),
which permits unrestricted use, distribution, and reproduction
in any medium, provided you give appropriate
credit to the original author(s) and the source,
provide a link to the Creative Commons license, and
indicate if changes were made.
License URL
Sponsor
Engineering & Physical Science Research Council (E
National Multiple Sclerosis Society
Grant Number
EP/K503733/1
PA 0103
Subjects
Deep learning
EEG
Electroencephalography
Ensemble learning
Biomedical Engineering
11 Medical And Health Sciences
09 Engineering
Publication Status
Published