Mamba with Split-Based Pyramidal Convolution and Kolmogorov-Arnold Network-Channel-Spatial Attention for Electroencephalogram Classification

Zhe Li ^*

• Hankou University, Jiangxiangzhen, China

Deep learning is widely used in brain electrical signal studies, among which the brain-computer interface is an important direction. Deep learning can effectively improve the performance of BCI machines, which is of great medical and commercial value.. This paper introduces an efficient deep learning model for classifying brain electrical signals based on a Mamba structure enhanced with split-based pyramidal convolution (PySPConv) and Kolmogorov-Arnold network (KAN)-channel-spatial attention (KSA) mechanisms. Incorporating KANs into the attention module of the proposed KSA-Mamba-PySPConv model better approximates the sample function while obtaining local network features. PySPConv, on the other hand, swiftly and efficiently extracts multi-scale fusion features from input data. This integration allows the model to reinforce feature extraction at each layer in Mamba's structure. The model achieves a 96.76% accuracy on the eegmmidb dataset and demonstrates state-of-the-art performance across metrics such as the F1 score, precision, and recall. KSA-Mamba-PySPConv promises to be an effective tool in electroencephalogram classification in brain-computer interface systems.