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

用于高性能和高能效目标检测的整数值训练与脉冲驱动推理脉冲神经网络

Integer-Valued Training and Spike-Driven Inference Spiking Neural Network for High-performance and Energy-efficient Object Detection

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Xi’an Jiaotong University(西安交通大学)

Xinhao Luo, Man Yao, Yuhong Chou, Bo Xu, Guoqi Li

AI总结:

针对SNNs在目标检测任务中性能不足的问题,本研究设计SpikeYOLO架构并提出新型脉冲神经元,在COCO和Gen1数据集上显著提升检测性能,同时提高能效。

摘要:

受大脑启发的脉冲神经网络(SNNs)相比人工神经网络(ANNs)具有生物合理性和低功耗优势。目前SNNs的应用局限于简单分类任务,因为其性能较差。本工作聚焦于弥合ANNs与SNNs在目标检测任务上的性能差距。我们的设计围绕网络架构和脉冲神经元展开。首先,YOLO系列转换为相应的脉冲版本时,过于复杂的模块设计会导致脉冲退化。我们通过简化原始YOLO并融入元SNN块,设计了SpikeYOLO架构来解决该问题。其次,在脉冲神经元将膜电位转换为二进制脉冲的过程中,目标检测对量化误差更敏感。为应对这一挑战,我们设计了一种新型脉冲神经元:训练阶段激活整数值,推理阶段通过扩展虚拟时间步保持脉冲驱动。所提方法在静态和神经形态目标检测数据集上均得到验证。在静态COCO数据集上,我们获得66.2%的mAP@50和48.9%的mAP@50:95,分别比此前最先进的SNN高出15.0%和18.7%。在神经形态Gen1数据集上,我们实现了67.2%的mAP@50,比具有等效架构的ANN高出2.5%,且能效提升了5.7倍。代码:https://github.com/BICLab/SpikeYOLO

英文摘要:

Brain-inspired Spiking Neural Networks (SNNs) have bio-plausibility and low-power advantages over Artificial Neural Networks (ANNs). Applications of SNNs are currently limited to simple classification tasks because of their poor performance. In this work, we focus on bridging the performance gap between ANNs and SNNs on object detection. Our design revolves around network architecture and spiking neuron. First, the overly complex module design causes spike degradation when the YOLO series is converted to the corresponding spiking version. We design a SpikeYOLO architecture to solve this problem by simplifying the vanilla YOLO and incorporating meta SNN blocks. Second, object detection is more sensitive to quantization errors in the conversion of membrane potentials into binary spikes by spiking neurons. To address this challenge, we design a new spiking neuron that activates Integer values during training while maintaining spike-driven by extending virtual timesteps during inference. The proposed method is validated on both static and neuromorphic object detection datasets. On the static COCO dataset, we obtain 66.2% mAP@50 and 48.9% mAP@50:95, which is +15.0% and +18.7% higher than the prior state-of-the-art SNN, respectively. On the neuromorphic Gen1 dataset, we achieve 67.2% mAP@50, which is +2.5% greater than the ANN with equivalent architecture, and the energy efficiency is improved by 5.7*. Code: https://github.com/BICLab/SpikeYOLO

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

↑