File size: 7,179 Bytes
f420184
 
 
 
 
 
 
 
 
 
 
 
5faae8a
f420184
5faae8a
10f2f8f
5faae8a
2b1a463
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2a66559
2b1a463
 
 
 
 
2a66559
2b1a463
 
 
 
 
 
 
 
 
 
2a66559
2b1a463
 
 
 
 
 
 
 
 
 
2a66559
2b1a463
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
---
language:
- en
---
<div align="center">

<h1>	
Intrinsically Stable Spiking Neural Networks: Overcoming the Performance Barrier in the Absence of Batch Normalization</h1>

<!-- Badges -->
[![Paper](https://img.shields.io/badge/Arxiv-2606.31695-B31B1B.svg?style=flat-square)](https://arxiv.org/abs/2606.31695)
[![ECCV 2026](https://img.shields.io/badge/ECCV%202026-Poster-4b44ce.svg?style=flat-square)](https://www.ecva.net/papers.php)
[![Google Scholar](https://img.shields.io/badge/Google%20Scholar-Paper-4285F4?style=flat-square&logo=google-scholar&logoColor=white)](https://scholar.google.com/scholar?cluster=18039869873167931069)

[![GitHub](https://img.shields.io/badge/GitHub-Repository-black?logo=github)](https://github.com/Ruichen0424/IS-SNN)
[![Hugging Face](https://img.shields.io/badge/Hugging%20Face-Paper-FFD21E?style=flat-square&logo=huggingface&logoColor=black)](https://huggingface.co/papers/2606.31695)
![Hits](https://hits.sh/github.com/ruichen0424/is-snn.svg?label=Hits&color=4f46e5)
</div>

## πŸš€ Introduction

This is the official model repository of the paper **Intrinsically Stable Spiking Neural Networks: Overcoming the Performance Barrier in the Absence of Batch Normalization**, accepted at **ECCV 2026**.

## ✨ Key Highlights

*   **⚑ Zero-Runtime-Overhead SNN:** Removes all dynamic batch normalization (BN) layers. By folding weight standardization operations offline, it restores a pure, hardware-friendly **accumulation-only (addition-only)** inference datapath.
*   **πŸ“‰ Solves Firing-Rate Decay:** Addresses the fundamental issue of catastrophic firing-rate decay/saturation in BN-free SNNs through theoretical *Topology-Aware Weight Standardization* and *Modified Residual Connections*.
*   **πŸ† Scalable & Deep SNNs:** Breaks the depth limitations of prior normalization-free SNNs, delivering high-performance training across deep VGG, ResNet, and **Spiking Transformer** architectures.
*   **πŸ“Š State-of-the-Art Accuracy:** Reaches a competitive **68.05%** top-1 accuracy on **ImageNet** (T=4), matching or outperforming computationally heavy dynamic BN methods.
*   **πŸ”‹ Edge-Hardware Friendly:** Reduces FPGA lookup table (LUT) resource consumption for neuron implementations by **96.4%**, paving the way for ultra-low-power neuromorphic deployment.

## πŸ“„ Abstract
The performance of deep spiking neural networks (SNNs) often relies on batch normalization (BN). However, the advanced dynamic BN variants used in state-of-the-art models introduce runtime multiplications, which weaken the hardware-efficiency motivation of SNNs. To address this tension, we identify catastrophic firing-rate decay as a primary cause of severe performance degradation in normalization-free SNNs. Guided by this insight, this work proposes the Intrinsically Stable SNN (IS-SNN) architecture, which removes activation-normalization layers by enforcing signal homeostasis through topology-aware weight standardization and modified residual connections. By folding the standardization operations into static weights offline, IS-SNN removes the runtime statistics tracking and multiplications introduced by activation normalization, restoring an accumulation-oriented inference datapath. Comprehensive experiments show that IS-SNN achieves performance competitive with or superior to computationally expensive dynamic BN techniques across VGG, ResNet, and Transformer-based models. Notably, it achieves a competitive accuracy of 68.05% on ImageNet and overcomes the severe depth limitations of prior BN-free attempts. Together with a 96.4% reduction in FPGA lookup table resource consumption for neuron implementations, these results support IS-SNN as a practical framework for building accurate and hardware-friendly deep neuromorphic systems.

## πŸ› οΈ Requirements
- python==3.10
- numpy==1.23.5
- spikingjelly==0.0.0.0.14
- torch==2.2.0
- torchvision==0.17.0
- timm==1.0.7

### Environment Setup

We recommend using Anaconda to create a virtual environment:

```bash
conda create -n issnn python=3.10
conda activate issnn
```

Install PyTorch and dependencies:

```bash
# Install PyTorch (Choose based on your CUDA version)
# CUDA 11.8
conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

# Install SpikingJelly and timm
pip install timm==1.0.7
pip install spikingjelly==0.0.0.0.14
```

## πŸ’» Usage
Training on the CIFAR-10 dataset with AMP and Mixup/Cutmix: 
``` bash
CUDA_VISIBLE_DEVICES=0 \
python train.py --batch_size 128 --dataset_path '/ssd/Datasets/CIFAR10/' --dataset 'cifar10'\
                --class_number 10 --epochs 256 --lr 0.02 --weight_decay 5e-4 --amp\
                --timestep 4 --alpha 0.5 --mixup --workers 4 --print_freq 30 --name 'CIFAR10'
```
Training on the ImageNet dataset with AMP: 
``` bash
python train.py --batch_size 256 --dataset_path '/ssd/Datasets/ImageNet/' --dataset 'imagenet'\
                --class_number 1000 --epochs 128 --lr 0.2 --weight_decay 0 --amp\
                --timestep 4 --alpha 0.5 --workers 16 --print_freq 200 --name 'ImageNet'\
                --multiprocessing_distributed
```

## πŸ“Š Main Results

Here is the performance summary of **IS-SNN** across various standard datasets and network architectures. By removing activation-normalization layers, IS-SNN achieves competitive accuracy with zero runtime normalization overhead.

<table>
  <thead>
    <tr>
      <th align="center">Dataset</th>
      <th align="center">Architecture</th>
      <th align="center">Timestep</th>
      <th align="center">Accuracy (%)</th>
    </tr>
  </thead>
  <tbody>
    <tr align="center">
      <td rowspan="1" align="center" style="vertical-align: middle;"><b>ImageNet</b></td>
      <td>SEW-ResNet-34</td>
      <td>4</td>
      <td><b>68.05</b></td>
    </tr>
    <tr align="center">
      <td rowspan="2" align="center" style="vertical-align: middle;"><b>CIFAR-10</b></td>
      <td>VGG-11</td>
      <td>4</td>
      <td><b>95.06</b></td>
    </tr>
    <tr align="center">
      <td>SEW-ResNet-19</td>
      <td>6 / 4 / 2</td>
      <td><b>96.12 / 96.02 / 95.65</b></td>
    </tr>
    <tr align="center">
      <td rowspan="2" align="center" style="vertical-align: middle;"><b>CIFAR-100</b></td>
      <td>VGG-11</td>
      <td>4</td>
      <td><b>77.13</b></td>
    </tr>
    <tr align="center">
      <td>SEW-ResNet-19</td>
      <td>6 / 4 / 2</td>
      <td><b>80.72 / 79.97 / 79.03</b></td>
    </tr>
    <tr align="center">
      <td rowspan="1" align="center" style="vertical-align: middle;"><b>DVS-Gesture</b></td>
      <td>SEW-7B-Net</td>
      <td>16</td>
      <td><b>96.88</b></td>
    </tr>
  </tbody>
</table>
<br>

## πŸ“œ Citation
If you find our code useful for your research, or use the IS-SNN architecture, please consider citing:

```bibtex
@article{ma2026intrinsically,
  title={Intrinsically Stable Spiking Neural Networks: Overcoming the Performance Barrier in the Absence of Batch Normalization},
  author={Ma, Ruichen and Zhang, Xiaoyang and Bai, Jian and Qiao, Guanchao and Meng, Liwei and Ning, Ning and Liu, Yang and Hu, Shaogang},
  journal={arXiv preprint arXiv:2606.31695},
  year={2026}
}
```