Adaptive Gradient Learning for Spiking Neural Networks by Exploiting Membrane Potential Dynamics

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jiang, Jiaqiang, Wang, Lei, Jiang, Runhao, Fan, Jing, Yan, Rui
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909614622113792
author Jiang, Jiaqiang
Wang, Lei
Jiang, Runhao
Fan, Jing
Yan, Rui
author_facet Jiang, Jiaqiang
Wang, Lei
Jiang, Runhao
Fan, Jing
Yan, Rui
contents Brain-inspired spiking neural networks (SNNs) are recognized as a promising avenue for achieving efficient, low-energy neuromorphic computing. Recent advancements have focused on directly training high-performance SNNs by estimating the approximate gradients of spiking activity through a continuous function with constant sharpness, known as surrogate gradient (SG) learning. However, as spikes propagate among neurons, the distribution of membrane potential dynamics (MPD) will deviate from the gradient-available interval of fixed SG, hindering SNNs from searching the optimal solution space. To maintain the stability of gradient flows, SG needs to align with evolving MPD. Here, we propose adaptive gradient learning for SNNs by exploiting MPD, namely MPD-AGL. It fully accounts for the underlying factors contributing to membrane potential shifts and establishes a dynamic association between SG and MPD at different timesteps to relax gradient estimation, which provides a new degree of freedom for SG learning. Experimental results demonstrate that our method achieves excellent performance at low latency. Moreover, it increases the proportion of neurons that fall into the gradient-available interval compared to fixed SG, effectively mitigating the gradient vanishing problem.
format Preprint
id arxiv_https___arxiv_org_abs_2505_11863
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive Gradient Learning for Spiking Neural Networks by Exploiting Membrane Potential Dynamics
Jiang, Jiaqiang
Wang, Lei
Jiang, Runhao
Fan, Jing
Yan, Rui
Neural and Evolutionary Computing
Brain-inspired spiking neural networks (SNNs) are recognized as a promising avenue for achieving efficient, low-energy neuromorphic computing. Recent advancements have focused on directly training high-performance SNNs by estimating the approximate gradients of spiking activity through a continuous function with constant sharpness, known as surrogate gradient (SG) learning. However, as spikes propagate among neurons, the distribution of membrane potential dynamics (MPD) will deviate from the gradient-available interval of fixed SG, hindering SNNs from searching the optimal solution space. To maintain the stability of gradient flows, SG needs to align with evolving MPD. Here, we propose adaptive gradient learning for SNNs by exploiting MPD, namely MPD-AGL. It fully accounts for the underlying factors contributing to membrane potential shifts and establishes a dynamic association between SG and MPD at different timesteps to relax gradient estimation, which provides a new degree of freedom for SG learning. Experimental results demonstrate that our method achieves excellent performance at low latency. Moreover, it increases the proportion of neurons that fall into the gradient-available interval compared to fixed SG, effectively mitigating the gradient vanishing problem.
title Adaptive Gradient Learning for Spiking Neural Networks by Exploiting Membrane Potential Dynamics
topic Neural and Evolutionary Computing
url https://arxiv.org/abs/2505.11863