Impact of class imbalance on convolutional neural network training in multi-class problems

Ilham A.; Silva J.; Mercado-Caruso N.; Tapias-Ruiz D.; Lezama O.B.P. · Advances in Intelligent Systems and Computing (2021) · DOI: 10.1007/978-3-030-51859-2_28

Resumen

Image classification is the process of assigning an image one or multiple tags that describe its content. To perform the classification, a model must be designed for learning the labels to be assigned to a given image. The assignment is made through a learning process that uses a set of previously labeled training images, which must be large enough to guarantee efficient training. Many approaches have been researched to find optimal solutions to classification problems, however, databases with large amounts of images and the increased processing power of GPUs have made convolutional neural networks (CNNs) the best choice, as they outperform traditional algorithms. This paper presents a systematic analysis aimed at understanding how the issue of class inequality affects the efficiency of a convolutionary neural network trained for a task of image classification, and presents a technique for correcting the overtraining and that the network generalization. © The Editor(s) (if applicable) and The Author(s), under exclusive license to Springer Nature Switzerland AG 2021.

Instituciones: Universidad Peruana de Ciencias Aplicadas