Contrastive Learning for Sleep Staging based on Inter Subject Correlation

Tongxu Zhang,Bei Wang
2023-05-05
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 cross-subject 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.
Signal Processing,Machine Learning
What problem does this paper attempt to address?
The paper aims to address the cross-subject issue in automatic sleep staging. Specifically, the paper focuses on the following points: 1. **Cross-subject issue**: Existing deep learning models are affected by individual differences in electrophysiological signals when processing sleep data from different subjects. Therefore, designing an automatic sleep staging model that can handle the cross-subject issue is crucial. 2. **Application of contrastive learning methods**: The authors propose a method based on contrastive learning, utilizing cross-subject correlation theory to explore subject invariance and develop a cross-subject analysis method. This approach aims to overcome the impact of individual differences and improve the model's generalization ability. 3. **Improvement of the MViTime model**: To achieve this goal, the authors improved the MobileViT model to make it suitable for time series tasks and combined it with contrastive learning methods to achieve optimal sleep staging classification results. In summary, the main objective of this paper is to solve the cross-subject issue in automatic sleep staging by introducing contrastive learning methods and the perspective of cross-subject correlation, thereby providing a feasible and implementable solution.