Impact of Training Set Batch Size on the Performance of Convolutional Neural Networks for Diverse Datasets

Information Technology and Management Science 2017
Pavlo M. Radiuk

A problem of improving the performance of convolutional neural networks is considered. A parameter of the training set is investigated. The parameter is the batch size. The goal is to find an impact of training set batch size on the performance. To get consistent results, diverse datasets are used. They areMNIST and CIFAR-10. Simplicity of the MNIST dataset stands against complexity of the CIFAR-10 dataset, although the simpler dataset has 10 classes as well as the more complicated one. To achieve acceptable testing results, various convolutional neural network architectures are selected for the MNIST and CIFAR-10 datasets, with two and five convolutional layers, respectively. The assumption about the dependence of the recognition accuracy on the batch size value is confirmed: the larger the batch size value, the higher the recognition accuracy. Another assumption about the impact of the type of the batch size value on the CNN performance is not confirmed.

Atslēgas vārdi
Batch size, convolutional neural network, dataset, testing accuracy

Radiuk, P. Impact of Training Set Batch Size on the Performance of Convolutional Neural Networks for Diverse Datasets. Information Technology and Management Science, 2017, Vol. 20, 22.-24. lpp. ISSN 2255-9086. e-ISSN 2255-9094.

Publikācijas valoda
English (en)