Back to Search Start Over

Extreme learning machine with hybrid cost function of G-mean and probability for imbalance learning

Authors :
JongHyok Ri
Guanzhong Tian
Yong Liu
Wei-hua Xu
Jun-gang Lou
Source :
International Journal of Machine Learning and Cybernetics. 11:2007-2020
Publication Year :
2020
Publisher :
Springer Science and Business Media LLC, 2020.

Abstract

Extreme learning machine(ELM) is a simple and fast machine learning algorithm. However, similar to other conventional learning algorithms, the classical ELM can not well process the problem of imbalanced data distribution. In this paper, in order to improve the learning performance of classical ELM for imbalanced data learning, we present a novel variant of the ELM algorithm based on a hybrid cost function which employs the probability that given training sample belong in each class to calculate the G-mean. We perform comparable experiments for our approach and the state-of-the-arts methods on standard classification datasets which consist of 58 binary datasets and 9 multiclass datasets under different degrees of imbalance ratio. Experimental results show that our proposed algorithm can improve the classification performance significantly compared with other state-of-the-art methods.

Details

ISSN :
1868808X and 18688071
Volume :
11
Database :
OpenAIRE
Journal :
International Journal of Machine Learning and Cybernetics
Accession number :
edsair.doi...........bc247ca55b0bf40fac20189cf4fc3d1b