A protocol for trustworthy EEG decoding with neural networks

被引:0
|
作者
Borra, Davide [1 ]
Magosso, trustworthy Elisa
Ravanelli, Mirco [2 ,3 ]
机构
[1] Univ Bologna, Dept Elect Elect & Informat Engn Guglielmo Marconi, Cesena, Forli, Italy
[2] Concordia Univ, Dept Comp Sci & Software Engn, Montreal, PQ, Canada
[3] Mila Quebec Inst, Montreal, PQ, Canada
关键词
Electroencephalography; Single-trial EEG decoding; Deep learning; Convolutional neural networks; Hyperparameter search; Brain-Computer Interfaces; BRAIN-COMPUTER INTERFACES; MOTOR IMAGERY; COMMUNICATION; CLASSIFICATION; PROSTHESIS; ARM;
D O I
10.1016/j.neunet.2024.106847
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning solutions have rapidly emerged for EEG decoding, achieving state-of-the-art performance on a variety of decoding tasks. Despite their high performance, existing solutions do not fully address the challenge posed by the introduction of many hyperparameters, defining data pre-processing, network architecture, network training, and data augmentation. Automatic hyperparameter search is rarely performed and limited to network-related hyperparameters. Moreover, pipelines are highly sensitive to performance fluctuations due to random initialization, hindering their reliability. Here, we design a comprehensive protocol for EEG decoding that explores the hyperparameters characterizing the entire pipeline and that includes multi-seed initialization for providing robust performance estimates. Our protocol is validated on 9 datasets about motor imagery, P300, SSVEP, including 204 participants and 26 recording sessions, and on different deep learning models. We accompany our protocol with extensive experiments on the main aspects influencing it, such as the number of participants used for hyperparameter search, the split into sequential simpler searches (multi-step search), the use of informed vs. non-informed search algorithms, and the number of random seeds for obtaining stable performance. The best protocol included 2-step hyperparameter search via an informed search algorithm, with the final training and evaluation performed using 10 random initializations. The optimal trade-off between performance and computational time was achieved by using a subset of 3-5 participants for hyperparameter search. Our protocol consistently outperformed baseline state-of-the-art pipelines, widely across datasets and models, and could represent a standard approach for neuroscientists for decoding EEG in a trustworthy and reliable way.
引用
收藏
页数:22
相关论文
共 50 条
  • [1] MNEflow: Neural networks for EEG/MEG decoding and interpretation
    Zubarev, Ivan
    Vranou, Gavriela
    Parkkonen, Lauri
    SOFTWAREX, 2022, 17
  • [2] Deep Learning With Convolutional Neural Networks for EEG Decoding and Visualization
    Schirrmeister, Robin Tibor
    Springenberg, Jost Tobias
    Fiederer, Lukas Dominique Josef
    Glasstetter, Martin
    Eggensperger, Katharina
    Tangermann, Michael
    Hutter, Frank
    Burgard, Wolfram
    Ball, Tonio
    HUMAN BRAIN MAPPING, 2017, 38 (11) : 5391 - 5420
  • [3] Decoding EEG by Visual-guided Deep Neural Networks
    Jiao, Zhicheng
    You, Haoxuan
    Yang, Fan
    Li, Xin
    Zhang, Han
    Shen, Dinggang
    PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2019, : 1387 - 1393
  • [4] DEEP LEARNING WITH CONVOLUTIONAL NEURAL NETWORKS FOR DECODING AND VISUALIZATION OF EEG PATHOLOGY
    Schirrmeister, R.
    Gemein, L.
    Eggensperger, K.
    Hutter, F.
    Ball, T.
    2017 IEEE SIGNAL PROCESSING IN MEDICINE AND BIOLOGY SYMPOSIUM (SPMB), 2017,
  • [5] An empirical comparison of neural networks and machine learning algorithms for EEG gait decoding
    Nakagome, Sho
    Trieu Phat Luu
    He, Yongtian
    Ravindran, Akshay Sujatha
    Contreras-Vidal, Jose L.
    SCIENTIFIC REPORTS, 2020, 10 (01)
  • [6] An empirical comparison of neural networks and machine learning algorithms for EEG gait decoding
    Sho Nakagome
    Trieu Phat Luu
    Yongtian He
    Akshay Sujatha Ravindran
    Jose L. Contreras-Vidal
    Scientific Reports, 10
  • [7] Neural decoding of music from the EEG
    Daly, Ian
    SCIENTIFIC REPORTS, 2023, 13 (01):
  • [8] Neural decoding of music from the EEG
    Ian Daly
    Scientific Reports, 13 (1)
  • [9] Invertible Neural Networks for Trustworthy AI
    Schwab, Malgorzata
    Biswas, Ashis Kumer
    2023 IEEE 35TH INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE, ICTAI, 2023, : 480 - 485
  • [10] EEG Motor Execution Decoding via Interpretable Sinc-Convolutional Neural Networks
    Borra, Davide
    Fantozzi, Silvia
    Magosso, Elisa
    XV MEDITERRANEAN CONFERENCE ON MEDICAL AND BIOLOGICAL ENGINEERING AND COMPUTING - MEDICON 2019, 2020, 76 : 1113 - 1122