Sorry, you need to enable JavaScript to visit this website.

SELF-SUPERVISED LEARNING FOR SLEEP STAGE CLASSIFICATION WITH TEMPORALAUGMENTATION AND FALSE NEGATIVE SUPPRESSION

DOI:
10.60864/he43-p476
Citation Author(s):
Fangyao Shen, Zehao Zhang, Yong Peng , Hongjie Guo, Lina Chen, Hong Gao
Submitted by:
Zehao Zhang
Last updated:
6 June 2024 - 10:28am
Document Type:
Poster
Document Year:
2024
Event:
Presenters:
Zehao Zhang
Paper Code:
BISP-P3
 

Self-supervised learning has been gaining attention in the field of sleep stage classification. It learns representations with unlabeled electroencephalography (EEG) signals, which alleviates the cost of labeling for specialists. However, most self-supervised approaches assume only the two augmented views from the same EEG sample is a positive pair, which suffers from the false negative problem. Therefore, we propose a new model named Temporal Augmentation and False Negative Suppression (TA-FNS) to solve the problem. Specifically, it first generates two augmented views for each EEG sample. Then the temporal augmentation module is proposed to learn temporal features during sleep from augmented views. Based on temporal features, intra-view and inter-view sample similarity matrices are calculated. Finally, the false negative suppression module identifies and eliminates potential false negatives according to the consistency between intra-view and inter-view similarity matrices. TA-FNS not only achieves state-of-the-art performances on Sleep-EDF and ISRUC datasets, but also learns semantic representation from EEG of different sleep stages, which demonstrates the effectiveness of it in mitigating the false negative problem.

up
0 users have voted: