Multi-level brain-guided fusion to reinforce spiking neural network in image classification

Zahra Imani,Mehdi Ezoji
DOI: https://doi.org/10.1007/s11042-024-19665-z
IF: 2.577
2024-06-21
Multimedia Tools and Applications
Abstract:Spiking Neural Networks (SNNs) can be a suitable alternative to deep artificial neural networks due to efficient computation and low-power event-driven information analysis. Nevertheless, SNN cannot compete with Artificial Neural Network (ANN) in terms of performance due to the challenges in the training process. In this work, an attempt has been made to improve the performance of SNN in the image classification task without changing the training method. To this end, SNN is used to encode image in a brain-guided model. Batch normalization through time layers are used in SNN architecture to train from scratch with low latency. In this work, EEG-evoked signal information, provided by an LSTM-based network, is used to construct a bio-visual path. A pooling-based spatial coding is proposed to down-sample the visual feature maps. Manifold transferring is applied to project each visual feature into the corresponding implicit EEG-based feature. The image-based reduced feature and its EEG-based feature, detected by manifold transferring, are fused to prepare the input for an SVM classifier. In the inference phase, without the need for EEG acquisition, the output of the manifold transfer block, provided in the training process, is combined with reduced visual features to classify the input image. The experimental results show that the multi-level fusion strategy in the brain-guided network improves the classification accuracy of SNN in 3-class and 5-class image classification problems by up to 4% and 15%, respectively.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?