Tiny Image Classification using Four-Block Convolutional Neural Network

被引:6
作者
Sharif, Mohsin [1 ]
Kausar, Asia [1 ]
Park, JinHyuck [1 ]
Shin, Dong Ryeol [1 ]
机构
[1] Sungkyunkwan Univ, Dept Elect & Comp Engn, Suwon, South Korea
来源
2019 10TH INTERNATIONAL CONFERENCE ON INFORMATION AND COMMUNICATION TECHNOLOGY CONVERGENCE (ICTC): ICT CONVERGENCE LEADING THE AUTONOMOUS FUTURE | 2019年
基金
新加坡国家研究基金会;
关键词
Low-Resolution Images; Convolutional Neural Network; CINIC-10; Multi-class image classication; Batch Normalization; BACKPROPAGATION;
D O I
10.1109/ictc46691.2019.8940002
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The task of classifying images into predefined classes is a major problem in computer vision and artificial intelligence. Deep neural network such as Convolutional Neural Network (CNN) have shown great success in large-scale dataset of high resolution image classification. Here, it is important to note that most real time images may not have high resolution. With the increasing demand of surveillance camera-based applications, the low resolution images are major problem. To overcome this problem, we propose two Four-Block CNN model; one with four-layers and the other one with three-layers. Our proposed Four-block four-layer CNN model contains four convolution layers, first three layers contains 3 x 3 kernel size with stride-1 and fourth layer used with stride-2 for dimensionality reduction.The Four-block three-layer has two convolutional layers with stride-1 and third layer with stride-2. We trained our model on CINIC10 and CIFAR-10 datasets having low-resolution images. For taking average of whole feature map we are using Global Average Pooling layer as a classifier in both models. To reduce training time complexity, we use non-saturating neurons. Overfitting problem has been addressed by dropout and batch normalization methods. On the validation data, we achieved the best accuracy of 81.62%, 92.21% for CINIC-10 and CIFAR-10 respectively, using Four-block four-layer model.
引用
收藏
页码:1 / 6
页数:6
相关论文
共 23 条
  • [1] Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
  • [2] [Anonymous], P 3 INT C LEARNING R
  • [3] [Anonymous], PROC CVPR IEEE
  • [4] [Anonymous], 2015, ARXIV PREPRINT ARXIV
  • [5] [Anonymous], 2017, COMMUN ACM, DOI DOI 10.1145/3065386
  • [6] [Anonymous], 2014, INT C LEARN REPR ICL
  • [7] [Anonymous], 2014, P 2 INT C LEARNING R
  • [8] Bastidas A., 2017, TINY IMAGENET IMAGE
  • [9] Caruana R, 2001, ADV NEUR IN, V13, P402
  • [10] Cogswell Michael, 2016, arXiv preprint arXiv:1511.06068