Bayes Imbalance Impact Index: A Measure of Class Imbalanced Data Set for Classification Problem

Yang Lu, Yiu Ming Cheung*, Yuan Yan Tang

*Corresponding author for this work

Research output: Contribution to journalJournal articlepeer-review

49 Citations (Scopus)

Abstract

Recent studies of imbalanced data classification have shown that the imbalance ratio (IR) is not the only cause of performance loss in a classifier, as other data factors, such as small disjuncts, noise, and overlapping, can also make the problem difficult. The relationship between the IR and other data factors has been demonstrated, but to the best of our knowledge, there is no measurement of the extent to which class imbalance influences the classification performance of imbalanced data. In addition, it is also unknown which data factor serves as the main barrier for classification in a data set. In this article, we focus on the Bayes optimal classifier and examine the influence of class imbalance from a theoretical perspective. We propose an instance measure called the Individual Bayes Imbalance Impact Index (IBI3) and a data measure called the Bayes Imbalance Impact Index (BI3). IBI3 and BI3 reflect the extent of influence using only the imbalance factor, in terms of each minority class sample and the whole data set, respectively. Therefore, IBI3 can be used as an instance complexity measure of imbalance and BI3 as a criterion to demonstrate the degree to which imbalance deteriorates the classification of a data set. We can, therefore, use BI3 to access whether it is worth using imbalance recovery methods, such as sampling or cost-sensitive methods, to recover the performance loss of a classifier. The experiments show that IBI3 is highly consistent with the increase of the prediction score obtained by the imbalance recovery methods and that BI3 is highly consistent with the improvement in the F1 score obtained by the imbalance recovery methods on both synthetic and real benchmark data sets.

Original languageEnglish
Pages (from-to)3525-3539
Number of pages15
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume31
Issue number9
Early online date1 Nov 2019
DOIs
Publication statusPublished - Sept 2020

Scopus Subject Areas

  • Software
  • Computer Science Applications
  • Computer Networks and Communications
  • Artificial Intelligence

User-Defined Keywords

  • Bayes classifier
  • class imbalance learning
  • data complexity
  • imbalance measure
  • imbalance recovery methods

Fingerprint

Dive into the research topics of 'Bayes Imbalance Impact Index: A Measure of Class Imbalanced Data Set for Classification Problem'. Together they form a unique fingerprint.

Cite this