atlas

Research map / Architectures and efficiency / Mixture of experts

Spiking Transformer with Experts Mixture

NeurIPS 2024 · Architectures and efficiency › Mixture of experts · approach: Sparse MoE as regularization and attention · 0 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 the Spiking Experts Mixture Mechanism (SEMM), a spike-driven MoE formulation for SNNs, and embeds it into Spiking Transformers as EMSA (head-wise experts in attention) and EMSP (channel-wise experts in MLP), reporting gains on static and neuromorphic datasets.

Problem

Mixture-of-Experts relies on softmax weighting and TopK hard sparsification, which conflict with SNN spike-driven, multiplication-free computation; there was no MoE design for Spiking Transformers.

Approach

Replace softmax/TopK routing with spiking routers whose binary outputs gate spiking experts via element-wise Hadamard product and addition, giving spike-driven dynamic sparse conditional computation with parameter counts comparable to the original SSA and MLP.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add SEMM to Spikformer-8-512Spikformer-8-512 · ImageNet-1kTop-1 accuracy: 73.38 -> 75.93Helped
Add SEMM to SD-Transformer-8-512SD-Transformer-8-512 · ImageNet-1kTop-1 accuracy: 74.57 -> 76.62Helped
Add SEMM to Spikingformer-8-512Spikingformer-8-512 · ImageNet-1kTop-1 accuracy: 74.79 -> 76.03Helped
Add SEMM to SpikformerSpikformer · CIFAR10-DVS (T=16)accuracy: 80.90 -> 82.90Helped
Add SEMM to Spike-Driven TransformerSpike-Driven Transformer · DVS128 Gesture (T=16)accuracy: 99.30 -> 99.30No clear effect
Add SEMM to SpikingformerSpikingformer · CIFAR-10 (T=4)accuracy: 95.81 -> 96.16Helped
Add SEMM to Spike-Driven TransformerSpike-Driven Transformer · CIFAR-100 (T=4)accuracy: 78.40 -> 80.26Helped
Add depth-wise convolution to EMSP (ablation)Spikformer EMSP · CIFAR-100accuracy: 78.17 -> 78.53Helped
Add depth-wise convolution to EMSP (ablation)Spikformer EMSP · CIFAR10-DVSaccuracy: 81.30 -> 82.32Helped
Add depth-wise convolution to EMSP (ablation)Spike-driven Transformer EMSP · CIFAR-100accuracy: 78.95 -> 79.81Helped
Add depth-wise convolution to EMSP (ablation)Spike-driven Transformer EMSP · CIFAR10-DVSaccuracy: 80.51 -> 81.10Helped
Add depth-wise convolution to EMSP (ablation)Spikingformer EMSP · CIFAR-100accuracy: 79.44 -> 79.81Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1k CIFAR10 CIFAR100 CIFAR10-DVS DVS128 Gesture

Benchmarks: ImageNet-1k CIFAR10 CIFAR100 CIFAR10-DVS DVS128 Gesture

Metrics: top-1 accuracy average spiking rate

Baselines: Spikformer Spike-driven Transformer Spikingformer SEW-ResNet MS-ResNet tdBN PLIF DIET-SNN Dspike DSR ANN-MoE with heavy MLP experts

Builds on: Spikformer Spike-driven Transformer Spikingformer Sparsely-gated mixture-of-experts SEW ResNet

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