用通道独立的自编码器+单头Transformer,高效识别脑电异常
CwA-T: A Channelwise AutoEncoder with Transformer for EEG Abnormality Detection

- 通道独立压缩原始脑电信号,保留生物意义特征
- 在TUH数据集上达85.0%准确率、76.2%敏感性、91.2%特异性
- 仅需202M FLOPs和2.9M参数,适合临床部署
脑电图(EEG)信号对检测异常脑活动至关重要,但其高维度与复杂性给有效分析带来挑战。本文提出CwA-T框架,结合通道式基于CNN的自编码器与单头Transformer分类器,实现高效的EEG异常检测。通道式自编码器在压缩原始信号的同时保持通道独立性,降低计算成本并保留生物学相关特征。压缩后的表示输入Transformer分类器,有效建模长期依赖以区分正常与异常信号。在TUH异常脑电语料库上的评估显示,该模型在病例级达到85.0%准确率、76.2%敏感性和91.2%特异性,优于EEGNet、Deep4Conv和FusionCNN等基线模型。此外,CwA-T仅需202M FLOPs和2.9M参数,显著优于其他基于Transformer的方法。其通道式设计保持可解释性,展现出在神经科学研究与临床实践中的巨大潜力。源代码见https://github.com/YossiZhao/CAE-T。
原文摘要 · Abstract (English)
Electroencephalogram (EEG) signals are critical for detecting abnormal brain activity, but their high dimensionality and complexity pose significant challenges for effective analysis. In this paper, we propose CwA-T, a novel framework that combines a channelwise CNN-based autoencoder with a single-head transformer classifier for efficient EEG abnormality detection. The channelwise autoencoder compresses raw EEG signals while preserving channel independence, reducing computational costs and retaining biologically meaningful features. The compressed representations are then fed into the transformer-based classifier, which efficiently models long-term dependencies to distinguish between normal and abnormal signals. Evaluated on the TUH Abnormal EEG Corpus, the proposed model achieves 85.0% accuracy, 76.2% sensitivity, and 91.2% specificity at the per-case level, outperforming baseline models such as EEGNet, Deep4Conv, and FusionCNN. Furthermore, CwA-T requires only 202M FLOPs and 2.9M parameters, making it significantly more efficient than transformer-based alternatives. The framework retains interpretability through its channelwise design, demonstrating great potential for future applications in neuroscience research and clinical practice. The source code is available at https://github.com/YossiZhao/CAE-T.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。