1. Extreme learning machine with hybrid cost function of G-mean and probability for imbalance learning
- Author
-
JongHyok Ri, Guanzhong Tian, Yong Liu, Wei-hua Xu, and Jun-gang Lou
- Subjects
business.industry ,Computer science ,Process (computing) ,Binary number ,Computational intelligence ,Sample (statistics) ,Function (mathematics) ,Machine learning ,computer.software_genre ,Class (biology) ,Artificial Intelligence ,Pattern recognition (psychology) ,Computer Vision and Pattern Recognition ,Artificial intelligence ,business ,computer ,Software ,Extreme learning machine - 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.
- Published
- 2020