arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.01074eess.SPcs.AIcs.LGq-bio.NCq-bio.QM

Comparison of Attention-based Deep Learning Models for EEG Classification

University of Padova(帕多瓦大学);National, Inter-University Consortium for Telecommunications (CNIT)(全国大学校际电信联盟(CNIT));National Center of Neurology and Psychiatry(国立精神神经医疗研究中心);University of Milano-Bicocca(米兰比可卡大学);Wroclaw University of Science and Technology(弗罗茨瓦夫理工大学)

Giulia Cisotto, Alessio Zanga, Joanna Chlebus, Italo Zoppis, Sara Manzoni, Urszula Markowska-Kaczmar

英文摘要:

Objective: To evaluate the impact on Electroencephalography (EEG) classification of different kinds of attention mechanisms in Deep Learning (DL) models. Methods: We compared three attention-enhanced DL models, the brand-new InstaGATs, an LSTM with attention and a CNN with attention. We used these models to classify normal and abnormal (i.e., artifactual or pathological) EEG patterns. Results: We achieved the state of the art in all classification problems, regardless the large variability of the datasets and the simple architecture of the attention-enhanced models. We could also prove that, depending on how the attention mechanism is applied and where the attention layer is located in the model, we can alternatively leverage the information contained in the time, frequency or space domain of the dataset. Conclusions: with this work, we shed light over the role of different attention mechanisms in the classification of normal and abnormal EEG patterns. Moreover, we discussed how they can exploit the intrinsic relationships in the temporal, frequency and spatial domains of our brain activity. Significance: Attention represents a promising strategy to evaluate the quality of the EEG information, and its relevance, in different real-world scenarios. Moreover, it can make it easier to parallelize the computation and, thus, to speed up the analysis of big electrophysiological (e.g., EEG) datasets.

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

↑