Back to Search Start Over

Optimizing area under the ROC curve via extreme learning machines.

Authors :
Yang, Zhiyong
Zhang, Taohong
Lu, Jingcheng
Zhang, Dezheng
Kalui, Dorothy
Source :
Knowledge-Based Systems. Aug2017, Vol. 130, p74-89. 16p.
Publication Year :
2017

Abstract

Recently, Extreme learning machine (ELM), an efficient training algorithm for single-hidden-layer feedforward neural networks (SLFN), has gained increasing popularity in machine learning communities. In this paper the ELM based Area Under the ROC Curve (AUC) optimization algorithms are studied so as to further improve the performance of ELM for imbalanced datasets. For binary class problems, a novel ELM algorithm is proposed based on an efficient least square method. For multi-class problems, the following works are done in this paper: First of all, theoretical comparison analysis is proposed for the potential multi-class extensions of AUC; Secondly, a unified objective function for multi-class AUC optimization is proposed following the theoretical analysis; Subsequently, two ELM based multi-class AUC optimization algorithms called E L M M A U C and E L M m a c r o A U C respectively are proposed followed with complexity analyses; Finally, the generalization analysis is established for E L M M A U C in search of theoretical supports. Empirical study on a variety of real-world datasets show the effectiveness of our proposed algorithms. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
09507051
Volume :
130
Database :
Academic Search Index
Journal :
Knowledge-Based Systems
Publication Type :
Academic Journal
Accession number :
123866730
Full Text :
https://doi.org/10.1016/j.knosys.2017.05.013