Imputation-Based Ensemble Techniques for Class Imbalance Learning

Document Type

Article

Publication Date

5-1-2021

Publication Title

IEEE Transactions on Knowledge and Data Engineering

Volume

33

Issue

5

First Page

1988

Keywords

Class imbalance learning, ensembles learning, missing data imputation, oversampling

Last Page

2001

Abstract

Correct classification of rare samples is a vital data mining task and of paramount importance in many research domains. This article mainly focuses on the development of the novel class-imbalance learning techniques, which make use of oversampling methods integrated with bagging and boosting ensembles. Two novel oversampling strategies based on the single and the multiple imputation methods are proposed. The proposed techniques aim to create useful synthetic minority class samples, similar to the original minority class samples, by estimation of missing values that are already induced in the minority class samples. The re-balanced datasets are then used to train base-learners of the ensemble algorithms. In addition, the proposed techniques are compared with the commonly used class imbalance learning methods in terms of three performance metrics including AUC, F-measure, and G-mean over several synthetic binary class datasets. The empirical results show that the proposed multiple imputation-based oversampling combined with bagging significantly outperforms other competitors.

DOI

10.1109/TKDE.2019.2951556

ISSN

10414347

E-ISSN

15582191

Share

COinS