arXiv:2410.11200cs.LGcs.AI2024-10中稿 · ICDM2024被引 4

用单通道脑电图学出多通道效果,提升癫痫预测准确率

SplitSEE: A Splittable Self-supervised Framework for Single-Channel EEG Representation Learning

  • 设计可拆分结构,分别学习时频域特征并强制聚类一致
  • 单通道训练超越多通道基线,在临床数据集上表现更优
  • 只需一次微调即可适配新任务,适合医疗场景快速部署

尽管端到端多通道脑电图学习方法展现出显著潜力,但在神经诊断中常受限于颅内脑电资源。当仅有单通道脑电图时,如何学习对多通道鲁棒且可扩展的任务表征(如癫痫预测)?本文提出SplitSEE,一种结构可拆分的自监督框架,用于单通道脑电图的有效时频表征学习。核心思想是将时域与频域视为两个独立视角,要求学习到的表示具有相同的聚类分配。为此,我们设计了两个域特定模块,独立学习域特异性表征,解决传统谱图方法中的时频权衡问题;并引入新型聚类损失,度量信息相似性,促使两域表示一致描述同一输入。SplitSEE采用预训练-微调框架,具备:(a) 效果性:仅依赖单通道脑电图即超越多通道基线;(b) 鲁棒性:跨通道性能波动小,临床数据集上表现优异;(c) 可扩展性:仅需一次微调即可实现高且稳定的性能,适用于部分模型层的灵活使用。

原文摘要 · Abstract (English)

While end-to-end multi-channel electroencephalography (EEG) learning approaches have shown significant promise, their applicability is often constrained in neurological diagnostics, such as intracranial EEG resources. When provided with a single-channel EEG, how can we learn representations that are robust to multi-channels and scalable across varied tasks, such as seizure prediction? In this paper, we present SplitSEE, a structurally splittable framework designed for effective temporal-frequency representation learning in single-channel EEG. The key concept of SplitSEE is a self-supervised framework incorporating a deep clustering task. Given an EEG, we argue that the time and frequency domains are two distinct perspectives, and hence, learned representations should share the same cluster assignment. To this end, we first propose two domain-specific modules that independently learn domain-specific representation and address the temporal-frequency tradeoff issue in conventional spectrogram-based methods. Then, we introduce a novel clustering loss to measure the information similarity. This encourages representations from both domains to coherently describe the same input by assigning them a consistent cluster. SplitSEE leverages a pre-training-to-fine-tuning framework within a splittable architecture and has following properties: (a) Effectiveness: it learns representations solely from single-channel EEG but has even outperformed multi-channel baselines. (b) Robustness: it shows the capacity to adapt across different channels with low performance variance. Superior performance is also achieved with our collected clinical dataset. (c) Scalability: With just one fine-tuning epoch, SplitSEE achieves high and stable performance using partial model layers.

脑电图自监督单通道聚类

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