SpiLiFormer:通过横向抑制增强脉冲变换器
SpiLiFormer: Enhancing Spiking Transformers with Lateral Inhibition
AI总结:
SpiLiFormer通过引入横向抑制机制,提升脉冲变换器对相关上下文的注意力,从而在多个数据集上取得更优性能。
摘要:
基于变换器的脉冲神经网络(SNNs)因其优越的性能和高能效而受到广泛关注。然而,大多数现有基于变换器的SNNs的脉冲注意模块都是从模拟变换器中借鉴的,未能充分解决对无关上下文过度分配注意力的问题。为解决这个基础但被忽视的问题,我们提出了一种受横向抑制启发的脉冲变换器(SpiLiFormer)。它模拟大脑的横向抑制机制,引导模型增强对相关标记的注意力,同时抑制对无关标记的注意力。我们的模型在多个数据集上均实现了最先进的(SOTA)性能,包括CIFAR-10(+0.45%)、CIFAR-100(+0.48%)、CIFAR10-DVS(+2.70%)、N-Caltech101(+1.94%)和ImageNet-1K(+1.6%)。值得注意的是,在ImageNet-1K数据集上,SpiLiFormer(69.9M参数,4时间步,384分辨率)仅使用39%的参数和一半的时间步,就比E-SpikeFormer(173.0M参数,8时间步,384分辨率)高出0.46%。代码和模型检查点可在https://github.com/KirinZheng/SpiLiFormer上公开获取。
英文摘要:
Spiking Neural Networks (SNNs) based on Transformers have garnered significant attention due to their superior performance and high energy efficiency. However, the spiking attention modules of most existing Transformer-based SNNs are adapted from those of analog Transformers, failing to fully address the issue of over-allocating attention to irrelevant contexts. To fix this fundamental yet overlooked issue, we propose a Lateral Inhibition-inspired Spiking Transformer (SpiLiFormer). It emulates the brain's lateral inhibition mechanism, guiding the model to enhance attention to relevant tokens while suppressing attention to irrelevant ones. Our model achieves state-of-the-art (SOTA) performance across multiple datasets, including CIFAR-10 (+0.45%), CIFAR-100 (+0.48%), CIFAR10-DVS (+2.70%), N-Caltech101 (+1.94%), and ImageNet-1K (+1.6%). Notably, on the ImageNet-1K dataset, SpiLiFormer (69.9M parameters, 4 time steps, 384 resolution) outperforms E-SpikeFormer (173.0M parameters, 8 time steps, 384 resolution), a SOTA spiking Transformer, by 0.46% using only 39% of the parameters and half the time steps. The code and model checkpoints are publicly available at https://github.com/KirinZheng/SpiLiFormer.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。