Cross-domain MLP and CNN Transfer Learning for Biological Signal Processing: EEG and EMG

Jordan J. Bird, Jhonatan Kobylarz, Diego R. Faria, Aniko Ekart, Eduardo P. Ribeiro

Research output: Contribution to journalArticle

Abstract

In this work, we show the success of unsupervised transfer learning between Electroencephalographic (brainwave) classification and Electromyographic (muscular wave) domains with both MLP
and CNN methods. To achieve this, signals are measured from both the brain and forearm muscles and EMG
data is gathered from a 4-class gesture classification experiment via the Myo Armband, and a 3-class mental
state EEG dataset is acquired via the Muse EEG Headband. A hyperheuristic multi-objective evolutionary
search method is used to find the best network hyperparameters. We then use this optimised topology
of deep neural network to classify both EMG and EEG signals, attaining results of 84.76% and 62.37%
accuracy, respectively. Next, when pre-trained weights from the EMG classification model are used for
initial distribution rather than random weight initialisation for EEG classification, 93.82%(+29.95) accuracy
is reached. When EEG pre-trained weights are used for initial weight distribution for EMG, 85.12% (+0.36)
accuracy is achieved. When the EMG network attempts to classify EEG, it outperforms the EEG network
even without any training (+30.25% to 82.39% at epoch 0), and similarly the EEG network attempting
to classify EMG data outperforms the EMG network (+2.38% at epoch 0). All transfer networks achieve
higher pre-training abilities, curves, and asymptotes, indicating that knowledge transfer is possible between
the two signal domains. In a second experiment with CNN transfer learning, the same datasets are projected
as 2D images and the same learning process is carried out. In the CNN experiment, EMG to EEG transfer
learning is found to be successful but not vice-versa, although EEG to EMG transfer learning did exhibit
a higher starting classification accuracy. The significance of this work is due to the successful transfer of
ability between models trained on two different biological signal domains, reducing the need for building
more computationally complex models in future research.
Original languageEnglish
Article number9027853
Pages (from-to)54789-54801
Number of pages13
JournalIEEE Access
Volume8
Early online date9 Mar 2020
DOIs
Publication statusE-pub ahead of print - 9 Mar 2020

Bibliographical note

This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see https://creativecommons.org/licenses/by/4.0/.

Keywords

  • Applied machine learning
  • EEG
  • EMG
  • biological signal processing
  • knowledge adaptation
  • neural networks
  • transfer learning

Fingerprint Dive into the research topics of 'Cross-domain MLP and CNN Transfer Learning for Biological Signal Processing: EEG and EMG'. Together they form a unique fingerprint.

  • Cite this