atlas

Research map / Architectures and efficiency / Spiking networks

SpikeLM: Towards General Spike-Driven Language Modeling via Elastic Bi-Spiking Mechanisms

ICML 2024 · Architectures and efficiency › Spiking networks · approach: Spiking transformers · 1 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 SpikeLM, a fully spike-driven language model using elastic bi-spiking (ternary, amplitude-scaled, frequency-controlled spikes) that handles both discriminative (GLUE) and generative (summarization, translation) tasks.

Problem

Binary {0,1} spikes in SNNs carry too little semantic information and have uncontrolled firing rates, so SNNs have not been applied to general language tasks, especially generation, with fully spike-driven (matrix-multiplication-free) computation.

Approach

Replace binary LIF spikes with elastic bi-spiking: ternary {-1,0,1} spikes (bidirectional), a layerwise amplitude alpha merged into weights after training, and a distribution-aware threshold alpha = k * mean|m| to control firing rate, trained with a straight-through estimator.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace binary LIF spikes with elastic bi-spikingBERT-based language model · GLUE devAvg. score: 54.9 -> 76.5Helped
Add bidirectional encoding to binary LIF spikes (ablation)LIF-BERT (T=4) · GLUE devGLUE Avg. (%): 54.9 -> 55.7Helped
Add frequency/amplitude encoding to binary LIF spikes (ablation)LIF-BERT (T=4) · GLUE devGLUE Avg. (%): 54.9 -> 71.8Helped
Add frequency/amplitude encoding after bidirectional encoding (ablation)LIF-BERT (T=4) with bidirectional encoding · GLUE devGLUE Avg. (%): 55.7 -> 76.5Helped
Add bidirectional encoding after frequency/amplitude encoding (ablation)LIF-BERT (T=4) with frequency/amplitude encoding · GLUE devGLUE Avg. (%): 71.8 -> 76.5Helped
Use learnable spike thresholds instead of direct fixed thresholds (ablation)SpikeLM · GLUE devGLUE Avg. (%): 55.7 -> 76.8Helped
Set frequency hyperparameter k=2 rather than learnable thresholds (ablation)SpikeLM · GLUE devGLUE Avg. (%): 76.8 -> 76.5No clear effect
Increase frequency hyperparameter from k=2 to k=3 (ablation)SpikeLM · GLUE devGLUE Avg. (%): 76.5 -> 75.1Hurt
Increase frequency hyperparameter from k=2 to k=4 (ablation)SpikeLM · GLUE devGLUE Avg. (%): 76.5 -> 60.8Hurt
Increase frequency hyperparameter from k=2 to k=4 (ablation)SpikeLM · GLUE devEnergy (mJ): 13.7 -> 7.4Helped
Replace binary LIF spikes with elastic bi-spikingBART-based summarization model · XSUMROUGE-L: 28.32 -> 32.92Helped
Replace binary LIF spikes with elastic bi-spikingBART-based summarization model · CNN-DailyMailROUGE-L: 28.05 -> 29.06Helped

Surprising findings

Limitations

Experimental setup

Datasets: BooksCorpus English Wikipedia GLUE XSUM CNN-DailyMail WMT16 En-Ro

Benchmarks: GLUE XSUM CNN-DailyMail WMT16 En-Ro

Metrics: accuracy F1 Spearman correlation ROUGE-1 ROUGE-2 ROUGE-L BLEU energy (mJ)

Baselines: BERT-base LIF-BERT PSN-BERT SpikeBERT SpikingFormer BiPFT BiT BiBERT BinaryBERT TernaryBERT Q2BERT BART-base

Builds on: LIF neurons BERT BART dynamical isometry SpikingFormer

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