The effect of batch size on the generalizability of the convolutional neural networks on a histopathology dataset

Kandel, Ibrahem (avtor), Castelli, Mauro (avtor)

Many hyperparameters have to be tuned to have a robust convolutional neural network that will be able to accurately classify images. One of the most important hyperparameters is the batch size, which is the number of images used to train a single forward and backward pass. In this study, the effect of batch size on the performance of convolutional neural networks and the impact of learning rates will be studied for image classification, specifically for medical images. To train the network faster, a VGG16 network with ImageNet weights was used in this experiment. Our results concluded that a higher batch size does not usually achieve high accuracy, and the learning rate and the optimizer used will have a significant impact as well. Lowering the learning rate and decreasing the batch size will allow the network to train better, especially in the case of fine-tuning.

• Jezik: Angleški jezik
• Vrsta gradiva: Članek v reviji
• Tipologija: 1.01 - Izvirni znanstveni članek
• Organizacija: EF - Ekonomska fakulteta
• Različica publikacije: Objavljena publikacija
• Leto izida: 2020
• Št. strani: Str. 312-315
• Številčenje: Vol. 6, iss. 4
• PID: 20.500.12556/RUL-124243
• UDK: 681.5
• ISSN pri članku: 2405-9595
• DOI: 10.1016/j.icte.2020.04.010
• COBISS.SI-ID: 38422787
• Datum objave v RUL: 12.01.2021

Gradivo je del revije

• Naslov: ICT express
• Založnik: Elsevier
• ISSN: 2405-9595
• COBISS.SI-ID: 526132505

Licence

• Licenca: CC BY-NC-ND 4.0, Creative Commons Priznanje avtorstva-Nekomercialno-Brez predelav 4.0 Mednarodna
• Povezava: http://creativecommons.org/licenses/by-nc-nd/4.0/deed.sl
• Opis: Najbolj omejujoča licenca Creative Commons. Uporabniki lahko prenesejo in delijo delo v nekomercialne namene in ga ne smejo uporabiti za nobene druge namene.
• Začetek licenciranja: 12.01.2021