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

计算机工程与科学

• 高性能计算 • 上一篇    下一篇

基于脉冲神经网络与移动GPU计算的图像分类算法研究与实现

徐频捷1,2,王诲喆1,2,李策3,唐丹1,赵地1   

  1. XU Pin-jie1,2,WANG Hui-zhe1,2,LI Ce3,TANG Dan1,ZHAO Di1
  • 收稿日期:2019-09-03 修回日期:2019-11-26 出版日期:2020-03-25 发布日期:2020-03-25
  • 基金资助:
    国家自然科学基金(61420106013);国家重点研究发展计划(2018ZX10723203);北京市自然科学基金(4161004);北京市科技项目(Z161100000216143,Z171100000117001)
     

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

摘要:

 

计算机视觉旨在通过计算机模拟人的视觉系统,让计算机学会“看”,是人工智能、神经科学研究的一个热点。作为计算机视觉的经典任务,图像分类吸引了越来越多的研究,尤其是基于神经网络的算法在各种分类任务上表现优异。然而,传统浅层人工神经网络特征学习能力不强、生物可解释性不足,而深层神经网络存在过拟合、高功耗的缺点,因此在低功耗环境下具有生物可解释性的图像分类算法研究仍然是一个具有挑战性的任务。为了解决上述问题,结合脉冲神经网络,设计并实现了一种基于Jetson TK1和脉冲神经网络的图像分类算法。研究的主要创新点有:(1)设计了深度脉冲卷积神经网络算法,用于图像分类;
(2)实现了基于CUDA改进的脉冲神经网络模型,并部署在Jetson TK1开发环境上。

 

关键词: 图像分类, 脉冲神经网络, 移动GPU计算

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