3 个月前

EventRPG:基于相关性传播引导的事件数据增强

EventRPG:基于相关性传播引导的事件数据增强

摘要

事件相机(event camera)作为一种新型仿生视觉传感器,因其低延迟、低功耗和高动态范围等优势,近年来受到广泛关注。然而,在基于事件的分类任务中,脉冲神经网络(Spiking Neural Network, SNN)仍面临过拟合这一关键问题,主要源于其相对较弱的空间表征能力。数据增强是一种简单而高效缓解过拟合、提升神经网络泛化能力的方法,且在图像处理领域中,基于显著性(saliency)的数据增强方法已被证明具有显著效果。然而,目前尚无有效方法能够从SNN中提取显著性图(saliency maps)。为此,本文首次提出脉冲层时间逐层重要性传播规则(Spiking Layer-Time-wise Relevance Propagation, SLTRP)与脉冲层逐层重要性传播规则(Spiking Layer-wise Relevance Propagation, SLRP),使SNN能够生成稳定且准确的类激活图(Class Activation Maps, CAMs)与显著性图。基于此,我们进一步提出EventRPG方法,通过在脉冲神经网络上引入重要性传播机制,实现更高效的增强策略。所提方法在多种SNN架构上进行了验证,在物体识别任务中取得了当前最优性能,分别在N-Caltech101和CIFAR10-DVS数据集上达到85.62%和85.55%的准确率;在动作识别任务SL-Animals上,准确率高达91.59%。相关代码已开源,地址为:https://github.com/myuansun/EventRPG。

代码仓库

myuansun/eventrpg
官方
pytorch

基准测试

基准方法指标
action-recognition-on-dvs128-gestureSEW-Resnet18
Accuracy (% ): 96.53
action-recognition-on-sl-animalsSEW-Resnet18 (3sets)
Accuracy (% ): 93.75
action-recognition-on-sl-animalsSEW-Resnet18 (4sets)
Accuracy (% ): 91.59
object-recognition-on-cifar10-dvsSpike-VGG11
Accuracy (% ): 85.55
object-recognition-on-n-caltech-101Spike-VGG11
Accuracy (% ): 85.62
object-recognition-on-n-carsSpike-VGG11
Accuracy (% ): 96.00

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
EventRPG:基于相关性传播引导的事件数据增强 | 论文 | HyperAI超神经