arXiv:2606.02166cs.LG2026-06

用Transformer融合多路脑电特征,实现高精度癫痫发作预测

EEG-FuseFormer: A Transformer-Driven Feature Fusion Framework for Seizure Onset Prediction

论文配图:EEG-FuseFormer: A Transformer-Driven Feature Fusion Framework for Seizure Onset Prediction
图 1 · 摘自论文原文
  • 通过CNN-LSTM与ResNet-18分别提取原始信号和频谱特征,再用Transformer融合
  • 在CHB-MIT数据集上平均召回率达98.85%,优于多数现有方法
  • 跨患者测试中微调预训练模型可显著提升性能,适合临床部署

癫痫是全球最常见的神经系统疾病之一,以反复发作的癫痫发作为特征,严重影响患者生活质量。尽管诊断技术不断进步,但因发作不可预测,风险控制仍具挑战。准确预测发作可有效降低风险。本文提出EEG-FuseFormer,一种基于Transformer的特征融合框架,融合来自卷积神经网络-长短期记忆网络(CNN-LSTM)和ResNet-18的中间特征。CNN-LSTM直接从原始信号中捕获时空特征,而ResNet-18则从脑电图(EEG)信号的短时傅里叶变换(STFT)表示中提取特征。特征融合采用Transformer编码器,最终通过全连接层生成预测结果。在CHB-MIT数据集上验证,该模型平均召回率达98.85%,优于大多数先进方法。研究还评估了模型在跨患者测试中的泛化能力。在跨患者验证框架下,对少量目标患者数据进行预训练模型微调(目标适应),相比传统跨患者验证方法,能显著提升召回率、精确率和F1分数。最后,评估了模型在不同硬件平台上的运行时计算复杂度,揭示性能与复杂度之间的权衡。

原文摘要 · Abstract (English)

Epilepsy is one of the most common neurological disorders globally, characterized by recurring seizures and significantly impacting the quality of life. Despite advancements in diagnostic techniques, the mitigation of risks faced by epilepsy patients remains challenging due to the unpredictability of seizure events. An accurate forecast of seizure onset helps to reduce risks in epilepsy patients. In this paper, we propose EEG-FuseFormer, a transformer-based feature fusion framework for seizure-onset prediction that combines intermediate features extracted from Convolutional Neural Networks-Long Short-Term Memory (CNN-LSTM) and ResNet-18 networks. The CNN-LSTM architecture captures both spatial and temporal features directly from the raw signal, whereas the ResNet-18 extracts features from the Short-Time Fourier Transform (STFT) representation of the EEG signals. Fusion is carried out using a transformer encoder, and the final prediction is generated using fully connected dense layers. The CHB-MIT dataset was used to validate the proposed model. The results show that the proposed model achieves a mean recall of 98.85% and outperforms most of the state-of-the-art methods. This study evaluates the ability of the proposed feature fusion model to generalize in cross-patient testing scenarios. Fine-tuning pre-trained models on limited target patient data (target adaptation) within the cross-patient validation framework results in higher recall, precision, and F1-score metrics in comparison to the conventional cross-patient validation approach. Finally, the runtime-based computational complexity of the model is assessed across diverse hardware platforms to highlight the performance-complexity trade-off.

癫痫预测脑电图Transformer特征融合

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