arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.03430eess.SP

基于多通道多域的单通道脑电睡眠分期知识蒸馏算法

Multi-Channel Multi-Domain based Knowledge Distillation Algorithm for Sleep Staging with Single-Channel EEG

Chao Zhang, Yiqiao Liao, Siqi Han, Milin Zhang, Zhihua Wang, Xiang Xie

AI总结:

本文提出基于多通道多域的知识蒸馏算法,通过多通道预训练与单通道微调方案,结合输出和中间特征迁移,实现86.5%的单通道脑电睡眠分期准确率,有效将多域多通道知识迁移至单通道模型。

摘要:

本文提出了一种基于多通道多域(MCMD)的单通道脑电(EEG)睡眠分期知识蒸馏算法。该算法同时学习来自不同域和不同通道的知识。本研究采用多通道预训练和单通道微调方案,将源域中来自不同通道的知识迁移到目标域的单通道模型中。采用预训练的教师-学生模型方案,结合目标域中的输出迁移和中间特征迁移,将知识从多通道教师模型蒸馏到单通道学生模型。该算法实现了86.5%的最先进单通道睡眠分期准确率,仅比最先进的多通道模型降低0.6%,相比基线模型提升了2%。实验结果表明,来自多域(不同数据集)和多通道(如EMG、EOG)的知识可以迁移到单通道睡眠分期中。

英文摘要:

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%, with only 0.6% deterioration from the state-of-the-art multi-channel model. There is an improvement of 2% compared to the baseline model. The experimental results show that knowledge from multiple domains (different datasets) and multiple channels (e.g. EMG, EOG) could be transferred to single-channel sleep staging.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑