Class-specific extreme learning machine for handling binary class imbalance problem

被引:63
作者
Raghuwanshi, Bhagat Singh [1 ]
Shukla, Sanyam [1 ]
机构
[1] Maulana Azad Natl Inst Technol, Bhopal 462003, Madhya Pradesh, India
关键词
Extreme learning machine; Class-specific extreme learning machine; Class imbalance problem; Classification; DATA SETS; CLASSIFICATION; ALGORITHMS; ENGINES; TRENDS;
D O I
10.1016/j.neunet.2018.05.011
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Imbalance problem occurs when the majority class instances outnumber the minority class instances. Conventional extreme learning machine (ELM) treats all instances with same importance leading to the prediction accuracy biased towards the majority class. To overcome this inherent drawback, many variants of ELM have been proposed like Weighted ELM, class-specific cost regulation ELM (CCR-ELM) etc. to handle the class imbalance problem effectively. This work proposes class-specific extreme learning machine (CS-ELM), a variant of ELM for handling binary class imbalance problem more effectively. This work differs from weighted ELM as it does not require assigning weights to the training instances. The proposed work also has lower computational complexity compared to weighted ELM. This work uses class-specific regularization parameters. CCR-ELM also uses class-specific regularization parameters. In CCR-ELM the computation of regularization parameters does not consider class distribution and class overlap. This work uses class-specific regularization parameters which are computed using class distribution. This work also differ from CCR-ELM in the computation of the output weight, beta. The proposed work has lower computational overhead compared to CCR-ELM. The proposed work is evaluated using benchmark real world imbalanced datasets downloaded from the KEEL dataset repository. The results show that the proposed work has better performance than weighted ELM, CCR-ELM, EFSVM, FSVM, SVM for class imbalance learning. (C) 2018 Elsevier Ltd. All rights reserved.
引用
收藏
页码:206 / 217
页数:12
相关论文
共 35 条
  • [31] Deterministic neural classification
    Toh, Kar-Ann
    [J]. NEURAL COMPUTATION, 2008, 20 (06) : 1565 - 1595
  • [32] Deep Supervised Hashing with Triplet Labels
    Wang, Xiaofang
    Shi, Yi
    Kitani, Kris M.
    [J]. COMPUTER VISION - ACCV 2016, PT I, 2017, 10111 : 70 - 84
  • [33] Dynamic classifier ensemble model for customer classification with imbalanced class distribution
    Xiao, Jin
    Xie, Ling
    He, Changzheng
    Jiang, Xiaoyi
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2012, 39 (03) : 3668 - 3675
  • [34] Ensemble weighted extreme learning machine for imbalanced data classification based on differential evolution
    Zhang, Yong
    Liu, Bo
    Cai, Jing
    Zhang, Suhua
    [J]. NEURAL COMPUTING & APPLICATIONS, 2017, 28 : S259 - S267
  • [35] Weighted extreme learning machine for imbalance learning
    Zong, Weiwei
    Huang, Guang-Bin
    Chen, Yiqiang
    [J]. NEUROCOMPUTING, 2013, 101 : 229 - 242