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

基于脉冲活动剪枝的高效深度脉冲神经网络构建

Towards Efficient Deep Spiking Neural Networks Construction with Spiking Activity based Pruning

School of Computer Science and Technology, Dalian University of Technology(大连理工大学计算机科学与技术学院);Guangdong Laboratory of Artificial Intelligence and Digital Economy(广东省人工智能与数字经济实验室);College of Computer Science and Technology, Zhejiang University(浙江大学计算机科学与技术学院);Centre for Systems Engineering and Innovation, Imperial College London(伦敦帝国理工学院系统工程与创新中心)

Yaxin Li, Qi Xu, Jiangrong Shen, Hongming Xu, Long Chen, Gang Pan

AI总结:

针对深度脉冲神经网络冗余结构导致的压缩需求,提出基于卷积核活动水平的结构化剪枝框架SCA,通过训练中动态剪枝与再生优化网络架构,在保持性能的同时降低计算负载并加速推理。

摘要:

在多种复杂数据集上表现出高性能的深度大规模脉冲神经网络(SNNs)的出现,由于存在大量冗余结构单元,导致需要压缩网络模型,旨在更有效地利用其低功耗和生物可解释性优势。目前,SNNs的大多数模型压缩技术基于对单个连接的非结构化剪枝,这需要特定的硬件支持。因此,我们提出了一种基于卷积核活动水平的结构化剪枝方法,命名为脉冲通道活动基(SCA)网络剪枝框架。受突触可塑性机制启发,我们的方法通过在训练期间剪枝和再生卷积核来动态调整网络结构,增强模型对当前目标任务的适应性。在保持模型性能的同时,该方法优化了网络架构,最终减少了计算负载并加速了推理过程。这表明结构化动态稀疏学习方法能更好地促进深度SNNs在低功耗和高效率场景中的应用。

英文摘要:

The emergence of deep and large-scale spiking neural networks (SNNs) exhibiting high performance across diverse complex datasets has led to a need for compressing network models due to the presence of a significant number of redundant structural units, aiming to more effectively leverage their low-power consumption and biological interpretability advantages. Currently, most model compression techniques for SNNs are based on unstructured pruning of individual connections, which requires specific hardware support. Hence, we propose a structured pruning approach based on the activity levels of convolutional kernels named Spiking Channel Activity-based (SCA) network pruning framework. Inspired by synaptic plasticity mechanisms, our method dynamically adjusts the network's structure by pruning and regenerating convolutional kernels during training, enhancing the model's adaptation to the current target task. While maintaining model performance, this approach refines the network architecture, ultimately reducing computational load and accelerating the inference process. This indicates that structured dynamic sparse learning methods can better facilitate the application of deep SNNs in low-power and high-efficiency scenarios.

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

↑