Exploration of classification confidence in ensemble learning

被引:69
作者
Li, Leijun [1 ]
Hu, Qinghua [1 ]
Wu, Xiangqian [1 ]
Yu, Daren [2 ]
机构
[1] Harbin Inst Technol, Sch Comp Sci & Technol, Biometr Comp Res Ctr, Harbin 150001, Peoples R China
[2] Harbin Inst Technol, Sch Energy Sci & Engn, Harbin 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
Ensemble learning; Ordered aggregation; Ensemble margin; Classification confidence; DIVERSITY;
D O I
10.1016/j.patcog.2014.03.021
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Ensemble learning has attracted considerable attention owing to its good generalization performance. The main issues in constructing a powerful ensemble include training a set of diverse and accurate base classifiers, and effectively combining them. Ensemble margin, computed as the difference of the vote numbers received by the correct class and the another class received with the most votes, is widely used to explain the success of ensemble learning. This definition of the ensemble margin does not consider the classification confidence of base classifiers. In this work, we explore the influence of the classification confidence of the base classifiers in ensemble learning and obtain some interesting conclusions. First, we extend the definition of ensemble margin based on the classification confidence of the base classifiers. Then, an optimization objective is designed to compute the weights of the base classifiers by minimizing the margin induced classification loss. Several strategies are tried to utilize the classification confidences and the weights. It is observed that weighted voting based on classification confidence is better than simple voting if all the base classifiers are used. In addition, ensemble pruning can further improve the performance of a weighted voting ensemble. We also compare the proposed fusion technique with some classical algorithms. The experimental results also show the effectiveness of weighted voting with classification confidence. (C) 2014 Elsevier Ltd. All rights reserved.
引用
收藏
页码:3120 / 3131
页数:12
相关论文
共 51 条
[1]  
[Anonymous], IEEE T SYST MAN CYBE
[2]  
[Anonymous], 1993, An introduction to the bootstrap
[3]  
[Anonymous], 4 EUR C COMP LEARN T
[4]  
[Anonymous], P INT C MACH LEARN I
[5]   Clustering ensembles of neural network models [J].
Bakker, B ;
Heskes, T .
NEURAL NETWORKS, 2003, 16 (02) :261-269
[6]  
Bartlett P., 1997, ADV NEURAL INFORM PR, V9
[7]   Parallel consensual neural networks [J].
Benediktsson, JA ;
Sveinsson, JR ;
Ersoy, OK ;
Swain, PH .
IEEE TRANSACTIONS ON NEURAL NETWORKS, 1997, 8 (01) :54-64
[8]  
Blake C. L., 1998, Uci repository of machine learning databases
[9]  
Breiman L., 2001, Learn, V45, P5
[10]  
Brown G., 2005, Information Fusion, V6, P5, DOI 10.1016/j.inffus.2004.04.004