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

MD-SNN: 基于量化脉冲神经网络的膜电位感知蒸馏

MD-SNN: Membrane Potential-aware Distillation on Quantized Spiking Neural Network

Donghyun Lee, Abhishek Moitra, Youngeun Kim, Ruokai Yin, Priyadarshini Panda

AI总结:

本文提出MD-SNN,通过膜电位感知蒸馏缓解量化后的不匹配问题,提升SNNs在静态和动态数据场景中的性能与硬件效率。

摘要:

脉冲神经网络(SNNs)因其稀疏二进制激活而成为传统神经网络的有前途且节能的替代方案。然而,由于复杂的时空动态和训练过程中跨时间步的多次反向传播计算,它们面临内存和计算开销的挑战。为了缓解这一开销,对SNNs应用压缩技术,如量化。然而,简单地将量化应用于SNNs会导致膜电位的不匹配,而膜电位是产生脉冲的关键因素,从而导致精度下降。在本文中,我们引入了基于量化脉冲神经网络的膜电位感知蒸馏(MD-SNN),该方法利用膜电位来缓解权重、膜电位和批量归一化量化后的不一致。据我们所知,这项研究是首次在SNNs中应用膜电位知识蒸馏。我们在各种数据集上验证了我们的方法,包括CIFAR10、CIFAR100、N-Caltech101和TinyImageNet,证明了其在静态和动态数据场景中的有效性。此外,为了提高硬件效率,我们使用SpikeSim平台评估了MD-SNN,发现MD-SNN在N-Caltech101数据集上,在等精度条件下,比浮点SNNs实现了14.85倍更低的能耗-延迟-面积积(EDAP),2.64倍更高的TOPS/W,以及6.19倍更高的TOPS/mm²。

英文摘要:

Spiking Neural Networks (SNNs) offer a promising and energy-efficient alternative to conventional neural networks, thanks to their sparse binary activation. However, they face challenges regarding memory and computation overhead due to complex spatio-temporal dynamics and the necessity for multiple backpropagation computations across timesteps during training. To mitigate this overhead, compression techniques such as quantization are applied to SNNs. Yet, naively applying quantization to SNNs introduces a mismatch in membrane potential, a crucial factor for the firing of spikes, resulting in accuracy degradation. In this paper, we introduce Membrane-aware Distillation on quantized Spiking Neural Network (MD-SNN), which leverages membrane potential to mitigate discrepancies after weight, membrane potential, and batch normalization quantization. To our knowledge, this study represents the first application of membrane potential knowledge distillation in SNNs. We validate our approach on various datasets, including CIFAR10, CIFAR100, N-Caltech101, and TinyImageNet, demonstrating its effectiveness for both static and dynamic data scenarios. Furthermore, for hardware efficiency, we evaluate the MD-SNN with SpikeSim platform, finding that MD-SNNs achieve 14.85X lower energy-delay-area product (EDAP), 2.64X higher TOPS/W, and 6.19X higher TOPS/mm2 compared to floating point SNNs at iso-accuracy on N-Caltech101 dataset.

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

↑