May, 2023

通过精确的梯度反向传播改进下采样,提高基于 Transformer 的突触型神经网络性能

TL;DR本文提出了 ConvBN-MaxPooling-LIF(CML)这种新型降采样方式以解决现有状态下深度脉冲神经网络(包括 Spikformer 和 Spikingformer)中梯度误差向后传播不准确的问题,并且在 ImageNet、CIFAR10、CIFAR100、CIFAR10-DVS、DVS128-Gesture 等数据集上验证 CML 显著提高模型性能,与 Spikingformer 比较,例如在 ImageNet,CIFAR10-DVS 上分别可以增加 1.79%,81.4% 的性能。