高级搜索

留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

基于DT-LIF神经元与SSD的脉冲神经网络目标检测方法

周雅 栗心怡 武喜艳 赵宇飞 宋勇

周雅, 栗心怡, 武喜艳, 赵宇飞, 宋勇. 基于DT-LIF神经元与SSD的脉冲神经网络目标检测方法[J]. 电子与信息学报, 2023, 45(8): 2722-2730. doi: 10.11999/JEIT221367
引用本文: 周雅, 栗心怡, 武喜艳, 赵宇飞, 宋勇. 基于DT-LIF神经元与SSD的脉冲神经网络目标检测方法[J]. 电子与信息学报, 2023, 45(8): 2722-2730. doi: 10.11999/JEIT221367
Lei FENG, Kunyi XIE, Liang ZHU, Xuesong QIU, Shaoyong GUO. 5G Ultra-Reliable and Low Latency Communication Resource Scheduling for Power Business Quality Assurance[J]. Journal of Electronics & Information Technology, 2021, 43(12): 3418-3426. doi: 10.11999/JEIT210509
Citation: ZHOU Ya, LI Xinyi, WU Xiyan, ZHAO Yufei, SONG Yong. Object Detection Method with Spiking Neural Network Based on DT-LIF Neuron and SSD[J]. Journal of Electronics & Information Technology, 2023, 45(8): 2722-2730. doi: 10.11999/JEIT221367

基于DT-LIF神经元与SSD的脉冲神经网络目标检测方法

doi: 10.11999/JEIT221367
基金项目: 国家自然科学基金(82272130, U22A20103)
详细信息
    作者简介:

    周雅:女,副教授,研究方向为智能光电信息处理

    栗心怡:女,硕士生,研究方向为类脑计算

    武喜艳:女,博士生,研究方向为脉冲神经网络及其应用

    赵宇飞:男,博士后,研究方向为面向计算机视觉的类脑计算

    宋勇:男,教授,研究方向为类脑计算、智能交互等

    通讯作者:

    宋勇 yongsong@bit.edu.cn

  • 中图分类号: TN911.73; TP391.41

Object Detection Method with Spiking Neural Network Based on DT-LIF Neuron and SSD

Funds: The National Natural Science Foundation of China (82272130, U22A20103)
  • 摘要: 相对于传统人工神经网络(ANN),脉冲神经网络(SNN)具有生物可解释性、计算效率高等优势。然而,对于目标检测任务,SNN存在训练难度大、精度低等问题。针对上述问题,该文提出一种基于动态阈值LIF神经元(DT-LIF)与单镜头多盒检测器(SSD)的SNN目标检测方法。首先,设计了一种DT-LIF神经元模型,该模型可根据累积的膜电位动态调整神经元的阈值,以驱动深层网络的脉冲活动,提高推理速度。同时,以DT-LIF神经元为基元,构建了一种基于SSD的混合SNN。该网络以脉冲视觉几何群网络(Spiking VGG)和脉冲密集连接卷积网络(Spiking DenseNet)为主干(Backbone),具有由批处理归一化(BN)层、脉冲卷积(SC)层与DT-LIF神经元构成的3个额外层和SSD预测框头(Head)。实验结果表明,相对于LIF神经元网络,DT-LIF神经元网络在Prophesee GEN1数据集上的目标检测精度提高了25.2%。对比AsyNet算法,所提方法的目标检测精度提高了17.9%。
  • 目标检测是一种以图像分类为基础的视觉任务,是计算机视觉领域的重要研究方向之一。一般情况下,目标检测主要解决两个问题:一是通过绘制边界框精准定位单个或多个目标所在位置(定位),二是判断目标属于哪种类别(识别)。近年来,基于深度学习的目标检测算法得到了快速发展,并逐渐成为主流。此类算法可分为两类:一类是以区域卷积神经网络(Region-CNN, R-CNN)[1]、空间金字塔池化网络(Spatial Pyramid Pooling-Net, SPP-Net)[2]为代表Two-stage算法,即在特征提取后首先生成一个可能包含待检测目标的候选框,再通过卷积神经网络进行分类,进一步实现目标定位和识别;另一类是以“你只看1次”算法(You Only Look Once, YOLO)[3]、单镜头多盒检测器(Single Shot multibox Detector, SSD)[4]为代表的one-stage算法,它利用端对端的方法,仅通过一步即可同时实现目标位置与类别的预测,在推理速度方面具有优势。

    另一方面,虽然更深、更复杂的人工神经网络(Artificial Neural Network, ANN)可以实现更高的精度,但其对计算成本的需求也大大增加[5]。相对而言,脉冲神经网络(Spiking Neural Network, SNN)的神经元之间通过尖峰脉冲的独立、离散电信号相互通信,只有当膜电位达到放电阈值[6]时,才通过脉冲传递信息。这种推理方法更接近生物神经元的特性,使得SNN比常规ANN具有更小的功耗和更快的推理速度。

    目前,已有的SNN主要应用于图像分类等相对简单的任务,对于同时实现定位与分类的目标检测等相对复杂的任务,其精度较低。例如:Kim等人[7]利用间接有监督学习,同时通过信道归一化为深度SNN传输信息,将YOLO转换成SNN版本的目标检测架构,但这种转换方法一定程度上丢失了SNN低功耗的优势,其性能仍低于标准YOLO。Chakraborty等人[8]提出了一种基于RetinaNet[9]的SNN目标检测器,采用混合学习方式,在主干网络中添加了由ANN对应网络转换而来的Spiking网络块,但同样由于这种转换方式会带来性能上的损失,其检测结果较RetinaNet模型仍有差距。Kugele等人[10]提出了一种混合SNN-ANN的架构用以解决目标检测任务,通过结合反向传播和替代梯度规则进行端到端训练,与类似的ANN网络架构相比精度损失很小,然而该方法使用常规脉冲神经元,存在推理时间较长等问题。

    针对上述问题,本文提出一种基于动态阈值LIF神经元(Dynamic Threshold Leaky Integrate-and-Fire, DT-LIF)与SSD的脉冲神经网络目标检测方法。首先,设计了一种动态阈值LIF神经元DT-LIF,该神经元模型可以在每个时间步长上动态调整阈值来增加脉冲活动,以减少延迟;然后,以DT-LIF神经元为基元,构建了一种基于SSD的混合SNN。该网络以脉冲视觉几何群网络(Spiking Visual Geometry Group, Spiking VGG)和脉冲密集连接卷积网络 (Spiking Densely connected convolutional Network, Spiking DenseNet)为主干(Backbone),具有由批处理归一化(Batch Normalization, BN)层、脉冲卷积(Spiking Convolution, SC)层与DT-LIF神经元构成的3个额外层和SSD预测框头(Head)。最终,在无需额外转换(ANN-to-SNN)的前提下,实现基于SNN的高精度目标检测。

    与生物神经系统类似,SNN采用脉冲序列表示信息,通过发射脉冲在空间域和时间域两个维度传递信息,可模拟信息在人类大脑中的编码和处理过程。在脉冲事件稀疏性的驱动下,SNN兼具计算高效性和生物可解释性,较ANN具有更高的计算效率和更快的推理速度。目前,基于SNN的学习方法主要可分为3类[11]

    (1)基于脉冲时间依赖可塑性(Spike-Timing Dependent Plasticity, STDP)[12]的无监督学习。STDP是Hebb[13]无监督学习算法的一种变体,连接突触前、后神经元的权重根据二者之间的放电延迟而修改。其中,Kheradpisheh等人[14]提出了一种基于STDP的脉冲深度卷积神经网络,用于目标识别。由于STDP是利用突触可塑性进行局部调整,不足以实现突触个体与神经系统整体目标的协调,因而难以应用于大规模深度网络。

    (2)ANN-to-SNN间接有监督学习。该方法先训练一个ANN模型,再利用脉冲神经元平均脉冲发放率和ReLU激活函数之间的关系,在完成原始ANN训练后,再将学习完成的权重迁移到具有相同结构的SNN。其中,Diehl等人[15]实现了快速ANN-to-SNN转换,采用参数归一化在MNIST数据集上实现了几乎无损的转换。然而,该方法实质上是利用SNN逼近ANN,损失了SNN的时间特性。

    (3)直接监督学习。近年来,基于反向传播(Back Propagation, BP)的SNN监督学习在提高网络性能方面取得了较大进展。Neftci等人[16]提出使用替代梯度学习训练SNN:在正向传播中,使用Heaviside阶跃函数产生脉冲;在反向传播中,采用替代梯度近似不可微函数的梯度(例如s型函数)。这项工作表明SNN能够在通用深度学习框架中通过时间的反向传播进行学习。Wu等人[17]提出了一种时空反向传播(Spatio-Temporal BackPropagation, STBP)学习算法,同时为了解决SNN中不可微分的问题,提出了适合于梯度下降训练的脉冲活动近似导数。基于该学习算法,Zheng等人[18]提出了阈值依赖批处理归一化(threshold-dependent Batch Normalization, tdBN)训练方法,用于直接训练高性能的深层SNN。其中,使用替代梯度学习规则,产生了基于Pytorch的SNN训练框架,例如SpikingJelly[19]。借助于自动微分和GPU加速,新的框架已经使得SNN获得了更深层的训练,在分类问题上取得了重要进展[19]

    SNN的基本计算单元是脉冲神经元,LIF神经元模型[20]目前是SNN中最常见的脉冲神经元之一。LIF神经元可以等效为一个阻容(RC)电路,如图1所示,电路内部并联了一个膜电容Cm和膜电阻Rm。没有电流输入时,膜电位V(t)处于静息电压Vreset。如果突触前神经元给突触后神经元传来脉冲,则在连接的突触上产生相应的电流I(t)。输入电流I(t)一部分给Cm充电,此时V(t)开始累积电压;另一部分则由Rm进行电子转移,即泄漏电流。当Cm两端的V(t)超过神经元阈值电压Vth时,神经元会触发一个脉冲传给通过突触连接的下一个神经元,随后Cm产生放电现象,V(t)复位到静息电压。

    图 1  LIF神经元模型等效电路

    在LIF神经元中引入膜时间参数τm=RCm,膜电位V(t)的变化过程可由式(1)表示

    τmdV(t)dt=(V(t)Vreset)+RmI(t)
    (1)

    X(t)=RmI(t)作为外界对神经元的输入,将式(1)改写为

    τmdV(t)dt=(V(t)Vreset)+X(t)
    (2)

    将LIF神经元模型的连续时间离散化到一个时间步长中,可以用式(3)—式(5)分别描述神经元的充电、放电、复位动力学过程

    Ht=f(Vt1,Xt)=Vt1+1τm((Vt1Vreset)+Xt)
    (3)
    St=Θ(HtVth)
    (4)
    Vt=Ht(1St)+VresetSt
    (5)

    其中,Htt时刻神经元充电后的膜电位,Vt为触发脉冲后神经元的膜电位,Xt为外部对神经元的输入,Vth为发射脉冲阈值,τm表示膜时间参数。Θ(x)是阶跃函数,St表示t时刻的输出脉冲,如果有脉冲产生则等于1,否则为0。

    SSD结合了YOLO和Faster-RCNN的优点[21],是近年来广泛使用的one-stage算法之一,在检测精度与速度上都具有一定优势。一方面,SSD一般由一个主干特征提取器和多个预测头组成,它采用多尺度方法提取特征,根据与anchor相似的不同尺度先验框进行初步检测,以提高检测精度。另一方面,SSD采用端到端的训练,借鉴回归模式完成目标的检测与定位,可在保证精度的同时,更快速地完成实时检测[22]

    SSD的网络结构包括主干网络、额外层(extra layer)、预测头3个部分。其中,主干网络是一个改进的VGG16[23]网络,利用卷积层替换VGG16最后两个全连接层及输出层。额外层是在基础主干网络之后添加的4个卷积层,以生成更多不同尺度的特征,提高网络的目标特征提取能力。此外,SSD网络还采用了6个卷积特征图,包括Conv4_3(38, 38), Conv7(19, 19), Conv8_2(10, 10), Conv9_2(5, 5), Conv10_2(3, 3)和Conv11_2(1, 1)。不同尺度特征图设置的先验框数目不同,分别为30, 60, 111, 162, 213, 264。

    SSD算法流程主要分3部分。首先,将图片输入到预训练好的网络中获得不同大小的特征映射。然后,根据上述6个卷积特征图,从主干网络和额外层中提取特征并构建不同尺度的检测分支,输入到检测层中进行边框回归和分类。最后,采用非最大值抑制(Non-Maximum Suppression, NMS)消除高重叠预测,从而获得最优解。

    基于DT-LIF神经元与SSD的目标检测算法的结构主要由主干网络、3个额外层和SSD预测头等构成。其中,主干网络通过将SSD算法的CNN替换为Spiking VGG/Spiking DenseNet形成;3个额外层均由BN层、1×1 SC层、3×3 SC层和所设计的DT-LIF神经元构成(1×1 SC层用以减少通道数量,3×3 SC层的步长为2);SSD头部主要利用所接收3个额外层输出的特征图进行预测。图2为基于DT-LIF神经元与SSD的目标检测算法的结构。

    图 2  基于DT-LIF神经元与SSD的目标检测算法的结构

    已有研究表明生物神经元具有可变的阈值[24]。Azouz等人[24]发现,体内皮质神经元的电压阈值不是固定的,脉冲阈值与脉冲发生前的膜去极化速率成反比。除此之外,Fontaine等人[25]验证阈值可变性是神经元的内在特征之一。然而,目前常规LIF神经元模型一般采用固定阈值,需要积累足够的电压才能达到阈值并触发脉冲,从而限制了脉冲神经元的放电速率。当网络层次加深时,处于深层的神经元将很难被驱动,导致收敛速度下降。

    针对上述问题,本文设计了一种DT-LIF神经元模型(图3),以增加脉冲活动,加快收敛速度。

    图 3  DT-LIF神经元模型示意图

    首先,脉冲阈值与膜电位变化率之间的反比例关系的连续时间形式可以用式(6)表示

    θ=p+qeψt
    (6)

    其中,θ为脉冲阈值,ψ=dVm/dt为膜去极化速率,Vm为膜电位。同时,由于模拟SNN是一个离散的过程,因此对膜去极化速率的连续时间描述可离散化为

    ψ=ΔVmΔt=HtVt1Δt
    (7)

    其中, ΔVm为膜电位变化量,相当于式(3)中的1/τm((Vt1Vreset)+Xt)

    将式(7)代入式(6)中,可得到动态脉冲阈值θt的离散化描述,如式(8)所示

    θt=p+qe1τm((Vt1Vreset)+Xt)c
    (8)

    其中,p∈[0,1]是电压门控阈值,p, q满足p+q=1,c为常数。需要说明的是,为了避免过饱和发射脉冲,应满足θtp

    式(8)描述了DT-LIF神经元模型的动态阈值,将其与式(4)中的固定阈值进行比较可以发现,DT-LIF神经元的动态阈值更接近生物神经元内在特征。因此,将式(4)中的Vth替换为式(8)中的θt,以实现DT-LIF神经元模型的脉冲产生机制(如图3所示)

    St=Θ(Htθt)
    (9)

    从式(8)可以看出,脉冲阈值与膜电位变化量ΔVm成反比。如果ΔVm为负,则增大阈值以防止神经元放电,这与已有研究中膜超极化会降低去极化电位的上升速率[24]的结论一致;如果ΔVm为正,则降低阈值以增加神经元激发脉冲的数量,这与已有研究中膜去极化会导致较低的脉冲阈值的结论一致[24]。同时,当输入Xt=Vt–1Vreset时,式(8)中的θt等于式(4)中的Vth,式(9)等价于式(4)。在这种情况下,DT-LIF模型退化为LIF模型。

    算法1给出了DT-LIF神经元发射脉冲的动力学过程。

    算法1 DT-LIF发射脉冲过程
     参数:θ, p, q, Vth, τm
     (1) θ = Vth = 1; V = 0; Vreset = 0 // 初始化
     (2) for t = 1 to timesteps do
     (3)  for l = 2 to L do
     (4)   for i = 1 to neurons do
     (5)    Hli,t = Vli,t1 + (Xli,t – (Vli,t1Vreset)) * tau // Xli,t
          是正向传递的输入
     (6)    delta = Hli,tVli,t1
     (7)    θli,t = p + q exp (–delta / c)
     (8)    if Hli,tθli,t then
     (9)     Sli,t = 1
     (10)     Vli,t = Vreset
     (11)    end for
     (12)   end for
     (13) end for
    下载: 导出CSV 
    | 显示表格

    本文分别构建了基于VGG, DenseNet的Spiking VGG, Spiking DenseNet,作为所设计的目标检测算法结构的主干网络。同时,设计了一种脉冲分类器作为主干网络的最终层,并先将空间域的所有输出脉冲累加,然后在时间域上得到预测结果。

    网络中所使用的神经元基元均为本文所设计的DT-LIF神经元,并采用脉冲卷积(Spiking Convolution, SC)[26]、最大池化、批处理归一化(Batch Normalization, BN)操作。其中,BN层的参数可以在推理时与前后的卷积层参数融合,使得无论是否接收到脉冲,通过脉冲卷积学习到的权值会通过反向传播来更新,以加快收敛速度,本文将其放置在网络中DT-LIF神经元之前。此外,本文设计的脉冲分类器替代了VGG, DenseNet主干网络中用于分类的最终层,以实现与SNN的兼容性。脉冲分类器由BN层、DT-LIF神经元、输出1×1的num_classes通道组成。其中,BN层将脉冲的高度稀疏特征映射转换为密集的十进制表示。1×1卷积输出可以保证该分类器能处理所有大小的特征图,而无需使用平均池化层。同时,DT-LIF神经元可以取代ReLU激活功能。

    本文构建的两种主干网络具体描述如下:

    (1)Spiking VGG:VGG[23]是一个卷积神经网络。本文利用脉冲分类器取代VGG最后的分类层,其余架构与原网络保持一致,图4为Spiking VGG网络结构图(以VGG11为例)。该网络包含5个Spiking块与1个脉冲分类器,每个Spiking块均由BN层、SC层、DT-LIF神经元组成,SC层全部使用3×3的卷积核。每一块的SC层特征图大小分别为:SC1(224, 224), SC2(112, 112), SC3(56, 56), SC4(28, 28), SC5(14, 14)。

    图 4  Spiking VGG网络结构图(以VGG11为例)

    (2)Spiking DenseNet:为了促进训练中梯度的反向传播,ResNet[27]模型建立了前后层之间的“短路连接”,连接机制为元素级相加。DenseNet[28]的连接机制相较ResNet更密集,是一种通过使用通道级连接来促进梯度传播的架构。Spiking DenseNet的网络结构主要由Spiking Dense块和Transition层组成。Spiking Dense块中每层的特征图大小一致,每个Spiking Dense块均包含数个由[1×1 SC, 3×3 SC]组合的Bottleneck,每个SC层前均添加了BN层与DT-LIF神经元。Transition层主要是连接两个相邻的Spiking Dense块,并降低特征图尺寸,结构为BN层、1×1 SC层、DT-LIF神经元和最大池化层的组合。图5为Spiking DenseNet网络结构图(以DenseNet121为例),该网络包含4个Spiking Dense块,每一块的特征图大小分别为:(56, 56), (28, 28), (14, 14), (7, 7)。

    图 5  Spiking DenseNet网络结构图(以DenseNet121为例)

    本文的实验在CentOS系统下进行,GPU为NVIDIA RTX 2080,软件环境为基于PyTorch 1.11的SpikingJelly深度学习框架。训练时初始学习率设置为3e-4,batchsize设为8,在Prophesee GEN1数据集上训练了50个epoch,使用余弦退火的学习率下降方式(cosine annealing learning rate scheduler)将学习率逐渐降低到0。实验中所有SC操作均采用Kaiming统一方法初始化,所有BN层的初始化权重为1,偏差为0。DT-LIF神经元的膜时间参数τm均为2,固定阈值Vth为1,p=q=0.5, c=20,采用atan作为替代函数。

    本文在Prophesee GEN1检测数据集[29]上评估了所提出的脉冲神经网络目标检测方法。该数据集由一个空间分辨率为304×240的Prophesee GEN1传感器记录,共有39 h的城市、高速公路等场景,并将其剪切为数个60 s录制文件,包含超过255000个手动标注的汽车、行人边界框。数据包含(x, y, t, p)信息,分别为事件的2维像素坐标、时间戳与极性,图6为部分典型数据集示例。本文将数据集按照3:1的比例分为训练集和验证集。

    图 6  Prophesee GEN1数据集示例

    本文采用平均精度均值(mean Average Precision, mAP)作为性能评价指标。mAP是所有类别P-R曲线下面积的平均值,即计算mAP前需先得到准确率(Precision, P)和召回率(Recall, R)。

    以预测框与真实框的交并比(Intersection of Union, IoU)的阈值作为判断是否检测成功的指标,若IoU大于0.5则认为是正确检测出的正样本,反之则认为是检测错误的负样本[30]。本文使用的具体指标为mAP(0.5:0.95),即IoU阈值等于0.5时求出的mAP与IoU设为0.95时求出的mAP之比。

    首先,为了评价本文设计的DT-LIF神经元的性能,分别以Spiking VGG11和Spiking DenseNet121为主干网络,进行了LIF神经元和DT-LIF神经元的对比实验。其次,为了验证本文提出的基于DT-LIF与SSD的目标检测算法的性能,开展了其与AsyNet[31]目标检测算法的对比实验。实验结果如表1所示。

    表 1  Prophesee GEN1数据集上的对比实验结果
    方法mAP(0.5:0.95)
    Spiking VGG11+LIF0.127
    Spiking VGG11+DT-LIF0.159
    Spiking DenseNet+LIF0.148
    Spiking DenseNet+DT-LIF0.165
    AsyNet[31]0.140
    下载: 导出CSV 
    | 显示表格

    表1所示,以Spiking VGG11为主干网络时,相对LIF神经元模型,采用DT-LIF神经元模型的目标检测算法mAP提高了25.2%。类似地,以Spiking DenseNet121为主干网络时,采用DT-LIF神经元模型的目标检测算法mAP相对提高了11.5%,表明所设计的DT-LIF神经元模型可有效提高目标检测网络的检测精度。

    同时,本文对比了训练时每一轮的损失(Loss),训练损失曲线图如图7所示。可以看出,以Spiking VGG11为主干网络时,第3轮起采用不同神经元模型的训练损失差距变得明显。整个训练过程中LIF神经元网络的损失一直呈波动趋势,而第36轮时DT-LIF神经元网络的训练损失已下降至0.893,之后保持稳定。类似地,以Spiking DenseNet121为主干网络时,第4轮起采用不同神经元模型的训练损失差距开始变得明显。第43轮时LIF神经元网络的损失下降至0.807,第41轮时DT-LIF神经元网络的训练损失已下降至0.767,之后均趋于平稳。上述结果表明,相较于基于LIF神经元的目标检测网络,由于DT-LIF神经元具有阈值可变性,基于DT-LIF神经元的网络训练损失更早收敛到稳定的数值,推理速度更快。

    图 7  训练损失(Loss)曲线图

    另一方面,由表1可以看出,相对AsyNet目标检测算法,本文提出的基于DT-LIF与SSD的目标检测算法mAP提高了17.9%,表明本文提出的算法具有更强的处理真实事件数据特征的能力,有效提高了目标检测精度。

    本文提出一种基于DT-LIF神经元与SSD框架的SNN目标检测方法。首先,设计了一种DT-LIF神经元模型。同时,以DT-LIF神经元为基元,构建了以Spiking VGG和Spiking DenseNet为主干(Backbone),包括3个额外层和SSD预测框头(Head)的混合SNN。在Prophesee GEN1检测数据集上的实验结果表明,相对于采用LIF神经元模型,所设计的DT-LIF神经元模型的mAP提高了25.2%,且损失收敛速度更快,与AsyNet算法对比,所提出的目标检测算法的mAP提高了17.9%,从而为解决目标检测任务中SNN存在的训练难度大、精度低等问题,实现基于SNN的高精度目标检测奠定了基础。

  • 图  1  LIF神经元模型等效电路

    图  2  基于DT-LIF神经元与SSD的目标检测算法的结构

    图  3  DT-LIF神经元模型示意图

    图  4  Spiking VGG网络结构图(以VGG11为例)

    图  5  Spiking DenseNet网络结构图(以DenseNet121为例)

    图  6  Prophesee GEN1数据集示例

    图  7  训练损失(Loss)曲线图

    算法1 DT-LIF发射脉冲过程
     参数:θ, p, q, Vth, τm
     (1) θ = Vth = 1; V = 0; Vreset = 0 // 初始化
     (2) for t = 1 to timesteps do
     (3)  for l = 2 to L do
     (4)   for i = 1 to neurons do
     (5)    Hli,t = Vli,t1 + (Xli,t – (Vli,t1Vreset)) * tau // Xli,t
          是正向传递的输入
     (6)    delta = Hli,tVli,t1
     (7)    θli,t = p + q exp (–delta / c)
     (8)    if Hli,tθli,t then
     (9)     Sli,t = 1
     (10)     Vli,t = Vreset
     (11)    end for
     (12)   end for
     (13) end for
    下载: 导出CSV

    表  1  Prophesee GEN1数据集上的对比实验结果

    方法mAP(0.5:0.95)
    Spiking VGG11+LIF0.127
    Spiking VGG11+DT-LIF0.159
    Spiking DenseNet+LIF0.148
    Spiking DenseNet+DT-LIF0.165
    AsyNet[31]0.140
    下载: 导出CSV
  • [1] GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]. 2014 IEEE Conference on Computer Vision and Pattern Recognition, Columbus, USA, 2014: 580–587.
    [2] HE Kaiming, ZHANG Xiangyu, REN Shaoqing, et al. Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37(9): 1904–1916. doi: 10.1109/TPAMI.2015.2389824
    [3] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: Unified, real-time object detection[C]. 2016 IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, USA, 2016: 779–788.
    [4] LIU Wei, ANGUELOV D, ERHAN D, et al. SSD: Single shot MultiBox detector[C]. The 14th European Conference on Computer Vision, Amsterdam, The Netherlands, 2016: 21–37.
    [5] TAN Mingxing and LE Q. EfficientNet: Rethinking model scaling for convolutional neural networks[C]. The 36th International Conference on Machine Learning, Long Beach, USA, 2019: 6105–6114.
    [6] GERSTNER W and KISTLER W M. Spiking Neuron Models: Single Neurons, Populations, Plasticity[M]. Cambridge: Cambridge University Press, 2002: 421–454.
    [7] KIM S, PARK S, NA B, et al. Spiking-YOLO: Spiking neural network for energy-efficient object detection[C]. The 34th AAAI Conference on Artificial Intelligence, New York, USA, 2020: 11270–11277.
    [8] CHAKRABORTY B, SHE Xueyuan, and MUKHOPADHYAY S. A fully spiking hybrid neural network for energy-efficient object detection[J]. IEEE Transactions on Image Processing, 2021, 30: 9014–9029. doi: 10.1109/TIP.2021.3122092
    [9] LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection[C]. 2017 IEEE International Conference on Computer Vision, Venice, Italy, 2017: 2999–3007.
    [10] KUGELE A, PFEIL T, PFEIFFER M, et al. Hybrid SNN-ANN: Energy-efficient classification and object detection for event-based vision[C]. 43rd DAGM German Conference on Pattern Recognition, Bonn, Germany, 2022: 297–312.
    [11] 胡一凡, 李国齐, 吴郁杰, 等. 脉冲神经网络研究进展综述[J]. 控制与决策, 2021, 36(1): 1–26. doi: 10.13195/j.kzyjc.2020.1006

    HU Yifan, LI Guoqi, WU Yujie, et al. Spiking neural networks: A survey on recent advances and new directions[J]. Control and Decision, 2021, 36(1): 1–26. doi: 10.13195/j.kzyjc.2020.1006
    [12] TOYOIZUMI T, PFISTER J P, AIHARA K, et al. Spike-timing dependent plasticity and mutual information maximization for a spiking neuron model[C]. The 17th International Conference on Neural Information Processing Systems, Vancouver, Canada, 2004: 1409–1416.
    [13] HEBB D O. The Organization of Behavior: A Neuropsychological Theory[M]. New York: Psychology Press, 2002.
    [14] KHERADPISHEH S R, GANJTABESH M, THORPE S J, et al. STDP-based spiking deep convolutional neural networks for object recognition[J]. Neural Networks, 2018, 99: 56–67. doi: 10.1016/j.neunet.2017.12.005
    [15] DIEHL P U, NEIL D, BINAS J, et al. Fast-classifying, high-accuracy spiking deep networks through weight and threshold balancing[C]. 2015 International Joint Conference on Neural Networks (IJCNN), Killarney, Ireland, 2015: 1–8.
    [16] NEFTCI E O, MOSTAFA H, and ZENKE F. Surrogate gradient learning in spiking neural networks: Bringing the power of gradient-based optimization to spiking neural networks[J]. IEEE Signal Processing Magazine, 2019, 36(6): 51–63. doi: 10.1109/msp.2019.2931595
    [17] WU Yujie, DENG Lei, LI Guoqi, et al. Spatio-temporal backpropagation for training high-performance spiking neural networks[J]. Frontiers in Neuroscience, 2018, 12: 331. doi: 10.3389/fnins.2018.00331
    [18] ZHENG Hanle, WU Yujie, DENG Lei, et al. Going deeper with directly-trained larger spiking neural networks[C]. The 35th AAAI Conference on Artificial Intelligence, Palo Alto, USA, 2021: 11062–11070.
    [19] FANG Wei, YU Zhaofei, CHEN Yanqi, et al. Incorporating learnable membrane time constant to enhance learning of spiking neural networks[C]. 2021 IEEE/CVF International Conference on Computer Vision, Montreal, Canada, 2021: 2641–2651.
    [20] GERSTNER W, KISTLER W M, NAUD R, et al. Neuronal Dynamics: From Single Neurons to Networks and Models of Cognition[M]. Cambridge: Cambridge University Press, 2014.
    [21] 贺丰收, 何友, 刘准钆, 等. 卷积神经网络在雷达自动目标识别中的研究进展[J]. 电子与信息学报, 2020, 42(1): 119–131. doi: 10.11999/JEIT180899

    HE Fengshou, HE You, LIU Zhunga, et al. Research and development on applications of convolutional neural networks of radar automatic target recognition[J]. Journal of Electronics &Information Technology, 2020, 42(1): 119–131. doi: 10.11999/JEIT180899
    [22] 董小伟, 韩悦, 张正, 等. 基于多尺度加权特征融合网络的地铁行人目标检测算法[J]. 电子与信息学报, 2021, 43(7): 2113–2120. doi: 10.11999/JEIT200450

    DONG Xiaowei, HAN Yue, ZHANG Zheng, et al. Metro pedestrian detection algorithm based on multi-scale weighted feature fusion network[J]. Journal of Electronics &Information Technology, 2021, 43(7): 2113–2120. doi: 10.11999/JEIT200450
    [23] SIMONYAN K and ZISSERMAN A. Very deep convolutional networks for large-scale image recognition[C]. 3rd International Conference on Learning Representations, San Diego, USA, 2015.
    [24] AZOUZ R and GRAY C M. Dynamic spike threshold reveals a mechanism for synaptic coincidence detection in cortical neurons in vivo[J]. Proceedings of the National Academy of Sciences of the United States of America, 2000, 97(14): 8110–8115. doi: 10.1073/PNAS.130200797
    [25] FONTAINE B, PEÑA J L, and BRETTE R. Spike-threshold adaptation predicted by membrane potential dynamics in vivo[J]. PLoS Computational Biology, 2014, 10(4): e1003560. doi: 10.1371/journal.PCBI.1003560
    [26] XIAO Rong, TANG Huajin, MA Yuhao, et al. An event-driven categorization model for AER image sensors using multispike encoding and learning[J]. IEEE Transactions on Neural Networks and Learning Systems, 2020, 31(9): 3649–3657. doi: 10.1109/tnnls.2019.2945630
    [27] FANG Wei, YU Zhaofei, CHEN Yanqi, et al. Deep residual learning in spiking neural networks[C/OL]. The 34th International Conference on Neural Information Processing Systems, 2021: 21056–21069.
    [28] HUANG Gao, LIU Zhuang, VAN DER MAATEN L, et al. Densely connected convolutional networks[C]. 2017 IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, USA, 2017: 2261–2269.
    [29] DE TOURNEMIRE P, NITTI D, PEROT E, et al. A large scale event-based detection dataset for automotive[EB/OL]. https://doi.org/10.48550/arXiv.2001.08499, 2020.
    [30] 张德祥, 王俊, 袁培成. 基于注意力机制的多尺度全场景监控目标检测方法[J]. 电子与信息学报, 2022, 44(9): 3249–3257. doi: 10.11999/JEIT210664

    ZHANG Dexiang, WANG Jun, and YUAN Peicheng. Object detection method for multi-scale full-scene surveillance based on attention mechanism[J]. Journal of Electronics &Information Technology, 2022, 44(9): 3249–3257. doi: 10.11999/JEIT210664
    [31] MESSIKOMMER N, GEHRIG D, LOQUERCIO A, et al. Event-based asynchronous sparse convolutional networks[C]. 16th European Conference on Computer Vision, Glasgow, UK, 2020: 415–431.
  • 期刊类型引用(15)

    1. 丁嘉伟. 面向簇化移动机器人的网络资源调度算法. 机床与液压. 2024(11): 47-52 . 百度学术
    2. 曾琦,刘友波,李梓玮,马超,钟俊,刘星. 支持新型电力系统的5G/6G高可靠低时延电力物联网通信设计与分析. 中国电机工程学报. 2024(21): 8435-8444 . 百度学术
    3. 龚万炜. 可靠性约束下的5G通信频段自适应选择技术. 电子技术与软件工程. 2023(04): 29-32 . 百度学术
    4. 齐放,张成,柯旺松,胡晨. 基于风险均衡度的电力5G通信链路自动选择方法研究. 电子技术应用. 2023(04): 1-6 . 百度学术
    5. 李松,张继钰,陈瑞瑞. 面向工业物联网的短包安全通信资源智能分配算法. 南京邮电大学学报(自然科学版). 2023(02): 11-17 . 百度学术
    6. 梁建. 5G URLLC应用场景中低时延、高可靠关键技术分析. 数字通信世界. 2023(07): 89-91 . 百度学术
    7. 刘峻朋,夏玮玮,刘晗,修成林,燕锋,沈连丰. 面向电力业务质量保障的NR-U与Wi-Fi频谱共享. 电信科学. 2023(07): 11-22 . 百度学术
    8. 李欣,程松,智远,王世杰,刘扬. 基于N-k故障的电网数据通信业务保障分级方法. 自动化与仪器仪表. 2023(10): 163-166 . 百度学术
    9. 覃凤,任宝平,雷久淮. 电子通信低时延IPSEC安全加密传输仿真. 计算机仿真. 2023(09): 171-175 . 百度学术
    10. 李想. 基于深度学习的网络链路通信质量检测方法. 数字通信世界. 2022(05): 134-136 . 百度学术
    11. 王海洋,王玉东,刘晗,孙海蓬,李正浩,刘明刚,苏俊浩. 电力通信网与运营商5G网络融合方案. 移动通信. 2022(08): 106-112 . 百度学术
    12. 李世党,魏明生,赵娟,刘加跃,唐守锋. 蜂窝物联网中短包域能量效率最大化波束成形优化与设计. 电子与信息学报. 2022(09): 3075-3082 . 本站查看
    13. 王鑫,王霖,余芸,敖知琪,孙凌云. 数字孪生电网的特性、架构及应用综述. 电子与信息学报. 2022(11): 3721-3733 . 本站查看
    14. 张成娟. 5G通信中传输技术的应用. 微型电脑应用. 2022(11): 182-185 . 百度学术
    15. 张凯楠,冯瑛敏,黄丽妍,刘瑾,赵晶,任国岐. 基于GMM聚类的通信站点资源自动化研判分析. 电力大数据. 2022(06): 62-68 . 百度学术

    其他类型引用(9)

  • 加载中
图(7) / 表(2)
计量
  • 文章访问数:  1104
  • HTML全文浏览量:  515
  • PDF下载量:  201
  • 被引次数: 24
出版历程
  • 收稿日期:  2022-11-01
  • 修回日期:  2023-05-11
  • 网络出版日期:  2023-05-20
  • 刊出日期:  2023-08-21

目录

/

返回文章
返回