Robust, automated sleep scoring by a compact neural network with distributional shift correction

Journal Article · 12 December 2019 · PLoS ONE

DOI: https://doi.org/10.1371/journal.pone.0224642 · OSTI ID:1713216

Barger, Zeke  ^[1]; Frye, Charles G. ^[1]; Liu, Danqian ^[1]; Dan, Yang ^[1]; Bouchard, Kristofer E. ^[2]

---

1. Univ. of California, Berkeley, CA (United States)
2. Univ. of California, Berkeley, CA (United States); Lawrence Berkeley National Lab. (LBNL), Berkeley, CA (United States)

Studying the biology of sleep requires the accurate assessment of the state of experimental subjects, and manual analysis of relevant data is a major bottleneck. Recently, deep learning applied to electroencephalogram and electromyogram data has shown great promise as a sleep scoring method, approaching the limits of inter-rater reliability. As with any machine learning algorithm, the inputs to a sleep scoring classifier are typically standardized in order to remove distributional shift caused by variability in the signal collection process. However, in scientific data, experimental manipulations introduce variability that should not be removed. For example, in sleep scoring, the fraction of time spent in each arousal state can vary between control and experimental subjects. We introduce a standardization method, mixture z-scoring, that preserves this crucial form of distributional shift. Using both a simulated experiment and mouse in vivo data, we demonstrate that a common standardization method used by state-of-the-art sleep scoring algorithms introduces systematic bias, but that mixture z-scoring does not. We present a free, open-source user interface that uses a compact neural network and mixture z-scoring to allow for rapid sleep scoring with accuracy that compares well to contemporary methods. This work provides a set of computational tools for the robust automation of sleep scoring.

• Research Organization: Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)
• Sponsoring Organization: National Science Foundation (NSF); USDOE Laboratory Directed Research and Development (LDRD) Program; USDOE Office of Science (SC)
• Grant/Contract Number: AC02-05CH11231
• OSTI ID: 1713216
• Journal Information: PLoS ONE, Journal Name: PLoS ONE Journal Issue: 12 Vol. 14; ISSN 1932-6203
• Publisher: Public Library of ScienceCopyright Statement
• Country of Publication: United States
• Language: English