On-device Learning of EEGNet-based Network For Wearable Motor Imagery Brain-Computer Interface

Read original: arXiv:2409.00083 - Published 9/4/2024 by Sizhen Bian, Pixi Kang, Julian Moosmann, Mengxi Liu, Pietro Bonazzi, Roman Rosipal, Michele Magno
Total Score

0

On-device Learning of EEGNet-based Network For Wearable Motor Imagery Brain-Computer Interface

Sign in to get full access

or

If you already have an account, we'll log you in

Overview

  • This paper explores on-device learning of an EEGNet-based network for a wearable motor imagery brain-computer interface (MI-BCI)
  • Key points include:
    • Developing a compact neural network that can be efficiently trained and deployed on wearable devices
    • Implementing continuous learning capabilities to enable the network to adapt to individual users over time
    • Evaluating the performance and efficiency of the proposed approach on a custom dataset

Plain English Explanation

The researchers in this paper developed a new neural network architecture called EEGNet that can be used for brain-computer interfaces based on motor imagery. The key innovation is that this network can be efficiently trained and run directly on wearable devices, rather than requiring a connection to a more powerful computer.

This is important because it allows the system to continuously adapt to each individual user over time, improving performance without the need for frequent retraining on a server. The researchers tested this approach on a custom dataset and found that it could achieve good accuracy while being very efficient in terms of memory and computation requirements.

Technical Explanation

The researchers proposed an EEGNet-based neural network architecture that can be trained and deployed directly on wearable brain-computer interface (BCI) devices. EEGNet is a compact and efficient convolutional neural network design that has been shown to perform well on motor imagery EEG classification tasks.

To enable on-device learning, the researchers implemented a continuous learning approach where the network can be fine-tuned over time to adapt to each individual user's unique EEG patterns. This avoids the need for frequent retraining on a central server, which would be impractical for a wearable system.

The network was evaluated on a custom dataset of motor imagery EEG recordings collected from 20 participants. The results showed that the on-device learning approach could achieve classification accuracies comparable to a centralized model, while being much more efficient in terms of memory footprint and inference time, making it suitable for real-time wearable brain-computer interface applications.

Critical Analysis

The researchers acknowledge several limitations of their work, including the relatively small size of the custom dataset and the fact that it was collected in a controlled lab setting rather than real-world conditions. Additionally, the continuous learning approach was only evaluated over a short time period, and the long-term effects on performance and model drift are not explored.

Another potential concern is the security and privacy implications of performing on-device learning, as this could potentially expose user-specific data and model updates to potential attackers. The researchers do not address these issues in the paper.

That said, the core idea of enabling efficient, personalized brain-computer interface systems that can adapt to individual users is compelling and could have significant practical applications, particularly in areas like assistive technology and neurorehabilitation. The researchers' approach represents an important step towards realizing this vision.

Conclusion

This paper presents a novel approach for enabling on-device learning of an EEGNet-based neural network for motor imagery brain-computer interfaces. By allowing the network to continuously adapt to individual users, the researchers have developed a system that can achieve good performance while being highly efficient in terms of memory and computation, making it suitable for real-time wearable applications. While the work has some limitations, it represents an important step forward in the field of personalized, adaptive brain-computer interface technology.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

On-device Learning of EEGNet-based Network For Wearable Motor Imagery Brain-Computer Interface
Total Score

0

On-device Learning of EEGNet-based Network For Wearable Motor Imagery Brain-Computer Interface

Sizhen Bian, Pixi Kang, Julian Moosmann, Mengxi Liu, Pietro Bonazzi, Roman Rosipal, Michele Magno

Electroencephalogram (EEG)-based Brain-Computer Interfaces (BCIs) have garnered significant interest across various domains, including rehabilitation and robotics. Despite advancements in neural network-based EEG decoding, maintaining performance across diverse user populations remains challenging due to feature distribution drift. This paper presents an effective approach to address this challenge by implementing a lightweight and efficient on-device learning engine for wearable motor imagery recognition. The proposed approach, applied to the well-established EEGNet architecture, enables real-time and accurate adaptation to EEG signals from unregistered users. Leveraging the newly released low-power parallel RISC-V-based processor, GAP9 from Greeenwaves, and the Physionet EEG Motor Imagery dataset, we demonstrate a remarkable accuracy gain of up to 7.31% with respect to the baseline with a memory footprint of 15.6 KByte. Furthermore, by optimizing the input stream, we achieve enhanced real-time performance without compromising inference accuracy. Our tailored approach exhibits inference time of 14.9 ms and 0.76 mJ per single inference and 20 us and 0.83 uJ per single update during online training. These findings highlight the feasibility of our method for edge EEG devices as well as other battery-powered wearable AI systems suffering from subject-dependant feature distribution drift.

Read more

9/4/2024

EEGMobile: Enhancing Speed and Accuracy in EEG-Based Gaze Prediction with Advanced Mobile Architectures
Total Score

0

EEGMobile: Enhancing Speed and Accuracy in EEG-Based Gaze Prediction with Advanced Mobile Architectures

Teng Liang, Andrews Damoah

Electroencephalography (EEG) analysis is an important domain in the realm of Brain-Computer Interface (BCI) research. To ensure BCI devices are capable of providing practical applications in the real world, brain signal processing techniques must be fast, accurate, and resource-conscious to deliver low-latency neural analytics. This study presents a model that leverages a pre-trained MobileViT alongside Knowledge Distillation (KD) for EEG regression tasks. Our results showcase that this model is capable of performing at a level comparable (only 3% lower) to the previous State-Of-The-Art (SOTA) on the EEGEyeNet Absolute Position Task while being 33% faster and 60% smaller. Our research presents a cost-effective model applicable to resource-constrained devices and contributes to expanding future research on lightweight, mobile-friendly models for EEG regression.

Read more

8/9/2024

🏷️

Total Score

0

EEGEncoder: Advancing BCI with Transformer-Based Motor Imagery Classification

Wangdan Liao, Weidong Wang

Brain-computer interfaces (BCIs) harness electroencephalographic signals for direct neural control of devices, offering a significant benefit for individuals with motor impairments. Traditional machine learning methods for EEG-based motor imagery (MI) classification encounter challenges such as manual feature extraction and susceptibility to noise.This paper introduces EEGEncoder, a deep learning framework that employs modified transformers and TCNs to surmount these limitations. We innovatively propose a fusion architecture, namely Dual-Stream Temporal-Spatial Block (DSTS), to capture temporal and spatial features, improving the accuracy of Motor Imagery classification task. Additionally, we use multiple parallel structures to enhance the performance of the model. When tested on the BCI Competition IV-2a dataset, our model results outperform current state-of-the-art techniques.

Read more

6/26/2024

EDPNet: An Efficient Dual Prototype Network for Motor Imagery EEG Decoding
Total Score

0

EDPNet: An Efficient Dual Prototype Network for Motor Imagery EEG Decoding

Can Han, Chen Liu, Crystal Cai, Jun Wang, Dahong Qian

Motor imagery electroencephalograph (MI-EEG) decoding plays a crucial role in developing motor imagery brain-computer interfaces (MI-BCIs). However, decoding intentions from MI remains challenging due to the inherent complexity of EEG signals relative to the small-sample size. In this paper, we propose an Efficient Dual Prototype Network (EDPNet) to enable accurate and fast MI decoding. EDPNet employs a lightweight adaptive spatial-spectral fusion module, which promotes more efficient information fusion between multiple EEG electrodes. Subsequently, a parameter-free multi-scale variance pooling module extracts more comprehensive temporal features. Furthermore, we introduce dual prototypical learning to optimize the feature space distribution and training process, thereby improving the model's generalization ability on small-sample MI datasets. Our experimental results show that the EDPNet outperforms state-of-the-art models with superior classification accuracy and kappa values (84.11% and 0.7881 for dataset BCI competition IV 2a, 86.65% and 0.7330 for dataset BCI competition IV 2b). Additionally, we use the BCI competition III IVa dataset with fewer training data to further validate the generalization ability of the proposed EDPNet. We also achieve superior performance with 82.03% classification accuracy. Benefiting from the lightweight parameters and superior decoding accuracy, our EDPNet shows great potential for MI-BCI applications. The code is publicly available at https://github.com/hancan16/EDPNet.

Read more

7/4/2024