StAGN: Spatial-Temporal Adaptive Graph Network via Contrastive Learning for Sleep Stage Classification

Sleep stage classification is a critical concern in sleep quality assessment and disease diagnosis. Graph network based studies for sleep stages classification have achieved promising performance. However, these studies still ignored the importance of learning morphological feature information with the spatial-temporal relationship among multi-modal physiological signals. To address this issue, we propose a Spatial-temporal Adaptive Graph Network named StAGN for sleep stage classification. The main advantage of StAGN is to adaptively learn the time-dependent and channel-wise interdependent waveform morphological features in multi-modal physiological signals. Such features will be extracted by a modified 1-dimensional ResNet with a projection short-cut connection and adjusted by a joint spatial-temporal attention, thereby best serving the followed brain topological connection graph network for sleep stage classification. Meanwhile, we leverage the contrastive learning scheme with label information to further improve classification accuracy without changing the signal morphology. Experiment results on two publicly available sleep datasets of ISRUC-S1 and ISRUC-S3 show that the proposed StAGN can achieve a competitive performance for sleep stage classification, which is superior to the state-of-the-art counterparts.