• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science

Previous Articles     Next Articles

Image classification algorithm based on
spiking neural network and mobile GPU computing

(1.中国科学院计算技术研究所,北京 100190;2.中国科学院大学,北京 100049;
3.中国矿业大学(北京) 机电与信息工程学院,北京 100083)
 
  

  1. (1.Institute of Computing Technology,Chinese Academy of Sciences,Beijing 100190;
    2.University of Chinese Academy of Sciences,Beijing 100049;
    3.School of Mechanical Electronic & Information Engineering,
    China University of Mining & Technology-Beijing,Beijing 100083,China)
  • Received:2019-09-03 Revised:2019-11-26 Online:2020-03-25 Published:2020-03-25

Abstract:

Computer vision is designed to simulate human visual systems through machines, which is a hot spot in artificial intelligence and neuroscience research. As a classical task of computer vision, image classification has attracted more and more researches, especially the image classification algorithms based on neural networks perform well on various classification tasks. However, the traditional shallow artificial neural networks have weak feature learning ability and insufficient bio-interpretability, while the deep neural networks have the disadvantages of over-fitting and high power consumption. Therefore, the bio-interpretable classification algorithm in low power environments is still a challenging task. In order to solve the above problems, a classification method based on spiking neural network in Jetson TK1 development environment is designed. The main innovations of the research are as follows:
(1) Designing a spiking convolution neural network for image classification;
(2) Implementing the improved spiking neural network based on CUDA and deploying it in Jetson TK1.
 
 

Key words: image classification, spiking neural network, mobile GPU computing