Evolutionary under-sampling based bagging ensemble method for imbalanced data classification

被引:56
作者
Sun, Bo [1 ,2 ]
Chen, Haiyan [1 ,2 ]
Wang, Jiandong [1 ]
Xie, Hua [2 ]
机构
[1] Nanjing Univ Aeronaut & Astronaut, Coll Comp Sci & Technol, Nanjing 210016, Jiangsu, Peoples R China
[2] Nanjing Univ Aeronaut & Astronaut, Natl Key Lab ATFM, Nanjing 211106, Jiangsu, Peoples R China
基金
中国国家自然科学基金;
关键词
class imbalanced problem; under-sampling; bagging; evolutionary under-sampling; ensemble learning; machine learning; data mining; SUPPORT VECTOR MACHINES; DATA-SETS; SMOTE; CLASSIFIERS; STRATEGIES;
D O I
10.1007/s11704-016-5306-z
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In the class imbalanced learning scenario, traditional machine learning algorithms focusing on optimizing the overall accuracy tend to achieve poor classification performance especially for the minority class in which we are most interested. To solve this problem, many effective approaches have been proposed. Among them, the bagging ensemble methods with integration of the under-sampling techniques have demonstrated better performance than some other ones including the bagging ensemble methods integrated with the over-sampling techniques, the cost-sensitive methods, etc. Although these under-sampling techniques promote the diversity among the generated base classifiers with the help of random partition or sampling for the majority class, they do not take any measure to ensure the individual classification performance, consequently affecting the achievability of better ensemble performance. On the other hand, evolutionary under-sampling EUS as a novel undersampling technique has been successfully applied in searching for the best majority class subset for training a good-performance nearest neighbor classifier. Inspired by EUS, in this paper, we try to introduce it into the under-sampling bagging framework and propose an EUS based bagging ensemble method EUS-Bag by designing a new fitness function considering three factors to make EUS better suited to the framework. With our fitness function, EUS-Bag could generate a set of accurate and diverse base classifiers. To verify the effectiveness of EUS-Bag, we conduct a series of comparison experiments on 22 two-class imbalanced classification problems. Experimental results measured using recall, geometric mean and AUC all demonstrate its superior performance.
引用
收藏
页码:331 / 350
页数:20
相关论文
共 58 条
[1]  
[Anonymous], 2012, Genetic Algorithms: Concepts and Designs
[2]  
[Anonymous], 2003, C45 CLASS IMBALANCE
[3]  
[Anonymous], 2012, IEEE T SYST MAN CY C, DOI DOI 10.1109/TSMCC.2011.2161285
[4]  
[Anonymous], 2004, Machine Learning
[5]  
[Anonymous], P 23 ANN ACM C MULT
[6]  
Banfield R. E., 2005, Information Fusion, V6, P49, DOI 10.1016/j.inffus.2004.04.005
[7]   A comparison of decision tree ensemble creation techniques [J].
Banfield, Robert E. ;
Hall, Lawrence O. ;
Bowyer, Kevin W. ;
Kegelmeyer, W. P. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2007, 29 (01) :173-180
[8]   Influence Measures for CART Classification Trees [J].
Bar-Hen, Avner ;
Gey, Servane ;
Poggi, Jean-Michel .
JOURNAL OF CLASSIFICATION, 2015, 32 (01) :21-45
[9]   New applications of ensembles of classifiers [J].
Barandela, R ;
Sánchez, JS ;
Valdovinos, RM .
PATTERN ANALYSIS AND APPLICATIONS, 2003, 6 (03) :245-256
[10]   Strategies for learning in class imbalance problems [J].
Barandela, R ;
Sánchez, JS ;
García, V ;
Rangel, E .
PATTERN RECOGNITION, 2003, 36 (03) :849-851