arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.03470cs.NEcs.AI

SpikeZIP-TF:转换即一切——基于Transformer的SNN

SpikeZIP-TF: Conversion is All You Need for Transformer-based SNN

School of Electronic Information and Electrical Engineering, Shanghai Jiao Tong University(上海交通大学电子信息与电气工程学院);Huawei Technologies(华为技术有限公司)

Kang You, Zekai Xu, Chen Nie, Zhijie Deng, Qinghai Guo, Xiang Wang, Zhezhi He

AI总结:

针对Transformer-based SNN精度低于ANN的问题,提出SpikeZIP-TF转换方法,实现ANN与SNN完全等价,在ImageNet和SST-2上分别达83.82%和93.79%精度,超越现有最优。

摘要:

脉冲神经网络(SNN)因其高效率和准确性的特点而备受关注。目前,ANN到SNN的转换方法可以在计算机视觉(CV)任务中的CNN结构上,以超低延迟(8个时间步)获得与ANN精度相当的SNN。然而,由于基于Transformer的网络在CV和自然语言处理(NLP)方面均取得了卓越的精度,基于Transformer的SNN在精度上仍低于对应的ANN。在这项工作中,我们提出了一种新颖的ANN到SNN转换方法,称为SpikeZIP-TF,其中ANN和SNN完全等价,因此不会引起精度下降。SpikeZIP-TF在CV数据集(ImageNet)上达到83.82%的精度,在NLP数据集(SST-2)上达到93.79%的精度,均高于最先进的基于Transformer的SNN。代码可在GitHub上获取:https://github.com/Intelligent-Computing-Research-Group/SpikeZIP_transformer

英文摘要:

Spiking neural network (SNN) has attracted great attention due to its characteristic of high efficiency and accuracy. Currently, the ANN-to-SNN conversion methods can obtain ANN on-par accuracy SNN with ultra-low latency (8 time-steps) in CNN structure on computer vision (CV) tasks. However, as Transformer-based networks have achieved prevailing precision on both CV and natural language processing (NLP), the Transformer-based SNNs are still encounting the lower accuracy w.r.t the ANN counterparts. In this work, we introduce a novel ANN-to-SNN conversion method called SpikeZIP-TF, where ANN and SNN are exactly equivalent, thus incurring no accuracy degradation. SpikeZIP-TF achieves 83.82% accuracy on CV dataset (ImageNet) and 93.79% accuracy on NLP dataset (SST-2), which are higher than SOTA Transformer-based SNNs. The code is available in GitHub: https://github.com/Intelligent-Computing-Research-Group/SpikeZIP_transformer

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑