LLM2D
时空映射工具用于.spike神经网络
STEM: Spatial-Temporal Mapping Tool For Spiking Neural Networks
作者: Sherif Eissa, Sander Stuijk, Floran De Putter, Andrea Nardi-Dei, Federico Corradi, Henk Corporaal
发布日期: 2/6/2025
arXiv ID: oai:arXiv.org:2502.03287v1

摘要

arXiv:2502.03287v1 类型:交叉 摘要:脉冲神经网络(SNNs)是有前景的仿生第三代神经网络。最近的研究已经训练出了与人工神经网络(ANNs)在准确度上相当的深度SNN模型。尽管SNN的事件驱动和稀疏特性显示了比ANNs更高效的能源计算潜力,但SNN神经元具有随着时间演化的内部状态,这可能会增加数据移动和存储需求,从而使其在能量效率方面相对于ANNs失去优势。本文探讨了拥有神经元状态的能量影响,以及这些影响如何受到所选的映射到现实硬件架构的内存层次结构的影响。因此,我们开发了STEMS,一种SNN的映射设计空间探索工具。STEMS模型了SNN的状态行为,并探索了层内和层间映射优化,以最小化数据移动,同时考虑了空间和时间维度的SNN。使用STEMS,我们在两个基于事件的视觉SNN基准上展示了高达12倍的片外数据移动减少和5倍的能效提升(在层内优化基础上)。最后,神经元状态并不需要所有SNN层。通过优化一个基准的神经元状态,我们展示了神经元状态减少20倍且在无准确率损失的情况下性能提升了1.4倍。