arXiv:2501.16329cs.LGcs.AI2025-01被引 3

提出跨模态自蒸馏模型,提升多通道睡眠分期准确率

sDREAMER: Self-distilled Mixture-of-Modality-Experts Transformer for Automatic Sleep Staging

  • 设计三路混合模态专家结构,增强脑电与肌电信息交互
  • 自蒸馏训练使单通道输入性能超越同类单通道模型
  • 统一框架支持单/多通道输入,适合临床实际应用

基于脑电(EEG)和肌电(EMG)信号的自动睡眠分期是睡眠研究的重要方向。现有方法存在两大缺陷:模态间信息交互有限,且缺乏能处理多种输入源的统一模型。为此,我们提出新型睡眠分期模型sDREAMER,强调跨模态交互与单通道性能。具体设计三路径混合模态专家(MoME)结构,分别处理EEG、EMG及混合信号,采用部分共享权重。进一步提出自蒸馏训练策略,促进模态间信息融合。模型支持多通道输入训练,可对单通道或双通道输入进行分类。实验表明,该模型在多通道推理上优于现有基于Transformer的睡眠分期方法;在单通道推理中,也超越了仅用单通道数据训练的Transformer模型。

原文摘要 · Abstract (English)

Automatic sleep staging based on electroencephalography (EEG) and electromyography (EMG) signals is an important aspect of sleep-related research. Current sleep staging methods suffer from two major drawbacks. First, there are limited information interactions between modalities in the existing methods. Second, current methods do not develop unified models that can handle different sources of input. To address these issues, we propose a novel sleep stage scoring model sDREAMER, which emphasizes cross-modality interaction and per-channel performance. Specifically, we develop a mixture-of-modality-expert (MoME) model with three pathways for EEG, EMG, and mixed signals with partially shared weights. We further propose a self-distillation training scheme for further information interaction across modalities. Our model is trained with multi-channel inputs and can make classifications on either single-channel or multi-channel inputs. Experiments demonstrate that our model outperforms the existing transformer-based sleep scoring methods for multi-channel inference. For single-channel inference, our model also outperforms the transformer-based models trained with single-channel signals.

睡眠分期跨模态Transformer自蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。