arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2109.13751cs.CVcs.AIcs.LGcs.NE

StereoSpike:使用脉冲神经网络进行深度学习

StereoSpike: Depth Learning with a Spiking Neural Network

CerCo CNRS UMR 5549(法国国家科学研究中心 认知与脑科学中心 联合研究单元5549);Université Toulouse III(图卢兹第三大学)

Ulysse Rançon, Javier Cuadrado-Anibarro, Benoit R. Cottereau, Timothée Masquelier

AI总结:

本文提出StereoSpike,一种结合事件相机与脉冲神经网络的端到端深度估计架构,通过新颖的读出范式实现高精度密集深度预测,并支持低功耗神经形态芯片部署。

摘要:

深度估计是一项重要的计算机视觉任务,特别适用于自动驾驶车辆中的导航或机器人中的物体操作。在这里,我们使用端到端的神经形态方法解决了这个问题,结合两个事件相机和一个稍作修改的类U-Net编码器-解码器架构的脉冲神经网络(SNN),我们将其命名为StereoSpike。更具体地说,我们使用了多车辆立体事件相机数据集(MVSEC)。它提供了深度真值,用于以监督方式使用替代梯度下降训练StereoSpike。我们提出了一种新颖的读出范式,从解码器的脉冲中获取密集的模拟预测——每个像素的深度。我们证明该架构泛化能力极强,甚至优于其非脉冲对应物,达到了最先进的测试精度。据我们所知,这是首次由完全脉冲网络解决如此大规模的回归问题。最后,我们展示了可以通过正则化获得低发放率(<10%),且精度损失极小。这意味着StereoSpike可以高效地在神经形态芯片上实现,为低功耗和实时嵌入式系统打开了大门。

英文摘要:

Depth estimation is an important computer vision task, useful in particular for navigation in autonomous vehicles, or for object manipulation in robotics. Here we solved it using an end-to-end neuromorphic approach, combining two event-based cameras and a Spiking Neural Network (SNN) with a slightly modified U-Net-like encoder-decoder architecture, that we named StereoSpike. More specifically, we used the Multi Vehicle Stereo Event Camera Dataset (MVSEC). It provides a depth ground-truth, which was used to train StereoSpike in a supervised manner, using surrogate gradient descent. We propose a novel readout paradigm to obtain a dense analog prediction -- the depth of each pixel -- from the spikes of the decoder. We demonstrate that this architecture generalizes very well, even better than its non-spiking counterparts, leading to state-of-the-art test accuracy. To the best of our knowledge, it is the first time that such a large-scale regression problem is solved by a fully spiking network. Finally, we show that low firing rates (<10%) can be obtained via regularization, with a minimal cost in accuracy. This means that StereoSpike could be efficiently implemented on neuromorphic chips, opening the door for low power and real time embedded systems.

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

↑