arrow
返回

Comprehensive SNN Compression Using ADMM Optimization and Activity Regularization

delete2023-06-01
delete30
delete
OA
AI
邓
邓磊 (Lei Deng)
Y
Yujie Wu
Y
Yifan Hu
L
Ling Liang
Guoqi Li 封面图
Guoqi Li (Guoqi Li) *
X
Xing Hu *
Y
Yufei Ding
李
李朋 (Peng Li)
Y
Yuan Xie
DOI:10.1109/TNNLS.2021.3109064delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
As well known, the huge memory and compute costs of both artificial neural networks (ANNs) and spiking neural networks (SNNs) greatly hinder their deployment on edge devices with high efficiency. Model compression has been proposed as a promising technique to improve the running efficiency via parameter and operation reduction, whereas this technique is mainly practiced in ANNs rather than SNNs. It is interesting to answer how much an SNN model can be compressed without compromising its functionality, where two challenges should be addressed: 1) the accuracy of SNNs is usually sensitive to model compression, which requires an accurate compression methodology and 2) the computation of SNNs is event-driven rather than static, which produces an extra compression dimension on dynamic spikes. To this end, we realize a comprehensive SNN compression through three steps. First, we formulate the connection pruning and weight quantization as a constrained optimization problem. Second, we combine spatiotemporal backpropagation (STBP) and alternating direction method of multipliers (ADMMs) to solve the problem with minimum accuracy loss. Third, we further propose activity regularization to reduce the spike events for fewer active operations. These methods can be applied in either a single way for moderate compression or a joint way for aggressive compression. We define several quantitative metrics to evaluate the compression performance for SNNs. Our methodology is validated in pattern recognition tasks over MNIST, N-MNIST, CIFAR10, and CIFAR100 datasets, where extensive comparisons, analyses, and insights are provided. To the best of our knowledge, this is the first work that studies SNN compression in a comprehensive manner by exploiting all compressible components and achieves better results.
Keyword:
Neurons
Computational modeling
Quantization (signal)
Optimization
Encoding
Task analysis
Synapses
Activity regularization
alternating direction method of multiplier (ADMM)
connection pruning
spiking neural network (SNN) compression
weight quantization

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

T
tsinghua university
学者数:
11.9W
论文数: 10.0W
被引数: 137
U
University of California Santa Barbara
学者数:
1.2W
论文数: 9.6K
被引数: 3.6W
University of California System 封面图
University of California System
学者数:
37.7W
论文数: 33.8W
被引数: 6.6K
C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
学者 查看更多机构
引用论文

引用论文

err分享
err收藏
An epidemiologic study of insect allergy in children. I. Characteristics of the disease
err1982-04-01
err0
PREAI
errKenneth C. Schuberth; Lawrence M. Lichtenstein; Anne Kagey-Sobotka; Moyses Szklo; Kathleen A. Kwiterovich; Martin D. Valentine
err分享
err收藏
A Stochastic Spiking Neural Network for Virtual Screening一种用于虚拟筛选的随机脉冲神经网络
err2018-04-01
err40
PREAI
errMorro, A.; Canals, V.; Oliver, A.; Alomar, M. L.; Galan-Prado, F.; Ballester, P. J.; Rossello, J. L.
err分享
err收藏
err分享
err收藏
Crossbar-Aware Neural Network Pruning
err2018-01-01
err45
errOAAI
errLiang, Ling; Deng, Lei; Zeng, Yueling; Hu, Xing; Ji, Yu; Ma, Xin; Li, Guoqi; Xie, Yuan
err分享
err收藏
Baseline characteristics in the Aliskiren Trial in Type 2 Diabetes Using Cardio-Renal Endpoints (ALTITUDE)
err2012-02-14
err0
errOAAI
errHans-Henrik Parving; Barry M Brenner; John JV McMurray; Dick de Zeeuw; Steven M Haffner; Scott D Solomon; Nish Chaturvedi; Frederik Persson; Maria Nicolaides; Alexia Richard; Zhihua Xiang; Juergen Armbrecht; Marc A Pfeffer
err分享
err收藏
Compact Hardware Synthesis of Stochastic Spiking Neural Networks随机脉冲神经网络的紧凑硬件综合
err2019-09-25
err17
PREAI
errGalan-Prado, Fabio; Moran, Alejandro; Font, Joan; Roca, Miquel; Rossello, Josep L.
err分享
err收藏
Conversion of Continuous-Valued Deep Networks to Efficient Event-Driven Networks for Image Classification
err2017-12-07
err771
errOAAI
errRueckauer, Bodo; Lungu, Iulia-Alexandra; Hu, Yuhuang; Pfeiffer, Michael; Liu, Shih-Chii
err分享
err收藏
学者 查看更多内容