Validation of ‘Somnivore’, a Machine Learning Algorithm for Automated Scoring and Analysis of Polysomnography Data

• Published in: Frontiers in neuroscience Vol. 13; p. 207
• Main Authors: Allocca, Giancarlo, Ma, Sherie, Martelli, Davide, Cerri, Matteo, Del Vecchio, Flavia, Bastianini, Stefano, Zoccoli, Giovanna, Amici, Roberto, Morairty, Stephen R., Aulsebrook, Anne E., Blackburn, Shaun, Lesku, John A., Rattenborg, Niels C., Vyssotski, Alexei L., Wams, Emma, Porcheret, Kate, Wulff, Katharina, Foster, Russell, Chan, Julia K. M., Nicholas, Christian L., Freestone, Dean R., Johnston, Leigh A., Gundlach, Andrew L.
• Format: Journal Article
• Language: English
• Published: Switzerland Frontiers Research Foundation 18.03.2019; Frontiers Media S.A
• Subjects: Agreements; Algorithms; Animal cognition; Artificial intelligence; Automation; Classification; Cognitive ability; Data collection; Engineers; Laboratories; Learning algorithms; Machine learning; machine learning algorithms; Medical research; Mental health; Narcolepsy; Neuroscience; Neurosciences; Physiology; polysomnography; Reclassification; signal processing algorithms; Sleep; Sleep and wakefulness; sleep stage classification; Software; Statistical analysis; wake-sleep stage scoring; Australia; Italy; signal processing algorithms; polysomnography; sleep stage classification; wake–sleep stage scoring; machine learning algorithms
• ISSN: 1662-453X; 1662-4548; 1662-453X
• DOI: 10.3389/fnins.2019.00207

Manual scoring of polysomnography data is labor-intensive and time-consuming, and most existing software does not account for subjective differences and user variability. Therefore, we evaluated a supervised machine learning algorithm, Somnivore , for automated wake-sleep stage classification. We designed an algorithm that extracts features from various input channels, following a brief session of manual scoring, and provides automated wake-sleep stage classification for each recording. For algorithm validation, polysomnography data was obtained from independent laboratories, and include normal, cognitively-impaired, and alcohol-treated human subjects (total = 52), narcoleptic mice and drug-treated rats (total = 56), and pigeons ( = 5). Training and testing sets for validation were previously scored manually by 1-2 trained sleep technologists from each laboratory. -measure was used to assess precision and sensitivity for statistical analysis of classifier output and human scorer agreement. The algorithm gave high concordance with manual visual scoring across all human data (wake 0.91 ± 0.01; N1 0.57 ± 0.01; N2 0.81 ± 0.01; N3 0.86 ± 0.01; REM 0.87 ± 0.01), which was comparable to manual inter-scorer agreement on all stages. Similarly, high concordance was observed across all rodent (wake 0.95 ± 0.01; NREM 0.94 ± 0.01; REM 0.91 ± 0.01) and pigeon (wake 0.96 ± 0.006; NREM 0.97 ± 0.01; REM 0.86 ± 0.02) data. Effects of classifier learning from single signal inputs, simple stage reclassification, automated removal of transition epochs, and training set size were also examined. In summary, we have developed a polysomnography analysis program for automated sleep-stage classification of data from diverse species. Somnivore enables flexible, accurate, and high-throughput analysis of experimental and clinical sleep studies.