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

微调代理梯度学习以实现脉冲神经网络的最优硬件性能

Fine-Tuning Surrogate Gradient Learning for Optimal Hardware Performance in Spiking Neural Networks

Ilkin Aliyev, Tosiron Adegbija

AI总结:

本研究探讨脉冲神经网络中代理梯度函数、beta和膜阈值等超参数对硬件性能的影响,发现微调这些参数可在几乎不损失准确率的情况下显著降低推理延迟并提升加速器效率。

摘要:

脉冲神经网络(SNNs)中高度稀疏的激活在硬件中仔细利用时可提供巨大的能效优势。SNN中稀疏性的行为独特地由数据集和训练超参数塑造。这项工作揭示了训练对硬件性能影响的新见解。具体而言,我们探讨了模型准确性与硬件效率之间的权衡。我们关注三个关键超参数:代理梯度函数、beta和膜阈值。在基于FPGA的硬件平台上的结果表明,在SVHN数据集上,快速sigmoid代理函数相比反正切代理函数产生更低的发放率,同时准确率相似。此外,通过交叉扫描beta和膜阈值超参数,与默认设置相比,我们实现了硬件推理延迟降低48%,而推理准确率仅损失2.88%。总体而言,这项研究强调了微调模型超参数对于设计高效SNN硬件加速器至关重要,通过微调模型与最近的工作相比实现了加速器效率(FPS/W)1.72倍的提升。

英文摘要:

The highly sparse activations in Spiking Neural Networks (SNNs) can provide tremendous energy efficiency benefits when carefully exploited in hardware. The behavior of sparsity in SNNs is uniquely shaped by the dataset and training hyperparameters. This work reveals novel insights into the impacts of training on hardware performance. Specifically, we explore the trade-offs between model accuracy and hardware efficiency. We focus on three key hyperparameters: surrogate gradient functions, beta, and membrane threshold. Results on an FPGA-based hardware platform show that the fast sigmoid surrogate function yields a lower firing rate with similar accuracy compared to the arctangent surrogate on the SVHN dataset. Furthermore, by cross-sweeping the beta and membrane threshold hyperparameters, we can achieve a 48% reduction in hardware-based inference latency with only 2.88% trade-off in inference accuracy compared to the default setting. Overall, this study highlights the importance of fine-tuning model hyperparameters as crucial for designing efficient SNN hardware accelerators, evidenced by the fine-tuned model achieving a 1.72x improvement in accelerator efficiency (FPS/W) compared to the most recent work.

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

↑