Multi-Channel Multi-Domain based Knowledge Distillation Algorithm for Sleep Staging with Single-Channel EEG
IEEE Transactions on Circuits and Systems II: Express Briefs(2024)
摘要
This paper proposed a Multi-Channel Multi-Domain (MCMD) based knowledge
distillation algorithm for sleep staging using single-channel EEG. Both
knowledge from different domains and different channels are learnt in the
proposed algorithm, simultaneously. A multi-channel pre-training and
single-channel fine-tuning scheme is used in the proposed work. The knowledge
from different channels in the source domain is transferred to the
single-channel model in the target domain. A pre-trained teacher-student model
scheme is used to distill knowledge from the multi-channel teacher model to the
single-channel student model combining with output transfer and intermediate
feature transfer in the target domain. The proposed algorithm achieves a
state-of-the-art single-channel sleep staging accuracy of 86.5
deterioration from the state-of-the-art multi-channel model. There is an
improvement of 2
that knowledge from multiple domains (different datasets) and multiple channels
(e.g. EMG, EOG) could be transferred to single-channel sleep staging.
更多查看译文
关键词
Sleep staging,transfer learning,knowledge distillation,single-channel EEG,brain-computer interface
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要