arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.16626eess.SPcs.LG

使用图卷积变分自编码器从EEG中学习被试表征

Subject Representation Learning from EEG using Graph Convolutional Variational Autoencoders

Queen's University(女王大学)

Aditya Mishra, Ahnaf Mozib Samin, Ali Etemad, Javad Hashemi

AI总结:

提出基于图卷积的变分自编码器GC-VASE,结合对比学习与分离潜空间从EEG数据中学习被试表征,并引入注意力适配器降低适应新被试的计算成本,在多个数据集上取得SOTA表现。

摘要:

我们提出了GC-VASE,一种基于图卷积的变分自编码器,利用对比学习从EEG数据中进行被试表征学习。我们的方法使用为被试识别量身定制的分离潜空间架构,成功学习了稳健的被试特定潜表征。为了增强模型对未见被试的适应性而无需大量重新训练,我们引入了基于注意力的适配器网络进行微调,这降低了将模型适应新被试的计算成本。我们的方法显著优于其他深度学习方法,取得了最先进的结果,在ERP-Core数据集上被试平衡准确率为89.81%,在SleepEDFx-20数据集上为70.85%。使用适配器和注意力层进行被试自适应微调后,GC-VASE在ERP-Core上将被试平衡准确率进一步提高到90.31%。此外,我们进行了详细的消融研究,以突出我们方法中关键组件的影响。

英文摘要:

We propose GC-VASE, a graph convolutional-based variational autoencoder that leverages contrastive learning for subject representation learning from EEG data. Our method successfully learns robust subject-specific latent representations using the split-latent space architecture tailored for subject identification. To enhance the model's adaptability to unseen subjects without extensive retraining, we introduce an attention-based adapter network for fine-tuning, which reduces the computational cost of adapting the model to new subjects. Our method significantly outperforms other deep learning approaches, achieving state-of-the-art results with a subject balanced accuracy of 89.81% on the ERP-Core dataset and 70.85% on the SleepEDFx-20 dataset. After subject adaptive fine-tuning using adapters and attention layers, GC-VASE further improves the subject balanced accuracy to 90.31% on ERP-Core. Additionally, we perform a detailed ablation study to highlight the impact of the key components of our method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑