arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.07331cs.NE

脉冲神经网络中高效的基于事件的延迟学习

Efficient Event-based Delay Learning in Spiking Neural Networks

Balázs Mészáros, James C. Knight, Thomas Nowotny

AI总结:

本文提出了一种基于EventProp的SNNs延迟训练方法,可精确计算权重和延迟梯度,支持多脉冲且首次应用于循环SNNs,实验证明其能优化延迟提升分类准确率,且内存占用减半、速度最高快26倍。

摘要:

脉冲神经网络(SNNs)使用稀疏通信进行计算,并作为传统人工神经网络(ANNs)的更节能替代方案正吸引着越来越多的关注。虽然标准ANNs是无状态的,但脉冲神经元是有状态的,因此具有内在的循环性,使其非常适合时空任务。然而,这种内在记忆的持续时间受限于突触和膜时间常数。延迟是一种强大的附加机制,在本文中,我们提出了一种新颖的基于事件的SNNs延迟训练方法,该方法基于EventProp形式化,能够精确计算关于权重和延迟的梯度。我们的方法支持每个神经元发放多个脉冲,并且据我们所知,这是首个应用于循环SNNs的延迟学习算法。我们在一个简单的序列检测任务,以及Yin-Yang、Spiking Heidelberg Digits、Spiking Speech Commands和Braille letter reading数据集上评估了我们的方法,证明我们的算法能够从次优初始条件中优化延迟,并与无延迟架构相比提高了分类准确率。我们还发现,循环延迟在小型网络中特别有益。最后,我们表明我们的方法使用的内存不到当前最先进延迟学习方法的一半,并且速度最高快26倍。

英文摘要:

Spiking Neural Networks (SNNs) compute using sparse communication and are attracting increased attention as a more energy-efficient alternative to traditional Artificial Neural Networks~(ANNs). While standard ANNs are stateless, spiking neurons are stateful and hence intrinsically recurrent, making them well-suited for spatio-temporal tasks. However, the duration of this intrinsic memory is limited by synaptic and membrane time constants. Delays are a powerful additional mechanism and, in this paper, we propose a novel event-based training method for SNNs with delays, grounded in the EventProp formalism which enables the calculation of exact gradients with respect to weights and delays. Our method supports multiple spikes per neuron and, to the best of our knowledge, is the first delay learning algorithm to be applied to recurrent SNNs. We evaluate our method on a simple sequence detection task, as well as the Yin-Yang, Spiking Heidelberg Digits, Spiking Speech Commands and Braille letter reading datasets, demonstrating that our algorithm can optimise delays from suboptimal initial conditions and enhance classification accuracy compared to architectures without delays. We also find that recurrent delays are particularly beneficial in small networks. Finally, we show that our approach uses less than half the memory of the current state-of-the-art delay-learning method and is up to 26x faster.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑