AN IMPROVED DEEP CONVOLUTIONAL NEURAL NETWORK MODEL WITH KERNEL LOSS FUNCTION IN IMAGE CLASSIFICATION

被引:5
作者
Xia, Yuantian [1 ]
Zhou, Juxiang [1 ]
Xu, Tianwei [1 ]
Gao, Wei [2 ]
机构
[1] Yunnan Normal Univ, Minist Educ, Key Lab Educ Informatizat Nationalities, Kunming 650500, Yunnan, Peoples R China
[2] Yunnan Normal Univ, Sch Informat Sci & Technol, Kunming 650500, Yunnan, Peoples R China
来源
MATHEMATICAL FOUNDATIONS OF COMPUTING | 2020年 / 3卷 / 01期
基金
中国国家自然科学基金;
关键词
Convolutional neural network; image classification; convolution kernel; stride; dropout; fully connected layer;
D O I
10.3934/mfc.2020005
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
To further enhance the performance of the current convolutional neural network, an improved deep convolutional neural network model is shown in this paper. Different from the traditional network structure, in our proposed method the pooling layer is replaced by two continuous convolutional layers with 3 x 3 convolution kernel between which a dropout layer is added to reduce overfitting, and cross entropy kernel is used as loss function. Experimental results on Mnist and Cifar-10 data sets for image classification show that, compared to several classical neural networks such as Alexnet, VGGNet and GoogleNet, the improved network achieve better performance in learning efficiency and recognition accuracy at relatively shallow network depths.
引用
收藏
页码:51 / 64
页数:14
相关论文
共 24 条
[1]   Generalization of deep neural network for bearing fault diagnosis under different working conditions using multiple kernel method [J].
An, Zenghui ;
Li, Shunming ;
Wang, Jinrui ;
Xin, Yu ;
Xu, Kun .
NEUROCOMPUTING, 2019, 352 :42-53
[2]  
[Anonymous], 2016, CoRR abs/1602.02830
[3]  
[Anonymous], representation learning with deep convolutional generative
[4]  
[Anonymous], 2015, P INT C LEARN REP IC
[5]  
[Anonymous], 2017, Pattern Anal. Mach. Intell.
[6]  
Brevdo E., 2016, TENSOR
[7]   Deep neural networks for rotation-invariance approximation and learning [J].
Chui, Charles K. ;
Lin, Shao-Bo ;
Zhou, Ding-Xuan .
ANALYSIS AND APPLICATIONS, 2019, 17 (05) :737-772
[8]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[9]   An iterative multistep kernel based method for nonlinear Volterra integral and integro-differential equations of fractional order [J].
Heydari, Mojgan ;
Shivanian, Elyas ;
Azarnavid, Babak ;
Abbasbandy, Saeid .
JOURNAL OF COMPUTATIONAL AND APPLIED MATHEMATICS, 2019, 361 :97-112
[10]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269