Towards Lossless ANN-SNN Conversion under Ultra-Low Latency with Dual-Phase Optimization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Ziming, Lian, Shuang, Zhang, Yuhao, Cui, Xiaoxin, Yan, Rui, Tang, Huajin
Format: Preprint
Publié: 2022
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917617638309888
author Wang, Ziming
Lian, Shuang
Zhang, Yuhao
Cui, Xiaoxin
Yan, Rui
Tang, Huajin
author_facet Wang, Ziming
Lian, Shuang
Zhang, Yuhao
Cui, Xiaoxin
Yan, Rui
Tang, Huajin
contents Spiking neural networks (SNNs) operating with asynchronous discrete events show higher energy efficiency with sparse computation. A popular approach for implementing deep SNNs is ANN-SNN conversion combining both efficient training of ANNs and efficient inference of SNNs. However, the accuracy loss is usually non-negligible, especially under a few time steps, which restricts the applications of SNN on latency-sensitive edge devices greatly. In this paper, we first identify that such performance degradation stems from the misrepresentation of the negative or overflow residual membrane potential in SNNs. Inspired by this, we decompose the conversion error into three parts: quantization error, clipping error, and residual membrane potential representation error. With such insights, we propose a two-stage conversion algorithm to minimize those errors respectively. Besides, We show each stage achieves significant performance gains in a complementary manner. By evaluating on challenging datasets including CIFAR-10, CIFAR- 100 and ImageNet, the proposed method demonstrates the state-of-the-art performance in terms of accuracy, latency and energy preservation. Furthermore, our method is evaluated using a more challenging object detection task, revealing notable gains in regression performance under ultra-low latency when compared to existing spike-based detection algorithms. Codes are available at https://github.com/Windere/snn-cvt-dual-phase.
format Preprint
id arxiv_https___arxiv_org_abs_2205_07473
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Towards Lossless ANN-SNN Conversion under Ultra-Low Latency with Dual-Phase Optimization
Wang, Ziming
Lian, Shuang
Zhang, Yuhao
Cui, Xiaoxin
Yan, Rui
Tang, Huajin
Neural and Evolutionary Computing
Machine Learning
Spiking neural networks (SNNs) operating with asynchronous discrete events show higher energy efficiency with sparse computation. A popular approach for implementing deep SNNs is ANN-SNN conversion combining both efficient training of ANNs and efficient inference of SNNs. However, the accuracy loss is usually non-negligible, especially under a few time steps, which restricts the applications of SNN on latency-sensitive edge devices greatly. In this paper, we first identify that such performance degradation stems from the misrepresentation of the negative or overflow residual membrane potential in SNNs. Inspired by this, we decompose the conversion error into three parts: quantization error, clipping error, and residual membrane potential representation error. With such insights, we propose a two-stage conversion algorithm to minimize those errors respectively. Besides, We show each stage achieves significant performance gains in a complementary manner. By evaluating on challenging datasets including CIFAR-10, CIFAR- 100 and ImageNet, the proposed method demonstrates the state-of-the-art performance in terms of accuracy, latency and energy preservation. Furthermore, our method is evaluated using a more challenging object detection task, revealing notable gains in regression performance under ultra-low latency when compared to existing spike-based detection algorithms. Codes are available at https://github.com/Windere/snn-cvt-dual-phase.
title Towards Lossless ANN-SNN Conversion under Ultra-Low Latency with Dual-Phase Optimization
topic Neural and Evolutionary Computing
Machine Learning
url https://arxiv.org/abs/2205.07473