iBet uBet web content aggregator. Adding the entire web to your favor.
iBet uBet web content aggregator. Adding the entire web to your favor.



Link to original content: http://pubmed.ncbi.nlm.nih.gov/39338733/
A Spatio-Temporal Capsule Neural Network with Self-Correlation Routing for EEG Decoding of Semantic Concepts of Imagination and Perception Tasks - PubMed Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2024 Sep 15;24(18):5988.
doi: 10.3390/s24185988.

A Spatio-Temporal Capsule Neural Network with Self-Correlation Routing for EEG Decoding of Semantic Concepts of Imagination and Perception Tasks

Affiliations

A Spatio-Temporal Capsule Neural Network with Self-Correlation Routing for EEG Decoding of Semantic Concepts of Imagination and Perception Tasks

Jianxi Huang et al. Sensors (Basel). .

Abstract

Decoding semantic concepts for imagination and perception tasks (SCIP) is important for rehabilitation medicine as well as cognitive neuroscience. Electroencephalogram (EEG) is commonly used in the relevant fields, because it is a low-cost noninvasive technique with high temporal resolution. However, as EEG signals contain a high noise level resulting in a low signal-to-noise ratio, it makes decoding EEG-based semantic concepts for imagination and perception tasks (SCIP-EEG) challenging. Currently, neural network algorithms such as CNN, RNN, and LSTM have almost reached their limits in EEG signal decoding due to their own short-comings. The emergence of transformer methods has improved the classification performance of neural networks for EEG signals. However, the transformer model has a large parameter set and high complexity, which is not conducive to the application of BCI. EEG signals have high spatial correlation. The relationship between signals from different electrodes is more complex. Capsule neural networks can effectively model the spatial relationship between electrodes through vector representation and a dynamic routing mechanism. Therefore, it achieves more accurate feature extraction and classification. This paper proposes a spatio-temporal capsule network with a self-correlation routing mechaninsm for the classification of semantic conceptual EEG signals. By improving the feature extraction and routing mechanism, the model is able to more effectively capture the highly variable spatio-temporal features from EEG signals and establish connections between capsules, thereby enhancing classification accuracy and model efficiency. The performance of the proposed model was validated using the publicly accessible semantic concept dataset for imagined and perceived tasks from Bath University. Our model achieved average accuracies of 94.9%, 93.3%, and 78.4% in the three sensory modalities (pictorial, orthographic, and audio), respectively. The overall average accuracy across the three sensory modalities is 88.9%. Compared to existing advanced algorithms, the proposed model achieved state-of-the-art performance, significantly improving classification accuracy. Additionally, the proposed model is more stable and efficient, making it a better decoding solution for SCIP-EEG decoding.

Keywords: EEG decoding; brain-computer interface (BCI); capsule neural network; self-correlation routing; semantic concepts.

PubMed Disclaimer

Conflict of interest statement

The authors declare no conflicts of interest.

Figures

Figure 1
Figure 1
The structure of capsule neural networks.
Figure 2
Figure 2
The overall structure of the proposed Efficient-STCapsNet, consisting of two parts: Spatio-Temporal Capsule-Generation block for creating spatio-temporal capsules and self-correlation routing of spatio-temporal capsules.
Figure 3
Figure 3
The starting phase consists of five convolutional layers and two pooling layers that will extract temporally and spatially significant features in the EEG signal to generate capsules ST-Capsules, or sn,dl for short (l is the number of layers, n denotes the number of capsules, and d denotes the dimensionality of the capsule).
Figure 4
Figure 4
The capsules in layer l + 1 further predict the overall structure or attributes of the input data based on the predictions of the capsule in layer l, as well as prior knowledge and coupling coefficients.
Figure 5
Figure 5
Experimental procedure for generation the dataset of semantic concepts for imagination and perception tasks.
Figure 6
Figure 6
Comparison between the proposed model and state-of-the-art models for overall classification of imagination and perception tasks for all subjects under different sensory modalities.
Figure 7
Figure 7
Comparison of the stability of different models.
Figure 8
Figure 8
The influence of changes in the number of capsules on accuracy and the amounts of parameters.
Figure 9
Figure 9
The influence of changes in the dimension of capsules on classification accuracy.
Figure 10
Figure 10
The impact of different cross-validation folds on classification accuracy.
Figure 11
Figure 11
The Comparison of complexity of different routing mechanisms.

Similar articles

References

    1. Vaughan T.M., Wolpaw J.R. Special issue containing contributions from the Fourth International Brain-Computer Interface Meeting. J. Neural Eng. 2011;8:020201. doi: 10.1088/1741-2560/8/2/020201. - DOI - PubMed
    1. Mandal S.K., Naskar M.N.B. MI brain-computer interfaces: A concise overview. Biomed. Signal Process. Control. 2023;86:105293. doi: 10.1016/j.bspc.2023.105293. - DOI
    1. Lebedev M.A., Nicolelis M.A. Brain–machine interfaces: Past, present and future. Trends Neurosci. 2006;29:536–546. doi: 10.1016/j.tins.2006.07.004. - DOI - PubMed
    1. Dattola S., La Foresta F. Effect of Rehabilitation on Brain Functional Connectivity in a Stroke Patient Affected by Conduction Aphasia. Appl. Sci. 2022;12:5591. doi: 10.3390/app12125991. - DOI
    1. Scano A., Lanzani V., Brambilla C., d’Avella A. Transferring Sensor-Based Assessments to Clinical Practice: The Case of Muscle Synergies. Sensors. 2024;24:3934. doi: 10.3390/s24123934. - DOI - PMC - PubMed

LinkOut - more resources