6533b86cfe1ef96bd12c80a9

RESEARCH PRODUCT

Automatic sleep scoring: A deep learning architecture for multi-modality time series

Fan LiFengyu CongDongdong ZhouTapani RistaniemiRui Yan

subject

0301 basic medicineProcess (engineering)Computer sciencePolysomnographyPolysomnographyMachine learningcomputer.software_genreuni (lepotila)03 medical and health sciencesDeep Learning0302 clinical medicinepolysomnographymedicineHumansBlock (data storage)Sleep Stagesmedicine.diagnostic_testArtificial neural networksignaalinkäsittelybusiness.industryunitutkimusGeneral NeuroscienceDeep learningdeep learningsignaalianalyysiElectroencephalographyautomatic sleep scoringmulti-modality analysiskoneoppiminen030104 developmental biologyMemory moduleSleep StagesArtificial intelligenceSleepTransfer of learningbusinesscomputer030217 neurology & neurosurgery

description

Background: Sleep scoring is an essential but time-consuming process, and therefore automatic sleep scoring is crucial and urgent to help address the growing unmet needs for sleep research. This paper aims to develop a versatile deep-learning architecture to automate sleep scoring using raw polysomnography recordings. Method: The model adopts a linear function to address different numbers of inputs, thereby extending model applications. Two-dimensional convolution neural networks are used to learn features from multi-modality polysomnographic signals, a “squeeze and excitation” block to recalibrate channel-wise features, together with a long short-term memory module to exploit long-range contextual relation. The learnt features are finally fed to the decision layer to generate predictions for sleep stages. Result: Model performance is evaluated on three public datasets. For all tasks with different available channels, our model achieves outstanding performance not only on healthy subjects but even on patients with sleep disorders (SHHS: Acc-0.87, K-0.81; ISRUC: Acc-0.86, K-0.82; Sleep-EDF: Acc-0.86, K-0.81). The highest classification accuracy is achieved by a fusion of multiple polysomnographic signals. Comparison: Compared to state-of-the-art methods that use the same dataset, the proposed model achieves a comparable or better performance, and exhibits low computational cost. Conclusions: The model demonstrates its transferability among different datasets, without changing model architecture or hyper-parameters across tasks. Good model transferability promotes the application of transfer learning on small group studies with mismatched channels. Due to demonstrated availability and versatility, the proposed method can be integrated with diverse polysomnography systems, thereby facilitating sleep monitoring in clinical or routine care. peerReviewed

https://doi.org/10.1016/j.jneumeth.2020.108971