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

在脉冲神经网络中利用替代梯度下降实现节能目标检测

Enabling energy-Efficient object detection with surrogate gradient descent in spiking neural networks

Sun Yat-sen University(中山大学)

Jilong Luo, Shanlin Xiao, Yinsheng Chen, Zhiyi Yu

AI总结:

针对脉冲神经网络训练困难及缺乏目标检测解码策略的问题,提出电流均值解码方法并基于梯度替代构建SNN-YOLOv3模型,在PASCAL VOC上实现61.87% mAP,较SpikingYOLO提升近10%且能耗降低两个数量级。

摘要:

脉冲神经网络(SNNs)是一种生物合理的神经网络模型,在事件驱动处理和时空信息处理方面具有显著优势,使其成为节能目标检测的理想选择。然而,生物神经元动力学模型的不可微性给SNNs的训练带来了挑战。此外,目前缺乏适用于SNNs目标检测的解码策略。本研究提出了电流均值解码(CMD)方法,通过解决回归问题来促进用于目标检测任务的深度SNNs训练。基于梯度替代和CMD,我们提出了用于目标检测的SNN-YOLOv3模型。实验表明,SNN-YOLOv3在PASCAL VOC数据集上仅使用6个时间步即可实现61.87%的mAP的优异性能。与SpikingYOLO相比,我们将mAP提高了近10%,同时将能耗降低了两个数量级。

英文摘要:

Spiking Neural Networks (SNNs) are a biologically plausible neural network model with significant advantages in both event-driven processing and spatio-temporal information processing, rendering SNNs an appealing choice for energyefficient object detection. However, the non-differentiability of the biological neuronal dynamics model presents a challenge during the training of SNNs. Furthermore, a suitable decoding strategy for object detection in SNNs is currently lacking. In this study, we introduce the Current Mean Decoding (CMD) method, which solves the regression problem to facilitate the training of deep SNNs for object detection tasks. Based on the gradient surrogate and CMD, we propose the SNN-YOLOv3 model for object detection. Our experiments demonstrate that SNN-YOLOv3 achieves a remarkable performance with an mAP of 61.87% on the PASCAL VOC dataset, requiring only 6 time steps. Compared to SpikingYOLO, we have managed to increase mAP by nearly 10% while reducing energy consumption by two orders of magnitude.

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

↑