Skip to main navigation Skip to search Skip to main content

A novel imbalanced classification method based on decision tree and bagging

  • Hongjiao Guan*
  • , Yingtao Zhang
  • , Hengda Cheng
  • , Xianglong Tang
  • *Corresponding author for this work
  • School of Computer Science and Technology, Harbin Institute of Technology
  • Utah State University

Research output: Contribution to journalArticlepeer-review

Abstract

Imbalanced classification is a challenging problem in the field of big data research and applications. Complex data distributions, such as small disjuncts and overlapping classes, make traditional methods unable to easily recognize the minority class and thus, lead to low sensitivity. The misclassification costs of the minority class are usually higher than that of the majority class. To deal with imbalanced datasets, typical algorithmic-level methods either introduce cost information or simply rebalance class distribution without considering the distribution of the minority class. In this paper, we propose an optimization embedded bagging (OEBag) approach to increase the sensitivity by learning the complex distributions in the minority class more precisely. By learning these base classifiers, OEBag selectively learns the minority examples that are misclassified easily by referring to examples in out-of-bag. OEBag is implemented by using two specialized under-sampling bagging methods. Nineteen real datasets with diverse levels of classification difficulties are utilized in this paper. Experimental results demonstrate that OEBag performs significantly better in sensitivity and has a great overall performance in terms of AUC (area under ROC curve) and G-mean when compared with several state-of-the-art methods.

Original languageEnglish
Pages (from-to)1140-1148
Number of pages9
JournalInternational Journal of Performability Engineering
Volume14
Issue number6
DOIs
StatePublished - Jun 2018
Externally publishedYes

Keywords

  • Bagging
  • Decision tree
  • Imbalanced classification
  • Overlapping among classes
  • Small disjuncts

Fingerprint

Dive into the research topics of 'A novel imbalanced classification method based on decision tree and bagging'. Together they form a unique fingerprint.

Cite this