4 个月前

卷积脉冲神经网络用于时空特征提取

卷积脉冲神经网络用于时空特征提取

摘要

脉冲神经网络(SNNs)由于其基于事件的特性,可以在低功耗和嵌入式系统(如新兴的神经形态芯片)中使用。此外,与传统的前馈人工神经网络(ANNs)相比,它们具有较低的计算成本,同时保留了ANNs的性质。然而,卷积脉冲神经网络及其他类型SNNs中的时间编码尚未得到充分研究。在本文中,我们通过设计实验来利用这一特性,探讨了卷积SNNs的空间-时间特征提取机制。浅层卷积SNN在性能上超过了当前最先进的空间-时间特征提取方法,如C3D、ConvLSTM及类似网络。此外,我们提出了一种新的深层脉冲架构来解决实际问题(特别是分类任务),该架构在NMNIST(99.6%)、DVS-CIFAR10(69.2%)和DVS-Gesture(96.7%)数据集上的表现优于其他SNN方法,在UCF-101(42.1%)和HMDB-51(21.5%)数据集上的表现也优于ANN方法。值得注意的是,训练过程是基于本文所解释的空间-时间反向传播变体实现的。

代码仓库

aa-samad/conv_snn
官方
pytorch

基准测试

基准方法指标
event-data-classification-on-cifar10-dvs-1STS-ResNet
Accuracy: 69.2
gesture-recognition-on-dvs128-gestureSTS-ResNet
Accuracy (%): 96.7
image-classification-on-n-mnistSTS-ResNet
Accuracy: 99.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
卷积脉冲神经网络用于时空特征提取 | 论文 | HyperAI超神经