197
Views
0
CrossRef citations to date
0
Altmetric
Research Article

Improved dilation CapsuleNet for motor imagery and mental arithmetic classification based on fNIRS

, , , &
Article: 2335886 | Received 16 Oct 2023, Accepted 25 Mar 2024, Published online: 05 Apr 2024
 

Abstract

Purpose

This study aimed to improve the accuracy of brain-computer interface (BCI) systems based on motor imagery (MI) and mental arithmetic (MA) by utilizing functional near-infrared spectroscopy (fNIRS) and an improved dilation CapsuleNet (ID-CapsuleNet) model.

Methods

The study focused on the characteristics of fNIRS and employed large-kernel dilation convolution to extract hemodynamic features from fNIRS data. Inspired by CapsuleNet’s success in image classification, an ID-CapsuleNet model was designed, combining large-kernel dilation convolution and CapsuleNet. Four publicly available datasets (A, B, C, and D) were utilized for evaluating the proposed model. Datasets A and B were MA type, while datasets C and D were MI type. Ablation experiments were conducted to assess the usefulness of large-kernel convolution, dynamic routing, and dilation convolution.

Results

The average accuracies for each dataset were 95.01%, 76.88%, 74.03%, and 80.29% respectively. Cross-subject average accuracies were 88.72%, 75.80%, 75.78%, and 80.34%. Ablation experiments confirmed the importance of large-kernel convolution, dynamic routing, and dilation convolution in the ID-CapsuleNet model.

Conclusion

The developed ID-CapsuleNet model demonstrated promising potential for enhancing the performance of BCI systems based on MI and MA. The findings contribute to the advancement of BCI technology, offering improved assistive tools for disabled individuals.

Disclosure statement

No potential competing interest was reported by the authors.

Additional information

Funding

This work was supported by the Special Projects in Key Fields Supported by the Technology Development Project of Guangdong Province; Scientific Research of High-level Talents of Wuyi University; Special Fund Project for Science and Technology Innovation Strategy of Guangdong Province.