# Exploiting Spatial-temporal Data for Sleep Stage Classification via Hypergraph Learning

arXiv: https://arxiv.org/abs/2309.02124 · HTML: https://mlsanity.com/paper/2309.02124/

## Key facts

- Cited by 2 papers in the map (data through 14 Oct 2026).
- Builds on 7 of its 20 references that are in the map.
- First posted on arXiv 5 Sep 2023.

## Builds on (most cited first, 7 of 7)

- [Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting](https://mlsanity.com/paper/1709.04875/) (2017)
- [DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG](https://mlsanity.com/paper/1703.04046/) (2017)
- [Graph-Based Deep Learning for Medical Diagnosis and Analysis: Past, Present and Future](https://mlsanity.com/paper/2105.13137/) (2021)
- [A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series](https://mlsanity.com/paper/1707.03321/) (2017)
- [Multi-View Spatial-Temporal Graph Convolutional Networks with Domain Generalization for Sleep Stage Classification](https://mlsanity.com/paper/2109.01824/) (2021)
- [Learning from Heterogeneity: A Dynamic Learning Framework for Hypergraphs](https://mlsanity.com/paper/2307.03411/) (2023)
- [An Adaptive Federated Relevance Framework for Spatial Temporal Graph Learning](https://mlsanity.com/paper/2206.03420/) (2022)

## Built on by (most cited first, 2 of 2)

- [CHASE: A Causal Hypergraph based Framework for Root Cause Analysis in Multimodal Microservice Systems](https://mlsanity.com/paper/2406.19711/) (2024)
- [MetaSTH-Sleep: Towards Effective Few-Shot Sleep Stage Classification for Health Management with Spatial-Temporal Hypergraph Enhanced Meta-Learning](https://mlsanity.com/paper/2505.17142/) (2025)

## Abstract (arXiv)

Sleep stage classification is crucial for detecting patients' health conditions. Existing models, which mainly use Convolutional Neural Networks (CNN) for modelling Euclidean data and Graph Convolution Networks (GNN) for modelling non-Euclidean data, are unable to consider the heterogeneity and interactivity of multimodal data as well as the spatial-temporal correlation simultaneously, which hinders a further improvement of classification performance. In this paper, we propose a dynamic learning framework STHL, which introduces hypergraph to encode spatial-temporal data for sleep stage classification. Hypergraphs can construct multi-modal/multi-type data instead of using simple pairwise between two subjects. STHL creates spatial and temporal hyperedges separately to build node correlations, then it conducts type-specific hypergraph learning process to encode the attributes into the embedding space. Extensive experiments show that our proposed STHL outperforms the state-of-the-art models in sleep stage classification tasks.