提出ISAM-MTL模型,提升跨被试脑机接口的分类准确率与稳定性。
ISAM-MTL: Cross-subject multi-task learning model with identifiable spikes and associative memory networks
- 用可识别脉冲表示和关联记忆网络实现跨被试多任务学习
- 在两个竞赛数据集上平均准确率提升,被试间性能差异减小
- 支持少样本快速校准,结果可解释,适合临床脑机系统
EEG在不同被试间的差异会降低现有深度学习模型性能,制约脑机接口(BCI)发展。本文提出ISAM-MTL模型,基于可识别脉冲(IS)表征与关联记忆(AM)网络的多任务学习(MTL)EEG分类方法。该模型将每位被试的分类视为独立任务,利用跨被试数据训练实现特征共享。模型包含一个脉冲特征提取器,用于捕捉跨被试共性特征;以及一个受海布学习启发的被试特异性双向关联记忆网络,实现高效快速的被试内分类。通过标签引导的变分推断构建可识别脉冲表示,增强分类精度。在两个BCI竞赛数据集上的实验表明,ISAM-MTL提升了跨被试EEG分类的平均准确率,同时降低了被试间性能波动。模型还展现出少样本学习能力及可识别的神经活动特性,支持脑机系统快速、可解释的校准。
原文摘要 · Abstract (English)
Cross-subject variability in EEG degrades performance of current deep learning models, limiting the development of brain-computer interface (BCI). This paper proposes ISAM-MTL, which is a multi-task learning (MTL) EEG classification model based on identifiable spiking (IS) representations and associative memory (AM) networks. The proposed model treats EEG classification of each subject as an independent task and leverages cross-subject data training to facilitate feature sharing across subjects. ISAM-MTL consists of a spiking feature extractor that captures shared features across subjects and a subject-specific bidirectional associative memory network that is trained by Hebbian learning for efficient and fast within-subject EEG classification. ISAM-MTL integrates learned spiking neural representations with bidirectional associative memory for cross-subject EEG classification. The model employs label-guided variational inference to construct identifiable spike representations, enhancing classification accuracy. Experimental results on two BCI Competition datasets demonstrate that ISAM-MTL improves the average accuracy of cross-subject EEG classification while reducing performance variability among subjects. The model further exhibits the characteristics of few-shot learning and identifiable neural activity beneath EEG, enabling rapid and interpretable calibration for BCI systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。