Deep Over-sampling Framework for Classifying Imbalanced Data

• Published in: Machine Learning and Knowledge Discovery in Databases Vol. 10534; pp. 770 - 785
• Main Authors: Ando, Shin, Huang, Chun Yuan
• Format: Book Chapter
• Language: English
• Published: Switzerland Springer International Publishing AG 2017; Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: Class imbalance; Convolutional neural network; Deep learning; Representation learning; Synthetic over-sampling
• DOI: 10.1007/978-3-319-71249-9_46
• ISBN: 3319712489; 9783319712482
• ISSN: 0302-9743

Class imbalance is a challenging issue in practical classification problems for deep learning models as well as traditional models. Traditionally successful countermeasures such as synthetic over-sampling have had limited success with complex, structured data handled by deep learning models. In this paper, we propose Deep Over-sampling (DOS), a framework for extending the synthetic over-sampling method to the deep feature space acquired by a convolutional neural network (CNN). Its key feature is an explicit, supervised representation learning, for which the training data presents each raw input sample with a synthetic embedding target in the deep feature space, which is sampled from the linear subspace of in-class neighbors. We implement an iterative process of training the CNN and updating the targets, which induces smaller in-class variance among the embeddings, to increase the discriminative power of the deep representation. We present an empirical study using public benchmarks, which shows that the DOS framework not only counteracts class imbalance better than the existing method, but also improves the performance of the CNN in the standard, balanced settings.