arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.00411cs.NEcs.AIcs.CV

LM-HT SNN:通过可学习多阈值模型提升脉冲神经网络(SNN)对等人工神经网络(ANN)的性能

LM-HT SNN: Enhancing the Performance of SNN to ANN Counterpart through Learnable Multi-hierarchical Threshold Model

School of Computer Science, Peking University(北京大学计算机学院);Institute for Artificial Intelligence, Peking University(北京大学人工智能研究院)

Zecheng Hao, Xinyu Shi, Yujia Liu, Zhaofei Yu, Tiejun Huang

AI总结:

针对脉冲神经网络(SNN)性能落后于人工神经网络(ANN)的问题,提出可学习多阈值LM-HT模型,可动态调节电流与电位泄漏、支持重参数化部署,还能结合ANN-SNN转换框架提升低延迟性能,在多数据集上达SOTA,媲美量化ANN。

摘要:

与传统人工神经网络(ANN)相比,脉冲神经网络(Spiking Neural Network, SNN)凭借其以更节能方式传输信息的固有能力,已引起学术界的广泛关注。然而,尽管此前已有研究通过多种方法优化SNN的学习算法,SNN的性能仍落后于ANN。近期提出的多阈值模型为进一步增强SNN的学习能力提供了更多可能性。本文从数学角度严格分析了多阈值模型、普通脉冲模型与量化ANN之间的关系,随后提出了一种新型LM-HT模型——这是一种等距多阈值模型,能够在时间维度上动态调节全局输入电流和膜电位泄漏。LM-HT模型还可通过重参数化转化为普通单阈值模型,从而实现更灵活的硬件部署。此外,我们注意到LM-HT模型在特殊初始化条件下可与ANN-SNN转换框架无缝集成。这种新型混合学习框架能够有效改善转换后的SNN在低时间延迟下性能较差的问题。大量实验结果表明,我们的模型在各类数据集上的表现均优于此前的最优(state-of-the-art)工作,推动SNN达到了可与量化ANN相媲美的全新性能水平。代码可在https://github.com/hzc1208/LMHT_SNN获取。

英文摘要:

Compared to traditional Artificial Neural Network (ANN), Spiking Neural Network (SNN) has garnered widespread academic interest for its intrinsic ability to transmit information in a more energy-efficient manner. However, despite previous efforts to optimize the learning algorithm of SNNs through various methods, SNNs still lag behind ANNs in terms of performance. The recently proposed multi-threshold model provides more possibilities for further enhancing the learning capability of SNNs. In this paper, we rigorously analyze the relationship among the multi-threshold model, vanilla spiking model and quantized ANNs from a mathematical perspective, then propose a novel LM-HT model, which is an equidistant multi-threshold model that can dynamically regulate the global input current and membrane potential leakage on the time dimension. The LM-HT model can also be transformed into a vanilla single threshold model through reparameterization, thereby achieving more flexible hardware deployment. In addition, we note that the LM-HT model can seamlessly integrate with ANN-SNN Conversion framework under special initialization. This novel hybrid learning framework can effectively improve the relatively poor performance of converted SNNs under low time latency. Extensive experimental results have demonstrated that our model can outperform previous state-of-the-art works on various types of datasets, which promote SNNs to achieve a brand-new level of performance comparable to quantized ANNs. Code is available at https://github.com/hzc1208/LMHT_SNN.

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

↑