arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.06232cs.CV

尖峰点网络:用于点云的尖峰神经网络

Spiking PointNet: Spiking Neural Networks for Point Clouds

Intelligent Science & Technology Academy of CASIC(中国航天科工集团智能科技研究院);Scientific Research Laboratory of Aerospace Intelligent Systems and Technology(航天智能系统技术科研实验室)

Dayong Ren, Zhe Ma, Yuanpei Chen, Weihang Peng, Xiaode Liu, Yuhan Zhang, Yufei Guo

AI总结:

本文提出Spiking PointNet,首个用于点云的尖峰神经网络,通过单时间步训练多时间步推理的范式,解决了SNN优化和PointNet成本问题,在ModelNet10/40上超越ANN对应模型。

摘要:

近年来,尖峰神经网络(SNNs)凭借其极高的能效,在二维视觉识别领域引起了广泛的研究关注,并展现出逐渐增长的应用潜力。然而,SNNs能否推广到三维识别任务中,仍然是一个尚未充分探索的问题。为此,我们在本文中提出了Spiking PointNet,这是第一个用于点云高效深度学习的尖峰神经模型。我们发现,限制SNNs在点云中应用的两大障碍是:SNNs固有的优化障碍,阻碍了具有大时间步长的大型尖峰模型的训练;以及PointNet昂贵的存储和计算成本,使得训练大型尖峰点模型不切实际。为了同时解决这些问题,我们提出了一种“少训练、多学习”的Spiking PointNet范式,并提供了理论论证和深入的实验分析。具体而言,我们的Spiking PointNet仅使用单个时间步长进行训练,但在多个时间步长推理时可以获得更好的性能,优于直接使用多个时间步长训练的网络。我们在ModelNet10和ModelNet40上进行了各种实验,以证明Spiking PointNet的有效性。值得注意的是,我们的Spiking PointNet甚至能超越其ANN对应模型,这在SNN领域是罕见的,从而为后续工作提供了一个潜在的研究方向。此外,Spiking PointNet在训练阶段展现了显著的加速和存储节省。

英文摘要:

Recently, Spiking Neural Networks (SNNs), enjoying extreme energy efficiency, have drawn much research attention on 2D visual recognition and shown gradually increasing application potential. However, it still remains underexplored whether SNNs can be generalized to 3D recognition. To this end, we present Spiking PointNet in the paper, the first spiking neural model for efficient deep learning on point clouds. We discover that the two huge obstacles limiting the application of SNNs in point clouds are: the intrinsic optimization obstacle of SNNs that impedes the training of a big spiking model with large time steps, and the expensive memory and computation cost of PointNet that makes training a big spiking point model unrealistic. To solve the problems simultaneously, we present a trained-less but learning-more paradigm for Spiking PointNet with theoretical justifications and in-depth experimental analysis. In specific, our Spiking PointNet is trained with only a single time step but can obtain better performance with multiple time steps inference, compared to the one trained directly with multiple time steps. We conduct various experiments on ModelNet10, ModelNet40 to demonstrate the effectiveness of Spiking PointNet. Notably, our Spiking PointNet even can outperform its ANN counterpart, which is rare in the SNN field thus providing a potential research direction for the following work. Moreover, Spiking PointNet shows impressive speedup and storage saving in the training phase.

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

↑