An Interpretable Single-Channel EEG Sleep Staging Model Based on Prototype Matching and Multitask Learning

Sleep staging is an essential technique for evaluating sleep quality. Recently, deep learning-based sleep staging methods have achieved high performance. However, existing methods are unable to provide a decision basis for classification, which limits their clinical reliability. To address this issue, we propose a novel model, which integrates prototype matching with multitask learning (MTL) to enhance interpretability without sacrificing performance. Our method captures significant waveform prototypes from various stages of sleep EEG as templates. Subsequently, the similarity score is calculated by comparing the deep representations of testing EEG segments with these prototypes as the basis for classification. To further learn the sleep-stage transition rules and improve performance, we design an MTL framework. We take the sample-level sleep-stage classification and sleep transition stage discrimination as two auxiliary tasks and take the sequence-level sleep-stage classification as the main task. Compared to other state-of-the-art (SOTA) methods, our method demonstrates competitive performance, achieving an overall accuracy of 85.9% and macroaveraged $F1$ (MF1) score of 81.0% on the SleepEDF dataset, and accuracy of 86.5% and an MF1 score of 80.6% on the MASS-SS3 dataset. Importantly, our method provides a visualization of the prototype set, and the salient sleep waveforms in the prototype set representing unique sleep states are labeled by the expert physician. In summary, our method enhances model interpretability while ensuring robust sleep staging performance, promising a reliable and clinically applicable tool for automatic sleep staging.

I. Introduction

Sleep is fundamental for maintaining physical and mental health, providing essential rest and recovery that enhances overall well-being and life quality [1]. The rising prevalence of sleep disorders poses a global health concern, affecting a significant portion of the population [2], [3].