Optimization Techniques for Conversion of Quantization Aware Trained Deep Neural Networks to Lightweight Spiking Neural Networks

被引:0
作者
Lee, Kyungchul [1 ]
Choi, Sunghyun [1 ]
Lew, Dongwoo [1 ]
Park, Jongsun [1 ]
机构
[1] Korea Univ, Seoul, South Korea
来源
2021 36TH INTERNATIONAL TECHNICAL CONFERENCE ON CIRCUITS/SYSTEMS, COMPUTERS AND COMMUNICATIONS (ITC-CSCC) | 2021年
关键词
spiking neural networks; ANN-SNN conversion; quantization aware training;
D O I
10.1109/ITC-CSCC52171.2021.9501427
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this paper, we present spiking neural network (SNN) conversion technique optimized for converting low bit-width artificial neural networks (ANN) trained with quantization aware training (QAT). Conventional conversion technique suffers significant accuracy drop on QAT ANNs due to different activation function used for QAT ANNs. To minimize such accuracy drop of the conventional conversion, the proposed technique uses Spike-Norm Skip, which selectively applies threshold balancing. In addition, subtraction based reset is used to further reduce accuracy degradation. The proposed conversion technique achieves an accuracy of 89.92% (0.68% drop) with a 5-bit weight on CIFAR-10 using VGG-16.
引用
收藏
页数:3
相关论文
共 5 条
  • [1] Deep Learning With Edge Computing: A Review
    Chen, Jiasi
    Ran, Xukan
    [J]. PROCEEDINGS OF THE IEEE, 2019, 107 (08) : 1655 - 1674
  • [2] Diehl PU, 2015, IEEE IJCNN
  • [3] Diehl PU, 2016, IEEE IJCNN, P4278, DOI 10.1109/IJCNN.2016.7727758
  • [4] Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference
    Jacob, Benoit
    Kligys, Skirmantas
    Chen, Bo
    Zhu, Menglong
    Tang, Matthew
    Howard, Andrew
    Adam, Hartwig
    Kalenichenko, Dmitry
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 2704 - 2713
  • [5] Going Deeper in Spiking Neural Networks: VGG and Residual Architectures
    Sengupta, Abhronil
    Ye, Yuting
    Wang, Robert
    Liu, Chiao
    Roy, Kaushik
    [J]. FRONTIERS IN NEUROSCIENCE, 2019, 13