Class Balanced Loss for Image Classification

Citations Over TimeTop 15% of 2020 papers

Abstract

In the study of image classification, neural network learning relies heavily on datasets. Due to variability in the difficulty of collecting images in reality, datasets tend to have class imbalance problems, which undoubtedly increases the difficulty of classification. During the training of a neural network, classes with a large number of images are naturally trained more often than classes with a small number of images. Because of imbalanced training, the classification ability of neural networks on test and validation sets differs greatly in different categories. The test results of more training classes are better, and the test results of classes with less training are poor. In this paper, we propose two kinds of balanced loss functions, namely, CEFL loss and CEFL2 loss, by rebalancing the cross-entropy loss function and focal loss function. The experimental results show that the proposed loss functions are significantly able to improve classification accuracy on class-imbalanced datasets.

Related Papers

• → Application of Weighted Cross-Entropy Loss Function in Intrusion Detection(2021)36 cited
• 2-D minimum cross entropy based on genetic algorithm for dynamic image segmentation(2005)
• → Taming the Cross Entropy Loss(2018)5 cited
• → On the Optimal Combination of Cross-Entropy and Soft Dice Losses for Lesion Segmentation with Out-of-Distribution Robustness(2022)2 cited
• → Transitory cross entropy for model training on unbalanced datasets(2022)