| Literature DB >> 36004856 |
Abeer Abdulaziz AlArfaj1, Hanan A Hosni Mahmoud1, Alaaeldin M Hafez2.
Abstract
Detection of limb motor functions utilizing brain signals is a significant technique in the brain signal gain model (BSM) that can be effectively employed in various biomedical applications. Our research presents a novel technique for prediction of feet motor functions by applying a deep learning model with cascading transfer learning technique to use the electroencephalogram (EEG) in the training stage. Our research deduces the electroencephalogram data (EEG) of stroke incidence to propose functioning high-tech interfaces for predicting left and right foot motor functions. This paper presents a transfer learning with several source input domains to serve a target domain with small input size. Transfer learning can reduce the learning curve effectively. The correctness of the presented model is evaluated by the abilities of motor functions in the detection of left and right feet. Extensive experiments were performed and proved that a higher accuracy was reached by the introduced BSM-EEG neural network with transfer learning. The prediction of the model accomplished 97.5% with less CPU time. These accurate results confirm that the BSM-EEG neural model has the ability to predict motor functions for brain-injured stroke therapy.Entities:
Keywords: machine learning; motor function therapy; transfer learning
Year: 2022 PMID: 36004856 PMCID: PMC9404826 DOI: 10.3390/bs12080285
Source DB: PubMed Journal: Behav Sci (Basel) ISSN: 2076-328X
Figure 1EEG signal recording versus time in seconds.
Figure 2The motor functions of the foot (a–c).
The statistics of the motor function of left and right feet data.
| Motor Function | Mean | Standard Deviation | Minimum | Maximum |
|---|---|---|---|---|
| Right foot flexion | 18.9° | 3.4° | 0 | 30° |
| Left foot flexion | 20.5° | 2.68° | 0 | 30° |
| Right foot extension | 40.7° | 5.67° | 0 | 50° |
| Left foot extension | 42.7° | 6.3° | 0 | 50° |
| Right foot pronation | 25.96 | 2.87 | 0 | 30° |
| Left foot pronation | 26.71 | 3.63 | 0 | 30° |
| Right foot supination | 51.71 | 5.73 | 0 | 60° |
| Left foot supination | 48.96 | 4.87 | 0 | 60° |
Dataset statistics (total samples of EEG signals: 2000 from 271 cases).
| Foot Movement Associated with the EEG | Count |
|---|---|
| Right foot flexion | 222 |
| Left foot flexion | 200 |
| Right foot extension | 208 |
| Left foot extension | 300 |
| Right foot pronation | 250 |
| Left foot pronation | 200 |
| Right foot supination | 300 |
| Left foot supination | 320 |
Figure 3Methodology to obtain deep learning models for transfer learning (the flow diagram of the BSM-EEG model).
Figure 4(a) The architecture of the transfer learning training; (b) the architecture of prediction from actual labeled clinical data.
Environment.
| Hardware | |
|---|---|
| Processor | RAM |
| Sun station CPU X6-3320 V2@ 3.60 GHz* 16 | 64 GB |
|
| |
| Operating system | Simulation environment |
| Linux | Python 3.4 and Mat lab |
Hyperparameters utilized for training.
| Stage | Layer | Hyperparameter Value |
|---|---|---|
| First Convolution | Filters | 128 |
| Kernel size | 5 | |
| Strides | 3 | |
| Average pooling | 8 | |
| Second Convolution | Filters | 256 |
| Kernel size | 4 | |
| Average pooling | 4 | |
| Third Convolution | Filters | 512 |
| Kernel size | 2 | |
| Max pooling | 2 | |
| Training Parameters | Learning rate | 0.2 |
| Epochs | 80 | |
| Batch size | 26 | |
| Optimizer | Adam |
Prediction accuracy of various counts of neurons in convolutional layers.
| Neuron Counts | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 |
|---|---|---|---|---|---|---|---|---|
| Layer 1 | 0.9256 | 0.9359 | 0.9363 | 0.9282 | 0.9461 | 0.9709 | 0.9726 | 0.9655 |
| Layer 2 | 0.9665 | 0.9704 | 0.9729 | 0.9389 | 0.9509 | 0.9449 | 0.9366 | 0.9336 |
| Layer 3 | 0.9449 | 0.9727 | 0.9652 | 0.9466 | 0.9529 | 0.9506 | 0.9437 | 0.9363 |
| Layer 4 | 0.9406 | 0.9383 | 0.9372 | 0.9449 | 0.9277 | 0.9364 | 0.9333 | 0.9309 |
| Layer 6 | 0.9304 | 0.9429 | 0.9361 | 0.9309 | 0.9309 | 0.9329 | 0.9309 | 0.9271 |
Figure 5Prediction accuracy of various counts of neurons in convolutional layers.
The impact of learning rate on performance.
| Learning Rate | 0.05 | 0.07 | 0.09 | 0.11 | 0.13 | 0.15 |
|---|---|---|---|---|---|---|
| Accuracy | 0.954 | 0.972 | 0.958 | 0.931 | 0.932 | 0.930 |
Figure 6The impact of learning rate on performance.
Classification report of our model with transfer learning with one and two source domain model.
| Our Model with Transfer Learning with One Source Domain | Our Model with Transfer Learning with Two Source Domain | |||||
|---|---|---|---|---|---|---|
| Predicted | Precision | Recall | F2-score | Precision | Recall | F2-score |
| Right foot | 0.9 | 0.95 | 0.9 | 0.97 | 0.99 | 0.96 |
| Left foot flexion | 0.8 | 0.85 | 0.8 | 0.96 | 0.96 | 0.96 |
| Right foot | 0.94 | 0.85 | 0.91 | 0.92 | 0.96 | 0.97 |
| Left foot | 0.94 | 0.85 | 0.9 | 0.97 | 0.92 | 0.96 |
| Right foot | 0.89 | 0.93 | 0.91 | 0.96 | 0.94 | 0.97 |
| Left foot | 0.9 | 0.9 | 0.91 | 0.96 | 0.9 | 0.96 |
| Right foot | 0.84 | 0.9 | 0.8 | 0.94 | 0.9 | 0.96 |
| Left foot | 0.94 | 0.9 | 0.9 | 0.97 | 0.9 | 0.99 |
Confusion matrix for the proposed DL model without transfer learning.
| Motor Function | Right Foot Flexion | Left Foot Flexion | Right Foot Extension | Left Foot Extension | Right Foot Pronation | Left Foot Pronation | Right Foot Supination | Left Foot Supination | Total Cases |
|---|---|---|---|---|---|---|---|---|---|
| Right foot flexion | 94 | 2 | 50 | 3 | 52 | 1 | 20 | 0 | 222 |
| Left foot flexion | 3 | 100 | 4 | 33 | 2 | 22 | 1 | 35 | 200 |
| Right foot extension | 20 | 5 | 107 | 5 | 30 | 2 | 18 | 21 | 208 |
| Left foot extension | 10 | 40 | 0 | 150 | 10 | 40 | 11 | 39 | 300 |
| Right foot pronation | 22 | 8 | 30 | 10 | 130 | 10 | 30 | 10 | 250 |
| Left foot pronation | 6 | 19 | 11 | 31 | 4 | 110 | 9 | 30 | 200 |
| Right foot supination | 21 | 0 | 29 | 10 | 60 | 5 | 170 | 5 | 300 |
| Left foot supination | 4 | 51 | 0 | 49 | 11 | 30 | 5 | 170 | 320 |
Confusion matrix for the proposed DL model with transfer learning with one source domain.
| Motor Function | Right Foot Flexion | Left Foot Flexion | Right Foot Extension | Left Foot Extension | Right Foot Pronation | Left Foot Pronation | Right Foot Supination | Left Foot Supination | Total Cases |
|---|---|---|---|---|---|---|---|---|---|
| Right foot flexion | 184 | 2 | 10 | 3 | 12 | 1 | 10 | 0 | 222 |
| Left foot flexion | 1 | 170 | 2 | 10 | 2 | 9 | 1 | 5 | 200 |
| Right foot extension | 8 | 2 | 180 | 7 | 1 | 2 | 8 | 0 | 208 |
| Left foot extension | 2 | 8 | 0 | 270 | 1 | 9 | 3 | 7 | 300 |
| Right foot pronation | 10 | 1 | 7 | 10 | 220 | 1 | 9 | 2 | 250 |
| Left foot pronation | 1 | 7 | 2 | 5 | 1 | 175 | 2 | 7 | 200 |
| Right foot supination | 8 | 0 | 9 | 2 | 11 | 3 | 265 | 2 | 300 |
| Left foot supination | 3 | 10 | 1 | 11 | 3 | 9 | 4 | 280 | 320 |
Confusion matrix for the proposed DL model with transfer learning with two source domains.
| Motor Function | Right Foot Flexion | Left Foot Flexion | Right Foot Extension | Left Foot Extension | Right Foot Pronation | Left Foot Pronation | Right Foot Supination | Left Foot Supination | Total Cases |
|---|---|---|---|---|---|---|---|---|---|
| Right foot flexion | 211 | 0 | 4 | 0 | 5 | 0 | 2 | 0 | 222 |
| Left foot flexion | 0 | 195 | 0 | 1 | 1 | 2 | 0 | 1 | 200 |
| Right foot extension | 2 | 0 | 200 | 0 | 3 | 1 | 2 | 0 | 208 |
| Left foot extension | 0 | 1 | 0 | 295 | 0 | 2 | 0 | 2 | 300 |
| Right foot pronation | 1 | 0 | 2 | 0 | 244 | 1 | 2 | 0 | 250 |
| Left foot pronation | 0 | 1 | 0 | 2 | 0 | 196 | 0 | 1 | 200 |
| Right foot supination | 2 | 0 | 1 | 1 | 2 | 0 | 292 | 2 | 300 |
| Left foot supination | 0 | 1 | 1 | 2 | 0 | 1 | 0 | 315 | 320 |
Time complexity of the proposed model with and without transfer learning.
| Our Model with Transfer | Our Model with Transfer | |
|---|---|---|
| Training CPU time (h) | 12:32 | 18:57 |
| Classification time (s) | 119.9 s | 90.3 s |
Performance comparison of the proposed model with and without transfer learning.
| Model | Average Accuracy for All Motor Functions (%) | Average Training Time (h) | Average Classification Time |
|---|---|---|---|
| Our model without transfer learning | 57.10% | 8.1 | 113.1 |
| Our model with | 90.90% | 12.9 | 119.9 |
| Our model with transfer learning with two source domain | 97.30% | 17.3 | 90.3 |
Performance comparison.
| Model | BP Neural | TransferN | DLN | STL | Our Model without | Our Model with | Our Model with |
|---|---|---|---|---|---|---|---|
| Acc | 0.6136 | 0.6443 | 0.6666 | 0.6611 | 0.5668 | 0.91 | 0.97 |
| Time(s) | 64 | 106 | 113 | 132 | 120 | 119 | 90.3 |