Relevance-based channel selection in motor imagery brain-computer interface

被引:13
作者
Nagarajan, Aarthy [1 ]
Robinson, Neethu [1 ]
Guan, Cuntai [1 ]
机构
[1] Nanyang Technol Univ, Sch Comp Sci & Engn, 50 Nanyang Ave, Singapore 639798, Singapore
关键词
MI-BCI; channel selection; deep learning; explainable AI; layer-wise relevance propagation; NEURAL-NETWORKS; EEG; CLASSIFICATION; DECISIONS; GRADIENT;
D O I
10.1088/1741-2552/acae07
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Objective. Channel selection in the electroencephalogram (EEG)-based brain-computer interface (BCI) has been extensively studied for over two decades, with the goal being to select optimal subject-specific channels that can enhance the overall decoding efficacy of the BCI. With the emergence of deep learning (DL)-based BCI models, there arises a need for fresh perspectives and novel techniques to conduct channel selection. In this regard, subject-independent channel selection is relevant, since DL models trained using cross-subject data offer superior performance, and the impact of inherent inter-subject variability of EEG characteristics on subject-independent DL training is not yet fully understood. Approach. Here, we propose a novel methodology for implementing subject-independent channel selection in DL-based motor imagery (MI)-BCI, using layer-wise relevance propagation (LRP) and neural network pruning. Experiments were conducted using Deep ConvNet and 62-channel MI data from the Korea University EEG dataset. Main Results. Using our proposed methodology, we achieved a 61% reduction in the number of channels without any significant drop (p = 0.09) in subject-independent classification accuracy, due to the selection of highly relevant channels by LRP. LRP relevance-based channel selections provide significantly better accuracies compared to conventional weight-based selections while using less than 40% of the total number of channels, with differences in accuracies ranging from 5.96% to 1.72%. The performance of the adapted sparse-LRP model using only 16% of the total number of channels is similar to that of the adapted baseline model (p = 0.13). Furthermore, the accuracy of the adapted sparse-LRP model using only 35% of the total number of channels exceeded that of the adapted baseline model by 0.53% (p = 0.81). Analyses of channels chosen by LRP confirm the neurophysiological plausibility of selection, and emphasize the influence of motor, parietal, and occipital channels in MI-EEG classification. Significance. The proposed method addresses a traditional issue in EEG-BCI decoding, while being relevant and applicable to the latest developments in the field of BCI. We believe that our work brings forth an interesting and important application of model interpretability as a problem-solving technique.
引用
收藏
页数:21
相关论文
共 53 条
[1]  
Ang KK, 2010, IEEE ENG MED BIO, P5549, DOI 10.1109/IEMBS.2010.5626782
[2]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[3]  
Apicella A, 2024, Arxiv, DOI arXiv:2210.06554
[4]   Channel Selection for Optimal EEG Measurement in Motor Imagery-Based Brain-Computer Interfaces [J].
Arpaia, Pasquale ;
Donnarumma, Francesco ;
Esposito, Antonio ;
Parvis, Marco .
INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2021, 31 (03)
[5]  
Arvaneh M, 2012, IEEE IJCNN
[6]   Optimizing the Channel Selection and Classification Accuracy in EEG-Based BCI [J].
Arvaneh, Mahnaz ;
Guan, Cuntai ;
Ang, Kai Keng ;
Quek, Chai .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2011, 58 (06) :1865-1873
[7]   On Pixel-Wise Explanations for Non-Linear Classifier Decisions by Layer-Wise Relevance Propagation [J].
Bach, Sebastian ;
Binder, Alexander ;
Montavon, Gregoire ;
Klauschen, Frederick ;
Mueller, Klaus-Robert ;
Samek, Wojciech .
PLOS ONE, 2015, 10 (07)
[8]   Explainable Artificial Intelligence (XAI): Concepts, taxonomies, opportunities and challenges toward responsible AI [J].
Barredo Arrieta, Alejandro ;
Diaz-Rodriguez, Natalia ;
Del Ser, Javier ;
Bennetot, Adrien ;
Tabik, Siham ;
Barbado, Alberto ;
Garcia, Salvador ;
Gil-Lopez, Sergio ;
Molina, Daniel ;
Benjamins, Richard ;
Chatila, Raja ;
Herrera, Francisco .
INFORMATION FUSION, 2020, 58 :82-115
[9]  
Blalock D, 2020, Arxiv, DOI arXiv:2003.03033
[10]   Optimizing spatial filters for robust EEG single-trial analysis [J].
Blankertz, Benjamin ;
Tomioka, Ryota ;
Lemm, Steven ;
Kawanabe, Motoaki ;
Mueller, Klaus-Robert .
IEEE SIGNAL PROCESSING MAGAZINE, 2008, 25 (01) :41-56