atlas

Research map / Architectures and efficiency / Spiking networks

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

ICML 2024 · Architectures and efficiency › Spiking networks · approach: Training deep spiking networks · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SpikeZIP-TF, an ANN-to-SNN conversion method that makes quantized Transformers (ViT, RoBERTa) exactly equivalent to spiking networks by adding spike-equivalent self-attention, Spike-Softmax and Spike-LayerNorm.

Problem

Transformer-based SNNs lag behind their ANN counterparts, and existing ANN-to-SNN conversion methods cannot handle Transformer operators such as attention (activation-activation products), softmax, and layer normalization.

Approach

Introduces the ST-BIF+ neuron, whose accumulated output equals a quantized function, plus spike-equivalent self-attention (SESA) and differential-style Spike-Softmax/Spike-LayerNorm, so that a quantized ViT or RoBERTa converts to an SNN with no accuracy loss.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Convert ViT-S to an SNN with SpikeZIP-TFViT-S · CIFAR10-DVSaccuracy: 88.1 -> 90.5Helped
Convert ViT-S to an SNN with SpikeZIP-TFViT-S · CIFAR-10accuracy: 97.3 -> 98.7Helped
Convert ViT-S to an SNN with SpikeZIP-TFViT-S · CIFAR-100accuracy: 86.9 -> 89.7Helped
Convert ViT-L to an SNN with SpikeZIP-TFImageNet · ImageNettop-1 accuracy: 78.51 -> 83.82Helped
Use SpikeZIP-TF with RoBERTa-L instead of SpikeGPTSST-2 · SST-2accuracy: 88.76 -> 93.79Helped
Use SpikeZIP-TF with RoBERTa-L instead of SpikeBERTMR · MRaccuracy: +3.65%Helped
Increase the SpikeZIP-TF model size from RoBERTa-B to RoBERTa-L (ablation)SpikeZIP-TF · SST-2accuracy: 92.81 -> 93.79Helped
Increase the SpikeZIP-TF model size from ViT-S to ViT-L (ablation)SpikeZIP-TF · ImageNettop-1 accuracy: 81.45 -> 83.82Helped
Reduce quantization level (ablation)SpikeZIP-TF with ViT-B · ImageNetaccuracyHurt
Reduce quantization level (ablation)SpikeZIP-TF with ViT-B · ImageNettime-steps to accuracy increaseHelped
Increase inference time-steps (ablation)SpikeZIP-TF · ImageNetaccuracyHelped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet CIFAR-10 CIFAR-100 CIFAR10-DVS MR Subj SST-2 SST-5 ChnSenti Waimai

Benchmarks: ImageNet CIFAR-10 CIFAR-100 CIFAR10-DVS MR Subj SST-2 SST-5 ChnSenti Waimai

Metrics: top-1 accuracy time-steps power (W) training hours energy (kWh)

Baselines: MST Spikformer Spikformer V2 Spikingformer SpikeBERT SpikeGPT SNN-TextCNN QCFS

Builds on: ST-BIF neuron MST QCFS I-BERT LSQ

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy