arrow
Return

Class Balanced Loss for Image Classification

delete2020-01-01
delete16
delete
OA
AI
L
Lin Wang
孙
孙占全 (Zhanquan Sun)
S
Shuqun Cheng *
L
Lei Guo
DOI:10.1109/ACCESS.2020.2991237delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

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

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

N
Naval Medical University
Scholars:
2.1W
Papers: 1.3W
Citations: 24
Cited Papers

Cited Papers

A systematic study of the class imbalance problem in convolutional neural networks
err2018-10-01
err1.7K
errOAAI
errBuda, Mateusz; Maki, Atsuto; Mazurowski, Maciej A.
errShare
errSave
Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent
err2015-01-01
err0
PREAI
errTuan. A. Vu; Giang. H. Le; Canh. D. Dao; Lan. Q. Dang; Kien. T. Nguyen; Quang. K. Nguyen; Phuong. T. Dang; Hoa. T. K. Tran; Quang. T. Duong; Tuyen. V. Nguyen; Gun. D. Lee
errShare
errSave
errShare
errSave
Structural study of lanthanides(III) in aqueous nitrate and chloride solutions by EXAFS
err1999-02-01
err0
PREAI
errT. Yaita; H. Narita; Sh. Suzuki; Sh. Tachimori; H. Motohashi; H. Shiwaku
errShare
errSave
Development of a novel Maillard reaction-based time–temperature indicator for monitoring the fluorescent AGE content in reheated foods
err2020-01-01
err0
errOAAI
errBei Hu; Lin Li; Yi Hu; Di Zhao; Yuting Li; Mingquan Yang; Aijuan Jia; Sui Chen; Bing Li; Xia Zhang
errShare
errSave
Recycling of NdFeB Magnets from Electric Drive Motors of (Hybrid) Electric Vehicles
err2016-10-12
err0
PREAI
errTobias Elwert; Daniel Goldmann; Felix Roemer; Sabrina Schwarz
errShare
errSave
researcher View more