Contrastive Learning for Sleep Staging based on Inter Subject Correlation
Zhang, T.; Wang, B.
Show abstract
In recent years, multitudes of researches have applied deep learning to automatic sleep stage classification. Whereas actually, these works have paid less attention to the issue of cross-subject in sleep staging. At the same time, emerging neuroscience theories on inter-subject correlations can provide new insights for cross-subject analysis. This paper presents the MViTime model that have been used in sleep staging study. And we implement the inter-subject correlation theory through contrastive learning, providing a feasible solution to address the crosssubject problem in sleep stage classification. Finally, experimental results and conclusions are presented, demonstrating that the developed method has achieved state-of-the-art performance on sleep staging. The results of the ablation experiment also demonstrate the effectiveness of the cross-subject approach based on contrastive learning. The code can be accessed through: https://github.com/jukieCheung/MViTime.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- SingleChannelNet: A Model for Automatic Sleep Stage Classification with Raw Single-Channel EEG 99%
- Evaluating three different adaptive decomposition methods for EEG signal seizure detection and classification 94%
- A Transfer Entropy-based methodology to analyze information flow under eyes-open and eyes-closed conditions with a clinical perspective 94%
Similar papers in this journal
- SleepSatelightFTC: A Lightweight and Interpretable Deep Learning Model for Single-Channel EEG-Based Sleep Stage Classification 98%
- HeartNet: Self Multi-Head Attention Mechanism via Convolutional Network with Adversarial Data Synthesis for ECG-based Arrhythmia Classification 95%
- Measuring Connectivity in Linear Multivariate Processes with Penalized Regression Techniques 93%
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
- A foundational transformer leveraging full night, multichannel sleep study data accurately classifies sleep stages 95%
- Corticothalamic modelling of sleep neurophysiology with applications to mobile EEG 94%
- Evaluation of Dreem headband for sleep staging and EEG spectral analysis in people living with Alzheimer’s and older adults 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.