Accelerating Spiking Neural Networks Using Quantum Algorithm with High Success Probability and High Calculation Accuracy
Yanhu Chen,Cen Wang,Hongxiang Guo,Xiong Gao,Jian Wu
DOI: https://doi.org/10.1016/j.neucom.2022.02.004
IF: 6
2022-01-01
Neurocomputing
Abstract:Spiking neural networks (SNNs) are a kind of neuromorphic computing which meticulously imitates the operations of biological nervous systems. Thus, the SNN is seen as a promising approach to further improve the level of intelligence that the legacy artificial neural networks achieved. In a spiking neuron of an SNN, it is a key step with the highest computing complexity to find out the moment when the out-put stimuli occur. Classic computer (i.e., electrical computer) based acceleration cannot reduce the com-putational complexity of the operations in the spiking neurons. As a result, the SNN suffers from scaling and deeper emulation problems. Considering these, in this paper, we propose a quantum algorithm to reduce the complexity of the key steps in the SNN and use this algorithm to build a quantum spiking neu-ron network (QSNN). More specifically, first, we give mathematical proof that the problem of finding the output stimuli is approximately equal to the problem of calculating the unsigned vector inner products, which can transfer to a quantum operation. Second, we design a scalable quantum circuit of QSNN for data of any dimension and evaluate its basic success probability and calculation accuracy. Third, to improve the QSNN performance, we propose a method to improve the minimum success probability to 99.8%, by repeatedly performing the quantum circuit only 11 times. Fourth, we prove that the computa-tional complexity of the QSNN is a log-polynomial relationship with the data dimension, which is much lower than that of the linear complexity of the classic SNN. Finally, we apply the QSNN to solve classifi-cation tasks on two real-world datasets, i.e., MNIST and fashion MNIST which both have two noise levels (i.e., no noise and 50% of the noise). The experiment results show that in addition to the acceleration, QSNN and SNN have equal classification accuracy, which suggests the feasibility and robustness of the QSNN. (c) 2022 Elsevier B.V. All rights reserved.