Very deep convolutional neural network based image classification using small training sample size
Beijing University of Posts and Telecommunications
Abstract
Since Krizhevsky won the ImageNet Large Scale Visual Recognition Challenge (ILSVRC) 2012 competition with the brilliant deep convolutional neural networks (D-CNNs), researchers have designed lots of D-CNNs. However, almost all the existing very deep convolutional neural networks are trained on the giant ImageNet datasets. Small datasets like CIFAR-10 has rarely taken advantage of the power of depth since deep models are easy to overfit. In this paper, we proposed a modified VGG-16 network and used this model to fit CIFAR-10. By adding stronger regularizer and using Batch Normalization, we achieved 8.45% error rate on CIFAR-10 without severe overfitting. Our results show that the very deep CNN can be used to…
Citation impact
- FWCI
- 5.61
- Percentile
- 100%
- References
- 24
Authors
2Topics & keywords
- Overfitting
- Computer science
- Convolutional neural network
- Artificial intelligence
- Normalization (sociology)
- Deep learning
- Pattern recognition (psychology)
- Contextual image classification