A Novel Algorithm for Imbalance Data Classification Based on Genetic Algorithm Improved SMOTE

被引:82
作者
Jiang, Kun [1 ]
Lu, Jing [1 ]
Xia, Kuiliang [1 ]
机构
[1] HeiHe Univ, Heihe, Peoples R China
关键词
Imbalanced dataset; Classification; SMOTE; Sampling rate; Genetic algorithm; Rockburst;
D O I
10.1007/s13369-016-2179-2
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
The classification of imbalanced data has been recognized as a crucial problem in machine learning and data mining. In an imbalanced dataset, there are significantly fewer training instances of one class compared to another class. Hence, the minority class instances are much more likely to be misclassified. In the literature, the synthetic minority over-sampling technique (SMOTE) has been developed to deal with the classification of imbalanced datasets. It synthesizes new samples of the minority class to balance the dataset, by re-sampling the instances of the minority class. Nevertheless, the existing algorithms-based SMOTE uses the same sampling rate for all instances of the minority class. This results in sub-optimal performance. To address this issue, we propose a novel genetic algorithm-based SMOTE (GASMOTE) algorithm. The GASMOTE algorithm uses different sampling rates for different minority class instances and finds the combination of optimal sampling rates. The experimental results on ten typical imbalance datasets show that, compared with SMOTE algorithm, GASMOTE can increase 5.9% on F-measure value and 1.6% on G-mean value, and compared with Borderline-SMOTE algorithm, GASMOTE can increase 3.7% on F-measure value and 2.3% on G-mean value. GASMOTE can be used as a new over-sampling technique to deal with imbalance dataset classification problem. We have particularly applied the GASMOTE algorithm to a practical engineering application: prediction of rockburst in the VCR rockburst datasets. The experiment results indicate that the GASMOTE algorithm can accurately predict the rockburst occurrence and hence provides guidance to the design and construction of safe deep mining engineering structures.
引用
收藏
页码:3255 / 3266
页数:12
相关论文
共 41 条
[1]   An approach for classification of highly imbalanced data using weighting and undersampling [J].
Anand, Ashish ;
Pugalenthi, Ganesan ;
Fogel, Gary B. ;
Suganthan, P. N. .
AMINO ACIDS, 2010, 39 (05) :1385-1391
[2]  
[Anonymous], IC AI
[3]  
[Anonymous], CHIN J COMPUT
[4]  
[Anonymous], PAKDD
[5]  
[Anonymous], 2009, RES APPL GENETIC ALG
[6]  
[Anonymous], COMPUT SYST APPL
[7]  
[Anonymous], COMPUT ENG DES
[8]  
[Anonymous], 2012, IEEE T SYST MAN CY C, DOI DOI 10.1109/TSMCC.2011.2161285
[9]  
[Anonymous], J YANGTZE U NAT SCI
[10]  
[Anonymous], DATA MINING PRACTICA