LSTM-SENet: category decoding model with an attention mechanism
Penghui Ding, Chi Zhang, Linyuan Wang, Guoen Hu, Bin Yan, Tong Li · Fourteenth International Conference on Digital Image Processing (ICDIP 2022) · 2022
With the development of deep neural networks (DNN), building visual decoding models based on functional magnetic resonance imaging (fMRI) to simulate the visual system of the human brain and studying visual mechanisms have become a research hotspot. Although existing visual decoding models built using DNNs have achieved a certain accuracy, most models ignore the differences between different voxels. Among them, the BRNN-based category decoding model uses the bidirectional long short term memory (LSTM) network to simulate the visual bidirectional information flow, which improves the decoding accuracy, but it uses the voxels of each brain area as an overall input model. Therefore, we embed the channel attention module, the Squeeze-and-Excitation Networks (SENet), into the LSTM network to construct an LSTM-SENet vision that introduces an attention mechanism The decoding model allows the model to learn by itself and assign different weights to each voxel, focusing on important voxels, thereby improving the classification accuracy of natural images. The experimental results show that our method improves the accuracy of (three-level) category decoding than other methods, and the results further verify the effectiveness of building a visual decoding model based on the visual mechanism.