Impact of class imbalance on convolutional neural network training in multi-class problems
Date
2020
2020
Author
Ilham, Ahmad
Silva, Jesús
Mercado Caruso, Nohora Nubia
Tapias, Donato
Pineda, Omar
Metadata
Show full item record
Show full item record
Enlace externo del documento: https://www.scopus.com/inward/record.uri?eid=2-s2.0-85089228091&doi=10.1007%2f978-3-030-51859-2_28&partnerID=40&md5=ac05fd947220584bbc26c8914f14071e
Abstract
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.
Collections