arrow
Return

Parallel hyperparameter optimization of spiking neural networks

delete2024-12-01
delete1
delete
OA
AI
T
Thomas Firmin *
P
Pierre Boulet
E
El-Ghazali Talbi
DOI:10.1016/j.neucom.2024.128483delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Hyperparameter optimization of spiking neural networks (SNNs) is a difficult task which has not yet been deeply investigated in the literature. In this work, we designed a scalable constrained Bayesian based optimization algorithm that prevents sampling in non-spiking areas of an efficient high dimensional search space. These search spaces contain infeasible solutions that output no or only a few spikes during the training or testing phases, we call such a mode a silent network. Finding them is difficult, as many hyperparameters are highly correlated to the architecture and to the dataset. We leverage silent networks by designing a spike- based early stopping criterion to accelerate the optimization process of SNNs trained by spike timing dependent plasticity and surrogate gradient. We parallelized the optimization algorithm asynchronously, and ran largescale experiments on heterogeneous multi-GPU Petascale architecture. Results show that by considering silent networks, we can design more flexible high-dimensional search spaces while maintaining a good efficacy. The optimization algorithm was able to focus on networks with high performances by preventing costly and worthless computation of silent networks.
Keywords:
Spiking neural networks
Hyperparameter optimization
Parallel asynchronous optimization
Bayesian optimization
STDP
SLAYER
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Neurocomputing cover
Neurocomputing
IF:
6.5
Papers:
2.5W
Citations:
6.5W

Organization

I
Inria
Scholars:
3.5K
Papers: 2.5K
Citations: 343