首页 | 本学科首页   官方微博 | 高级检索  
     

基于深度卷积神经网络的快速图像分类算法
引用本文:王华利,邹俊忠,张 见,卫作臣,汪春梅.基于深度卷积神经网络的快速图像分类算法[J].计算机工程与应用,2017,53(13):181-188.
作者姓名:王华利  邹俊忠  张 见  卫作臣  汪春梅
作者单位:1.华东理工大学 信息科学与工程学院,上海 200237 2.上海师范大学 信息与机电工程学院,上海 200234
摘    要:为了应对大量图像的分类问题,提出一种基于深度卷积神经网络和CUDA-cuDNN并行运算的快速图像分类方法。该方法利用深度卷积神经网络自动学习特征的优势来解决手工设计特征普适性差等问题,同时结合基于CUDA架构的cuDNN并行运算策略来提高训练速度和加快分类速度,并且针对深度卷积神经网络易受参数扰动等缺点,引入批量正则化(Batch Normalization)以提高算法的鲁棒性。实验结果表明,该方法不仅大幅缩短了训练时间同时加快了图像的分类速度,而且进一步降低了图像分类的错误率。

关 键 词:深度卷积神经网络  CUDA-cuDNN方法  批量正则化  图像分类  深度学习  

Fast image classification algorithm based on deep convolutional neural network
WANG Huali,ZOU Junzhong,ZHANG Jian,WEI Zuochen,WANG Chunmei.Fast image classification algorithm based on deep convolutional neural network[J].Computer Engineering and Applications,2017,53(13):181-188.
Authors:WANG Huali  ZOU Junzhong  ZHANG Jian  WEI Zuochen  WANG Chunmei
Affiliation:1.School of Information Science and Engineering, East China University of Science and Technology, Shanghai 200237, China 2.School of Information, Mechanical and Electrical Engineering, Shanghai Normal University, Shanghai 200234, China
Abstract:In order to solve large amount of images classification issues, a method is introduced by combining with CUDA-cuDNN and Deep Convolutional Neural Network(DCNN). This method makes advantages of DCNNs to learn features automatically, which makes up the incapability of hand-crafted features. Meanwhile, a cuDNN parallel computing method based on CUDA is employed to improve the speed of training and validation. DCNN is susceptible to parameter perturbation, which employs Batch Normalization(BN) to enhance the robustness. Experimental results indicate that the proposed method not only reduces training time substantially and accelerates validation speed, but also obtains lower classification error rate.
Keywords:Deep Convolutional Neural Network(DCNN)  CUDA-cuDNN  batch normalization  image classification  deep learning  
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号