Sleep-Energy: An Energy Optimization Method to Sleep Stage Scoring.

Journal: IEEE access : practical innovations, open solutions
Published Date:

Abstract

Sleep is essential for physical and mental health. Polysomnography (PSG) procedures are labour-intensive and time-consuming, making diagnosing sleep disorders difficult. Automatic sleep staging using Machine Learning (ML) - based methods has been studied extensively, but frequently provides noisier predictions incompatible with typical manually annotated hypnograms. We propose an energy optimization method to improve the quality of hypnograms generated by automatic sleep staging procedures. The method evaluates the system's total energy based on conditional probabilities for each epoch's stage and employs an energy minimisation procedure. It can be used as a meta-optimisation layer over the sleep stage sequences generated by any classifier that generates prediction probabilities. The method improved the accuracy of state-of-the-art Deep Learning models in the Sleep EDFx dataset by 4.0% and in the DRM-SUB dataset by 2.8%.

Authors

  • Bruno Aristimunha
    Center for Mathematics, Computing and Cognition (CMCC)Federal University of ABC (UFABC) São Paulo 09210-580 Brazil.
  • Alexandre Janoni Bayerlein
    Center for Mathematics, Computing and Cognition (CMCC)Federal University of ABC (UFABC) São Paulo 09210-580 Brazil.
  • M Jorge Cardoso
    Department of Biomedical EngineeringSchool of Biomedical Engineering and Imaging SciencesKing's College London WC2R 2LS London U.K.
  • Walter Hugo Lopez Pinaya
    Department of Biomedical EngineeringSchool of Biomedical Engineering and Imaging SciencesKing's College London WC2R 2LS London U.K.
  • Raphael Yokoingawa De Camargo
    Center for Mathematics, Computing and Cognition (CMCC)Federal University of ABC (UFABC) São Paulo 09210-580 Brazil.

Keywords

No keywords available for this article.