arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.13672cs.CV

推进脉冲神经网络迈向多尺度时空交互学习

Advancing Spiking Neural Networks towards Multiscale Spatiotemporal Interaction Learning

Liaoning Technical University(辽宁工程技术大学);University of Electronic Science and Technology of China(电子科技大学);University of California, Santa Cruz(加州大学圣克鲁兹分校);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所)

Yimeng Shan, Malu Zhang, Rui-jie Zhu, Xuerui Qiu, Jason K. Eshraghian, Haicheng Qu

AI总结:

本文提出脉冲多尺度注意力模块与Attention ZoneOut正则化方法,捕获事件数据多尺度时空交互信息,在神经形态数据集及ImageNet-1K上取得最先进性能,缩小了SNN与ANN的差距。

摘要:

神经科学研究的近期进展推动了脉冲神经网络(SNN)的发展,该类网络不仅具有进一步推进神经科学研究的潜力,而且由于其脉冲驱动特性,可作为人工神经网络(ANN)的节能替代方案。然而,以往的研究常常忽略了事件数据中的多尺度信息及其时空相关性,导致SNN模型将输入事件的每一帧近似为静态图像。我们假设这种过度简化是导致SNN与传统ANN之间性能差距的重要原因。为解决这一问题,我们设计了一个脉冲多尺度注意力(SMA)模块,该模块能够捕获多尺度时空交互信息。此外,我们开发了一种名为Attention ZoneOut(AZO)的正则化方法,该方法利用时空注意力权重,通过伪集成训练来降低模型的泛化误差。我们的方法在主流神经形态数据集上取得了最先进的结果。此外,我们在使用SMA和AZO增强的104层ResNet架构上,在Imagenet-1K数据集上达到了77.1%的性能。这一成就证实了采用非Transformer架构的SNN的最先进性能,并强调了我们的方法在弥合SNN模型与传统ANN模型之间性能差距方面的有效性。

英文摘要:

Recent advancements in neuroscience research have propelled the development of Spiking Neural Networks (SNNs), which not only have the potential to further advance neuroscience research but also serve as an energy-efficient alternative to Artificial Neural Networks (ANNs) due to their spike-driven characteristics. However, previous studies often neglected the multiscale information and its spatiotemporal correlation between event data, leading SNN models to approximate each frame of input events as static images. We hypothesize that this oversimplification significantly contributes to the performance gap between SNNs and traditional ANNs. To address this issue, we have designed a Spiking Multiscale Attention (SMA) module that captures multiscale spatiotemporal interaction information. Furthermore, we developed a regularization method named Attention ZoneOut (AZO), which utilizes spatiotemporal attention weights to reduce the model's generalization error through pseudo-ensemble training. Our approach has achieved state-of-the-art results on mainstream neural morphology datasets. Additionally, we have reached a performance of 77.1% on the Imagenet-1K dataset using a 104-layer ResNet architecture enhanced with SMA and AZO. This achievement confirms the state-of-the-art performance of SNNs with non-transformer architectures and underscores the effectiveness of our method in bridging the performance gap between SNN models and traditional ANN models.

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

↑