首页|基于低时延和高精度脉冲神经网络的目标检测

基于低时延和高精度脉冲神经网络的目标检测

扫码查看
人工神经网络(artificial neural network,ANN)在众多领域取得了显著进展,但其对计算资源和能耗的高需求限制了其在硬件端的部署和应用.脉冲神经网络(spiking neural network,SNN)因其低功耗和快速推理的特性,在神经形态硬件上表现出色.然而,SNN的神经元动态和脉冲发放机制导致其训练过程复杂,目前主要研究集中在图像分类任务上,本文尝试将SNN应用于更为复杂的计算机视觉任务.本文以YOLOv3-tiny网络为基础,提出了Spiking YOLOv3 模型,其符合SNN特性的网络模型,在检测任务上实现了更高的准确度,并将平均推理时间减少至约原来工作的 1/4.此外,我们还分析了ANN-SNN转换过程中产生的转换误差,并采用量化激活函数对Spiking YOLOv3 模型进行了优化以减小转换误差.优化后的模型平均推理时间减少至约原来的 1/2,并在VOC与UAV数据集上实现在ANN-SNN无损转换,显著提升了基于该模型的检测效率.
Target Detection Based on Low Latency and High Accuracy Spiking Neural Network
Artificial neural network(ANN)has made significant progress in many fields,but its high demand for computing resources and energy consumption limits its deployment and application on the hardware side.Spiking neural network(SNN)performs well on neural morphology hardware due to its low power consumption and fast inference characteristics.However,the neural dynamics and pulse propagation mechanism of SNN make its training process complex.Current research primarily focuses on image classification tasks.This study attempts to apply SNN to more complex computer vision tasks.This study is based on the YOLOv3 tiny network and proposes the spiking YOLOv3 model,which conforms to the SNN characteristics of the network model.It achieves higher accuracy in detection tasks and reduces the average inference time to about 1/4 of the original work.In addition,this study also analyzes the conversion errors generated during the ANN-SNN conversion process and optimizes the Spiking YOLOv3 model using a quantization activation function to reduce conversion errors.The optimized model reduces the average inference time to about half of the original and achieves lossless conversion on the VOC and UAV datasets in ANN-SNN,significantly improving the detection efficiency based on this model.

spiking neural network(SNN)artificial neural network-spiking neural network(ANN-SNN)conversiontarget detectionlow latency and high accuracy

明晓钰、李翔宇

展开 >

南京理工大学计算机科学与工程学院,南京 210094

脉冲神经网络 人工神经网络-脉冲神经网络(ANN-SNN)转换 目标检测 低时延和高精度

2024

计算机系统应用
中国科学院软件研究所

计算机系统应用

CSTPCD
影响因子:0.449
ISSN:1003-3254
年,卷(期):2024.33(7)