arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2310.10550cs.LGcs.AI
收藏

使用空间注意力的深度学习应用于不同导联设置的EEG数据

Deep learning applied to EEG data with different montages using spatial attention

SCCN, INC, UCSD(SCCN、INC、加州大学圣地亚哥分校);CerCo CNRS, Paul Sabatier University(CerCo 法国国家科学研究中心、保罗·萨巴蒂埃大学)

Dung Truong, Muhammad Abdullah Khalid, Arnaud Delorme

AI总结:

本研究提出将空间注意力应用于EEG电极坐标,实现不同导联设置原始数据的通道协调,并在性别分类任务上验证了该方法能提升模型性能,且优于固定导联设置的模型。

摘要:

深度学习处理原始脑电图(EEG)数据中复杂大脑动态并提取相关信息的能力已在近期多项工作中得到验证。然而,深度学习模型也被证明在大型数据语料上表现最佳。在处理EEG时,一种自然的方法是将来自不同实验的EEG数据集合并,以训练大型深度学习模型。然而,大多数EEG实验使用自定义的通道导联设置,这要求将数据转换到一个公共空间。以往的方法使用原始EEG信号提取感兴趣的特征,并专注于在EEG数据集之间使用公共特征空间。虽然这是一种合理的方法,但它未能充分利用原始EEG数据的潜在丰富性。在此,我们探索将空间注意力应用于EEG电极坐标,以实现原始EEG数据的通道协调,从而使我们能够在使用不同导联设置的EEG数据上训练深度学习模型。我们在性别分类任务上测试了该模型。我们首先表明,空间注意力提高了模型性能。然后,我们表明,在使用不同通道导联设置的数据上训练的深度学习模型,其性能显著优于在固定的23通道和128通道数据导联设置上训练的深度学习模型。

英文摘要:

The ability of Deep Learning to process and extract relevant information in complex brain dynamics from raw EEG data has been demonstrated in various recent works. Deep learning models, however, have also been shown to perform best on large corpora of data. When processing EEG, a natural approach is to combine EEG datasets from different experiments to train large deep-learning models. However, most EEG experiments use custom channel montages, requiring the data to be transformed into a common space. Previous methods have used the raw EEG signal to extract features of interest and focused on using a common feature space across EEG datasets. While this is a sensible approach, it underexploits the potential richness of EEG raw data. Here, we explore using spatial attention applied to EEG electrode coordinates to perform channel harmonization of raw EEG data, allowing us to train deep learning on EEG data using different montages. We test this model on a gender classification task. We first show that spatial attention increases model performance. Then, we show that a deep learning model trained on data using different channel montages performs significantly better than deep learning models trained on fixed 23- and 128-channel data montages.

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

↑