CNN-based framework using spatial dropping for enhanced interpretation of neural activity in motor imagery classification.
Diego Fabian Collazos-HuertasA M Álvarez-MezaC D Acosta-MedinaG A Castaño-DuqueG Castellanos-DominguezPublished in: Brain informatics (2020)
Interpretation of brain activity responses using motor imagery (MI) paradigms is vital for medical diagnosis and monitoring. Assessed by machine learning techniques, identification of imagined actions is hindered by substantial intra- and inter-subject variability. Here, we develop an architecture of Convolutional Neural Networks (CNN) with an enhanced interpretation of the spatial brain neural patterns that mainly contribute to the classification of MI tasks. Two methods of 2D-feature extraction from EEG data are contrasted: Power Spectral Density and Continuous Wavelet Transform. For preserving the spatial interpretation of extracting EEG patterns, we project the multi-channel data using a topographic interpolation. Besides, we include a spatial dropping algorithm to remove the learned weights that reflect the localities not engaged with the elicited brain response. We evaluate two labeled scenarios of MI tasks: bi-class and three-class. Obtained results in an MI database show that the thresholding strategy combined with Continuous Wavelet Transform improves the accuracy and enhances the interpretability of CNN architecture, showing that the highest contribution clusters over the sensorimotor cortex with a differentiated behavior of rhythms [Formula: see text] and [Formula: see text].
Keyphrases
- convolutional neural network
- deep learning
- machine learning
- resting state
- functional connectivity
- artificial intelligence
- working memory
- big data
- electronic health record
- healthcare
- human milk
- smoking cessation
- quality improvement
- magnetic resonance imaging
- preterm infants
- brain injury
- optical coherence tomography
- cerebral ischemia
- emergency department
- adverse drug
- preterm birth
- neural network
- dual energy