EEG神经网络分类的深度学习方法
导读
大脑活动模式分类是理解大脑,进行神经诊断和设计闭环神经接口的重要工具。头皮脑电图(EEG)以其非侵入性和低成本的优点被用于神经信号的分类,研究人员已经使用了各种机器学习方法来进行研究。近年来,深度学习(DL)因其能够在阐明分类的相关特征的同时显著提高许多领域的分类性能而广受欢迎。本文旨在为对基于EEG的深度学习研究感兴趣的研究人员提供全面的参考来源。
前言
深度学习(DL)在EEG应用中的潜力
随着深度学习算法的使用,EEG分析领域呈指数级增长。对先前研究的分析为DL架构的发展趋势提供了丰富的来源。在EEG分析中,卷积神经网络(CNN)、深度信念网络(DBM)、循环神经网络(RNN)和长短期记忆网络(LSTM)越来越流行。这些深度学习算法已应用于各种EEG分类任务。具体而言,Craik等人分析了90项研究发现,大多数EEG分类研究可分为六大类:情绪识别、运动想象、心理负荷、癫痫检测、事件相关电位和睡眠阶段评分。根据任务类型对研究进行了分析,并给出了对最能成功分类的DL架构类型的建议。此外,Craik还指出了特征选择方法的总体分布,特别是针对六大任务类别的分布,如图1所示。除此之之外,少有研究尝试使用DL算法来提高我们对阿尔茨海默症,抑郁症等方面的理解。
图1
有些综述还提供了对深度学习架构设计选择的进一步分析。Craik等人就DBN、CNN和RNN(该综述中发现的三种性能最高的架构类型)的特定架构设计提供了建议。具体而言,具有三个受限玻尔兹曼机(RBM)的深度信念网络(DBN)平均性能优于其他DBN设计。当使用信号值作为输入时,具有四到五个卷积层的CNN的性能优于其他公式,而当使用图像作为输入时,具有两个卷积层的CNN的性能优于其他变式。最后,研究发现具有两个LSTM层的RNN性能优于此类架构的其他变式。Roy等人发现,应用于EEG的DL架构通常使用3-10层,而不考虑具体的DL架构类型或输入公式。
深度学习方法概述
图2概览了EEG文献中常用的DL方法。接下来描述和总结了以往的EEG DL分类任务研究中使用的典型输入特征和DL方法。
图2
输入特征
①时域特征
②频域特征
③图像特征
架构
①神经网络(NNs)
②自编码器(AEs)
③受限玻尔兹曼机(RBMs)
④卷积神经网络(CNNs)
⑤循环神经网络(RNNs)
RNNs是一种特殊类型的NN,它通过在整个数据持续时间内解析权重来保留输入时间序列中的权值。与CNNs相比,RNNs最初用于对数据建模,包括时间序列、语言建模和语音合成等顺序特征。由于RNNs能够对序列依赖性进行建模,因此它是基于EEG的BCI的必然选择,其中大脑信号被视为时间序列。
特定于任务的分类
图3.一般意志过程的示意图。
①意志过程
A.运动想象(MI)
B.精神疲劳
C.心理负荷
一些研究进行了不同的工作记忆任务,试图引发不同水平的任务难度。大多数研究都试图将RNN单独或与CNN联合添加到其模型中。其中,Bashivan等人首次使用了深度循环卷积神经网络(RCNN)来预测心理负荷。然后,他们将预处理的EEG转换为一组保留拓扑结构的多谱图像,这与之前不考虑这种空间信息的研究不同。然后,他们训练了用于心理认知负荷分类的RCNN,与当时最先进的(SOTA)相比,该模型的分类精度(初始精度为15.3%)显著提高了8.9%。
②外部刺激
与运动想象等意志过程不同,光和声音形式等外部刺激(图4)可以在脑电波中引发特定的反应,这些反应能够触发用于某些接口的系统,如BCI。使用外部刺激的一些优点是被试变异性相对较低。
图4.一般外部刺激的示意图。
A.视觉诱发电位(VEP)和快速序列视觉呈现任务(RSVP)
B.稳态视觉诱发电位(SSVEP)
SSVEP向被试提供恒定频率的闪烁刺激,并且以相对较高的信息传输速率(ITR)和信噪比(SNR)而闻名。关于EEGNet研究,其他一些研究指出的一个方面是,对于他们使用的每个数据集,都需要不同的预处理。在EEG分析的典型管道中,应用特定于任务的预处理来剔除伪迹和工程特征,但目前对端到端管道的研究兴趣也越来越大。DL的潜力之一是其自动特征提取能力。这在EEG分析中特别有用,因为我们可以输入原始数据,剔除伪迹,提取特征,并对输出进行分类,而无需依赖手动选择和特征加工。尽管这是最具挑战性的领域之一,但人们对其应用(EEG分析或其他方面)有着浓厚的兴趣,这通常被称为端到端学习。
③医疗应用
癫痫是一种神经系统疾病,患者的大脑活动会变得异常。根据目前的临床知识,尚无直接的方法来预测或治愈癫痫。但是,研究人员一般会使用EEG作为非侵入性工具(图5)来阐明这种情况。如(1)识别有风险的人群,(2)对大脑状态进行分类(如正常与发作),(3)预测癫痫发作,以及(4)通过脑部刺激减轻癫痫发作。前两个用途的相关研究最多,因为癫痫和非癫痫性EEG信号的识别可以被视为一种典型的分类问题。最后一个用途需要在大脑成像的基础上进行刺激,并且实证研究表明,深部脑刺激(DBS)可减轻癫痫的发作。
图5.一般医疗应用的示意图。
B.阿尔茨海默症
C.睡眠分期
据美国国立卫生研究院(NIH)的《患者和护理者指南》所示,睡眠有两种基本类型:快速眼动(REM)睡眠和非REM睡眠。非快速眼动睡眠进一步划分成三个阶段。许多睡眠阶段的研究都依赖于EEG。通过EEG进行睡眠阶段分类的系统综述显示,许多早期研究尝试使用DL方法(分类为人工神经网络)进行睡眠阶段分类。其中最著名的研究工作是DeepSleepNet,它是一种基于原始单通道EEG的专用深度神经网络,结合了1D CNN和双向LSTM模块。这种单通道深度学习模型表明,睡眠专家与基于kappa系数值为0.80的模型预测之间存在很大的一致性。
④情感脑机接口
从2014年开始,人们对通过EEG分析大脑活动、将DL技术应用于人类生理情感识别产生了极大的兴趣(图6)。Craik等人发现,人类情感识别是基于DL的EEG数据分类研究的第二大应用领域,占已发表研究的16%,仅次于运动想象任务分类(22%)。研究人员,特别是来自计算机视觉领域的研究人员,倾向于针对EEG分类问题进行人类情绪识别。利用DL技术进行情感分类的EEG数据集包含:基于生理信号进行情绪分析的数据库(DEAP),SJTU情绪EEG数据集(SEED),以及个体和群体情感、个性和情绪研究数据集(AMIGOS)。此外,MPED和DREAMER等其他情感识别EEG数据集,进一步扩展了BCI社区在情感识别领域的可用资源。
图6.一般情感计算的示意图。
值得注意的问题
①混合vs端到端
图7.典型的EEG处理管道。
B.端到端
目前,人们对EEG的端到端DL分类越来越感兴趣。端到端学习指的是创建一个数据处理和建模管道,该管道可以优化任务的总体目标。人们对端到端DL感兴趣,是因为它有可能消除预处理的必要性,而预处理往往依赖于特定领域知识或专家的可视化解释。然而,在当前阶段,如果不彻底分析DL如何学习和依赖什么进行判断,使用端到端的DL分类仍然是有风险的。
②解释
A.权重的可视化
B.激活分析
DL EEG研究中常用的另一种解释方法是激活分析。由于其模型结构是为了捕获空间和时间元素,因此对激活函数的输出进行平均将提供关于空间和时间滤波方面的有意义的信息。例如,EEGNet文献中展示了一个用拓扑图表示的空间滤波器,我们可以根据大脑的哪个区域来解释这些激活。此外,对激活区域的平均小波时频分析可以提供丰富的时域和频域信息。类似地,Irene等人使用逐层相关性传播(LRP)来解释网络做出的分类决策。使用LRP,决策分数被转换为热图,表明每个数据点对决策的重要性。
③再现性
表1.提高脑电深度学习研究再现性的六项指南。
Vaughan, T.M., et al.: Brain-computer interface technology: a review of the Second International Meeting (2003)
Lotte, F., Congedo, M., Lécuyer, A., Lamarche, F., Arnaldi, B.: A review of classification algorithms for EEG-based brain–computer interfaces. J. Neural Eng. 4(2), R1 (2007)
Yannick, R., Hubert, B., Isabela, A., Alexandre, G., Jocelyn, F., et al.: Deep learningbased electroencephalography analysis: a systematic review. arXiv preprint arXiv:1901.05498 (2019). https://github.com/hubertjb/dleeg-review
Sejnowski, T.J.: The unreasonable effectiveness of deep learning in artificial intelligence. In: Proceedings of the National Academy of Sciences (2020)
Craik, A., He, Y., Contreras-Vidal, J.L.P.: Deep learning for Electroencephalogram (EEG) classification tasks: a review. J. Neural Eng. 16, 031001 (2019)
Morabito, F.C., et al.: Deep convolutional neural networks for classification of mild cognitive impaired and Alzheimer’s disease patients from scalp EEG recordings. In: 2016 IEEE 2nd International Forum on Research and Technologies for Society and Industry Leveraging a Better Tomorrow (RTSI), pp. 1–6. IEEE (2016)
Kim, D., Kim, K.: Detection of early stage Alzheimer’s disease using EEG relative power with deep neural network. In: 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 352–355. IEEE (2018)
Zhao, Y., He, L.: Deep learning in the EEG diagnosis of Alzheimer’s disease. In: Asian Conference on Computer Vision, pp. 340–353. Springer (2014)
Acharya, U.R., Oh, S.L., Hagiwara, Y., Tan, J.H., Adeli, H., Subha, D.P.: Automated EEGbased screening of depression using deep convolutional neural network. Comput. Methods Programs Biomed. 161, 103–113 (2018)
Baltatzis, V., Bintsi, K.-M., Apostolidis, G.K., Hadjileontiadis, L.J.: Bullying incidences identification within an immersive environment using HD EEG-based analysis: a swarm decomposition and deep learning approach. Sci. Rep. 7(1), 1–8 (2017)
Guo, Y., Friston, K., Aldo, F., Hill, S., Peng, H.: Brain Informatics and Health: 8th International Conference, BIH 2015, London, 30 Aug–2 Sept 2015. Proceedings, vol. 9250. Springer (2015)
Bashivan, P., Rish, I., Yeasin, M., Codella, N.: Learning representations from EEG with deep recurrent-convolutional neural networks. arXiv preprint arXiv:1511.06448 (2015)
Le Roux, N., Bengio, Y.: Representational power of restricted Boltzmann machines and deep belief networks. Neural Comput. 20(6), 1631–1649 (2008)
Lawrence, S., Giles, C.L., Tsoi, A.C., Back, A.D.: Face recognition: a convolutional neuralnetwork approach. IEEE Trans. Neural Netw. 8(1), 98–113 (1997)
LeCun, Y.: Deep learning & convolutional networks. In: 27th IEEE Hot Chips Symposium, HCS 2015. Institute of Electrical and Electronics Engineers Inc (2016)
Pearlmutter, B.A.: Learning state space trajectories in recurrent neural networks. Neural Comput. 1(2), 263–269 (1989)
Onose, G., et al.: On the feasibility of using motor imagery EEG-based brain–computer interface in chronic tetraplegics for assistive robotic arm control: a clinical test and long-term post-trial follow-up. Spinal Cord 50(8), 599 (2012)
Pfurtscheller, G., Neuper, C.: Motor imagery and direct brain-computer communication. Proc. IEEE 89(7), 1123–1134 (2001)
Féry, Y.-A.: Differentiating visual and kinesthetic imagery in mental practice. Can. J. Exp. Psychol./Revue Canadienne de Psychologie Expérimentale 57(1)), 1 (2003)
Tangermann, M., et al.: Review of the BCI competition IV. Front. Neurosci. 6, 55 (2012)
Abbas, W., Khan, N.A.: DeepMI: deep learning for multiclass motor imagery classification. In: 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 219–222. IEEE (2018)
Lawhern, V.J., Solon, A.J., Waytowich, N.R., Gordon, S.M., Hung, C.P., Lance, B.J.: EEGNet: a compact convolutional neural network for EEG-based brain–computer interfaces. J. Neural Eng. 15(5), 056013 (2018)
Sakhavi, S., Guan, C., Yan, S.: Learning temporal information for brain-computer interface using convolutional neural networks. IEEE Trans. Neural Netw. Learn. Syst. 99, 1–11 (2018)
Luo, T.-J., Chao, F., et al.: Exploring spatial-frequency-sequential relationships for motor imagery classification with recurrent neural network. BMC Bioinformatics 19(1), 344 (2018)
Wang, Z., Cao, L., Zhang, Z., Gong, X., Sun, Y., Wang, H.: Short time Fourier transformation and deep neural networks for motor imagery brain computer interface recognition. Concurr. Comput. Pract. Exp. 30(23), e4413 (2018)
Tefft, B.C., et al.: Prevalence of motor vehicle crashes involving drowsy drivers, United States, 2009–2013. Citeseer (2014)
Hajinoroozi, M., Mao, Z., Huang, Y.: Prediction of driver’s drowsy and alert states from EEG signals with deep learning. In: 2015 IEEE 6th International Workshop on Computational Advances in Multi-Sensor Adaptive Processing (CAMSAP), pp. 493–496. IEEE (2015)
Zeng, H., Yang, C., Dai, G., Qin, F., Zhang, J., Kong, W.: EEG classification of driver mental states by deep learning. Cogn. Neurodyn. 12(6), 597–606 (2018)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Szegedy, C., Ioffe, S., Vanhoucke, V., Alemi, A.A.: Inception-v4, inception-resnet and the impact of residual connections on learning. In: Thirty-First AAAI Conference on Artificial Intelligence (2017)
Gao, Z., et al.: EEG-based spatio-temporal convolutional neural network for driver fatigue evaluation. IEEE Trans. Neural Netw. Learn. Syst. 30, 2755–2763 (2019)
Jeong, J.-H., Yu, B.-W., Lee, D.-H., Lee, S.-W.: Classification of drowsiness levels based on a deep spatio-temporal convolutional bidirectional LSTM network using electroencephalography signals. Brain Sci. 9(12), 348 (2019)
Borghini, G., Astolfi, L., Vecchiato, G., Mattia, D., Babiloni, F.: Measuring neurophysiological signals in aircraft pilots and car drivers for the assessment of mental workload, fatigue and drowsiness. Neurosci. Biobehav. Rev. 44, 58–75 (2014)
Aghajani, H., Garbey, M., Omurtag, A.: Measuring mental workload with EEG+ fNIRS. Front. Hum. Neurosci. 11, 359 (2017)
Young, M.S., Brookhuis, K.A., Wickens, C.D., Hancock, P.A.: State of science: mental workload in ergonomics. Ergonomics 58(1), 1–17 (2015)
Jiao, Z., Gao, X., Wang, Y., Li, J., Xu, H.: Deep convolutional neural networks for mental load classification based on EEG data. Pattern Recogn. 76, 582–595 (2018)
Zhang, P., Wang, X., Zhang, W., Chen, J.: Learning spatial–spectral–temporal EEG features with recurrent 3D convolutional neural networks for cross-task mental workload assessment. IEEE Trans. Neural Syst. Rehabil. Eng. 27, 31–42 (2018)
Munafò, M.R., et al.: A manifesto for reproducible science. Nat. Hum. Behav. 1, 0021 (2017)
Pernet, C., Appelhoff, S., Flandin, G., Phillips, C., Delorme, A., Oostenveld, R.: BIDS-EEG: an extension to the Brain Imaging Data Structure (BIDS) Specification for electroencephalography (2018). https://doi.org/10.31234/osf.io/63a4y
更多精彩课程推荐
| |||
本文内容仅供参考,一切内容以英文原文为准。
非常欢迎大家留言、转载、收藏或分享~ 本文来自微信公众号“茗创科技”。如需转载,请在“茗创科技”后台回复“转载”,并附上所需转载的文章标题以及您的ID。
MC_Brain
茗创科技工作室
觉得有帮助,欢迎转发收藏或者点个在看哦~
听说点在看的人SCI接收率都提升了18%呢!