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 条
  • [31] GCL: Graph Calibration Loss for Trustworthy Graph Neural Networks
    Wang, Min
    Yang, Hao
    Cheng, Qing
    PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, : 988 - 996
  • [32] EEG Decoding Reveals Neural Predictions for Naturalistic Material Behaviors
    Kaiser, Daniel
    Stecher, Rico
    Doerschner, Katja
    JOURNAL OF NEUROSCIENCE, 2023, 43 (29): : 5406 - 5413
  • [33] EEG-based Neural Decoding of Gait in Developing Children
    Luu, Trieu Phat
    Eguren, David
    Cestari, Manuel
    Contreras-Vidal, Jose L.
    2019 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN AND CYBERNETICS (SMC), 2019, : 3608 - 3612
  • [34] EEG TOPOGRAPHY RECOGNITION BY NEURAL NETWORKS
    HIRAIWA, A
    SHIMOHARA, K
    TOKUNAGA, Y
    IEEE ENGINEERING IN MEDICINE AND BIOLOGY MAGAZINE, 1990, 9 (03): : 39 - 42
  • [35] Automated EEG classification by neural networks
    Sgro, JA
    Emerson, RG
    Stanton, P
    CISST'98: PROCEEDINGS OF THE INTERNATIONAL CONFERENCE ON IMAGING SCIENCE, SYSTEMS AND TECHNOLOGY, 1998, : 295 - 301
  • [36] The removal of EMG in EEG by neural networks
    Chen, Yongjian
    Akutagawa, Masatake
    Emoto, Takahiro
    Kinouchi, Yohsuke
    PHYSIOLOGICAL MEASUREMENT, 2010, 31 (12) : 1567 - 1584
  • [37] EEG topography recognition by neural networks
    Hiraiwa, Alkira, 1600, (09):
  • [38] Explaining Neural Networks by Decoding Layer Activations
    Schneider, Johannes
    Vlachos, Michalis
    ADVANCES IN INTELLIGENT DATA ANALYSIS XIX, IDA 2021, 2021, 12695 : 63 - 75
  • [39] Decoding with Value Networks for Neural Machine Translation
    He, Di
    Lu, Hanqing
    Xia, Yingce
    Qin, Tao
    Wang, Liwei
    Liu, Tie-Yan
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 30 (NIPS 2017), 2017, 30
  • [40] Paraconsistent artificial neural networks and EEG
    Abe, Jair Minoro
    Lopes, Helder F. S.
    Nakamatsu, Kazumi
    INTERNATIONAL JOURNAL OF KNOWLEDGE-BASED AND INTELLIGENT ENGINEERING SYSTEMS, 2013, 17 (02) : 99 - 111