SpikeNAS-Bench: Benchmarking NAS Algorithms for Spiking Neural Network Architecture
In recent years, neural architecture search (NAS) has marked significant advancements, yet its efficacy is marred by the dependence on substantial computational resources. To mitigate this, the development of NAS benchmarks has emerged, offering datasets that enumerate all potential network architectures and their performances within a predefined search space. Nonetheless, these benchmarks predominantly focus on convolutional architectures, which are criticized for their limited interpretability and suboptimal hardware efficiency. Recognizing the untapped potential of spiking neural networks (SNNs)—often hailed as the third generation of neural networks due to their biological realism and computational thrift—this study introduces SpikeNAS-Bench. As a pioneering benchmark for SNN, SpikeNAS-Bench utilizes a cell-based search space, integrating leaky integrate-and-fire neurons with variable thresholds as candidate operations. It encompasses 15 625 candidate architectures, rigorously evaluated on CIFAR10, CIFAR100, and Tiny-ImageNet datasets. This article delves into the architectural nuances of SpikeNAS-Bench, leveraging various criteria to underscore the benchmark's utility and presenting insights that could steer future NAS algorithm designs. Moreover, we assess the benchmark's consistency through three distinct proxy types: zero-cost-based, early-stop-based, and predictor-based proxies. Additionally, the article benchmarks seven contemporary NAS algorithms to attest to SpikeNAS-Bench's broad applicability. We commit to providing training logs, diagnostic data for all candidate architectures, and we promise to release all code and datasets postacceptance, aiming to catalyze further exploration and innovation within the SNN domain.
Authors
- Zhengkun Liu (ORCID: https://orcid.org/0000-0002-6481-835X)
- Biao Sun (ORCID: https://orcid.org/0000-0002-4124-9350)
- Wenfeng Zhao
- Hang Su (ORCID: https://orcid.org/0000-0002-6877-6783)
- Ting Li (ORCID: https://orcid.org/0000-0001-9756-1442)
- Lin Gan
- Guo-Zheng Sun (ORCID: https://orcid.org/0009-0004-7277-9778)
Institutions
- Binghamton University (US)
- Tianjin University (CN)
- Northwestern Polytechnical University (CN)
- Chinese Academy of Medical Sciences & Peking Union Medical College (CN)
- Politecnico di Milano (IT)
Publication Details
- Journal
- IEEE Transactions on Artificial Intelligence
- Published
- 2025-01-27
- DOI
- https://doi.org/10.1109/tai.2025.3534136
- Citations
- 3
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 2.19