Return
Class Balanced Loss for Image Classification
DOI:10.1109/ACCESS.2020.2991237.png)
Abstract
En 中文
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.
Keywords:
Classification
class imbalance
loss
accuracy
AI Summary
Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.
Journal
IF:
3.6
Papers:
9.8W
Citations:
29.4W
Organization
Cited Papers
A systematic study of the class imbalance problem in convolutional neural networks
NEURAL NETWORKS
IF6.3
Thermal comfort, perceived air quality, and cognitive performance when personally controlled air movement is used by tropically acclimatized persons
Indoor Air
IF0
Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent
RSC Advances
IF0
Sparse Matrix Classification on Imbalanced Datasets Using Convolutional Neural Networks
IEEE ACCESS
IF3.6
Development of a novel Maillard reaction-based time–temperature indicator for monitoring the fluorescent AGE content in reheated foods
RSC Advances
IF0

