Explainable Machine Learning Methods for Classification of Brain States during Visual Perception

Robiul Islam, Andrey V. Andreev, Natalia N. Shusharina and Alexander E. Hramov ()
Additional contact information
Robiul Islam: Laboratory of Neuroscience and Cognitive Technology, Innopolis University, 1 Universitetskaya Str., 420500 Innopolis, Russia
Andrey V. Andreev: Laboratory of Neuroscience and Cognitive Technology, Innopolis University, 1 Universitetskaya Str., 420500 Innopolis, Russia
Natalia N. Shusharina: Baltic Center of Neurotechnology and Artificial Intelligence, Immanuil Kant Baltic Federal University, 14 A. Nevskogo ul., 236041 Kaliningrad, Russia
Alexander E. Hramov: Baltic Center of Neurotechnology and Artificial Intelligence, Immanuil Kant Baltic Federal University, 14 A. Nevskogo ul., 236041 Kaliningrad, Russia

Mathematics, 2022, vol. 10, issue 15, 1-25

Abstract: The aim of this work is to find a good mathematical model for the classification of brain states during visual perception with a focus on the interpretability of the results. To achieve it, we use the deep learning models with different activation functions and optimization methods for their comparison and find the best model for the considered dataset of 31 EEG channels trials. To estimate the influence of different features on the classification process and make the method more interpretable, we use the SHAP library technique. We find that the best optimization method is Adagrad and the worst one is FTRL. In addition, we find that only Adagrad works well for both linear and tangent models. The results could be useful for EEG-based brain–computer interfaces (BCIs) in part for choosing the appropriate machine learning methods and features for the correct training of the BCI intelligent system.

Keywords: deep learning; brain state; EEG; interpretable models; visual perception (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/10/15/2819/pdf (application/pdf)
https://www.mdpi.com/2227-7390/10/15/2819/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:10:y:2022:i:15:p:2819-:d:883322

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().