Back to Search Start Over

Image classification with category centers in class imbalance situation

Authors :
Huilin Chen
Liguo Shuai
Yulu Zhang
Yali Ren
Source :
2018 33rd Youth Academic Annual Conference of Chinese Association of Automation (YAC).
Publication Year :
2018
Publisher :
IEEE, 2018.

Abstract

In recent years, deep convolutional networks have made a milestone progress in the field of image recognition. However, the recognition ability of the deep convolution network declines in the case of unbalanced training data, in terms of categories with fewer training images, the recognition ability of the deep convolution network declines more seriously. Aiming at this kind of problem, this paper presents a new classification method which recognizes a query image by comparing distances between category centers of CNN features of the whole training dataset and the corresponding CNN feature of this query image. The experimental results on Cifar-10 and Cifar-100 show that the claimed method can more accurately identify images whose categories have only a few training samples and that the mean precision of the recognition can be improved effectively.

Details

Database :
OpenAIRE
Journal :
2018 33rd Youth Academic Annual Conference of Chinese Association of Automation (YAC)
Accession number :
edsair.doi...........d4dbd599eee845c821baddf1c2156c94
Full Text :
https://doi.org/10.1109/yac.2018.8406400