Akademska digitalna zbirka SLovenije - logo
E-viri
Recenzirano Odprti dostop
  • An end-to-end deep learning...
    Dose, Hauke; Møller, Jakob S.; Iversen, Helle K.; Puthusserypady, Sadasivan

    Expert systems with applications, 12/2018, Letnik: 114
    Journal Article

    •End-to-end neural network model for classifying motor imagery EEG signals.•Using 1-D CNN layers to learn temporal and spatial filters for feature extraction.•Application of transfer learning to calibrate the model for individual subjects.•Analysis of the temporal and spatial filters learned by the model. Goal: To develop and implement a Deep Learning (DL) approach for an electroencephalogram (EEG) based Motor Imagery (MI) Brain-Computer Interface (BCI) system that could potentially be used to improve the current stroke rehabilitation strategies. Method: The DL model is using Convolutional Neural Network (CNN) layers for learning generalized features and dimension reduction, while a conventional Fully Connected (FC) layer is used for classification. Together they build a unified end-to-end model that can be applied to raw EEG signals. This previously proposed model was applied to a new set of data to validate its robustness against data variations. Furthermore, it was extended by subject-specific adaptation. Lastly, an analysis of the learned filters provides insights into how such a model derives a classification decision. Results: The selected global classifier reached 80.38%, 69.82%, and 58.58% mean accuracies for datasets with two, three, and four classes, respectively, validated using 5-fold crossvalidation. As a novel approach in this context, transfer learning was used to adapt the global classifier to single individuals improving the overall mean accuracy to 86.49%, 79.25%, and 68.51%, respectively. The global models were trained on 3s segments of EEG data from different subjects than they were tested on, which proved the generalization performance of the model. Conclusion: The results are comparable with the reported accuracy values in related studies and the presented model outperforms the results in the literature on the same underlying data. Given that the model can learn features from data without having to use specialized feature extraction methods, DL should be considered as an alternative to established EEG classification methods, if enough data is available.