HR-SNN: an End-to-End Spiking Neural Network for Four-class Classification Motor Imagery Brain-Computer Interface

Yulin Li,Liangwei Fan,Hui Shen,Dewen Hu
DOI: https://doi.org/10.1109/tcds.2024.3395443
IF: 4.546
2024-01-01
IEEE Transactions on Cognitive and Developmental Systems
Abstract:Spiking Neural Network (SNN) excels in processing temporal information and conserving energy, particularly when deployed on neuromorphic hardware. These strengths position SNN as an ideal choice for developing wearable Brain-Computer Interface (BCI) devices. However, the application of SNN in complex BCI tasks, like four-class Motor Imagery classification, is limited. In light of this, this study introduces a powerful SNN architecture hybrid response SNN (HR-SNN). We employ parameter-wise gradient descent methods to optimize spike encoding efficiency. The SNN’s frequency perception is improved by integrating a hybrid response spiking module. In addition, a diff-potential spiking decoder is designed to optimize SNN output potential utilization. Validation experiments are performed on Physionet and BCI competition IV 2a datasets. On Physionet, our model achieves accuracies of 67.24% and 74.95% using global training and subject-specific transfer learning, respectively. On BCI competition IV 2a, our approach attains an average accuracy of 77.58%, surpassing all the compared SNN models and demonstrating competitiveness against SOTA convolution neural network (CNN) approaches. We validate the robustness of HR-SNN under noise and channel loss scenarios. Additionally, energy analysis reveals HR-SNN’s superior energy efficiency compared to existing CNN models. Notably, HR-SNN exhibits a 2-16 times energy consumption advantage over existing SNN methods.
What problem does this paper attempt to address?