IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0146116.html
   My bibliography  Save this article

Heterogeneous Ensemble Combination Search Using Genetic Algorithm for Class Imbalanced Data Classification

Author

Listed:
  • Mohammad Nazmul Haque
  • Nasimul Noman
  • Regina Berretta
  • Pablo Moscato

Abstract

Classification of datasets with imbalanced sample distributions has always been a challenge. In general, a popular approach for enhancing classification performance is the construction of an ensemble of classifiers. However, the performance of an ensemble is dependent on the choice of constituent base classifiers. Therefore, we propose a genetic algorithm-based search method for finding the optimum combination from a pool of base classifiers to form a heterogeneous ensemble. The algorithm, called GA-EoC, utilises 10 fold-cross validation on training data for evaluating the quality of each candidate ensembles. In order to combine the base classifiers decision into ensemble’s output, we used the simple and widely used majority voting approach. The proposed algorithm, along with the random sub-sampling approach to balance the class distribution, has been used for classifying class-imbalanced datasets. Additionally, if a feature set was not available, we used the (α, β) − k Feature Set method to select a better subset of features for classification. We have tested GA-EoC with three benchmarking datasets from the UCI-Machine Learning repository, one Alzheimer’s disease dataset and a subset of the PubFig database of Columbia University. In general, the performance of the proposed method on the chosen datasets is robust and better than that of the constituent base classifiers and many other well-known ensembles. Based on our empirical study we claim that a genetic algorithm is a superior and reliable approach to heterogeneous ensemble construction and we expect that the proposed GA-EoC would perform consistently in other cases.

Suggested Citation

  • Mohammad Nazmul Haque & Nasimul Noman & Regina Berretta & Pablo Moscato, 2016. "Heterogeneous Ensemble Combination Search Using Genetic Algorithm for Class Imbalanced Data Classification," PLOS ONE, Public Library of Science, vol. 11(1), pages 1-28, January.
  • Handle: RePEc:plo:pone00:0146116
    DOI: 10.1371/journal.pone.0146116
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0146116
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0146116&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0146116?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Liao, Jui-Jung & Shih, Ching-Hui & Chen, Tai-Feng & Hsu, Ming-Fu, 2014. "An ensemble-based model for two-class imbalanced financial problem," Economic Modelling, Elsevier, vol. 37(C), pages 175-183.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Maryam Mahsal Khan & Alexandre Mendes & Stephan K Chalup, 2018. "Evolutionary Wavelet Neural Network ensembles for breast cancer and Parkinson’s disease prediction," PLOS ONE, Public Library of Science, vol. 13(2), pages 1-15, February.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. du Jardin, Philippe, 2021. "Forecasting corporate failure using ensemble of self-organizing neural networks," European Journal of Operational Research, Elsevier, vol. 288(3), pages 869-885.
    2. Kristof Lommers & Ouns El Harzli & Jack Kim, 2021. "Confronting Machine Learning With Financial Research," Papers 2103.00366, arXiv.org, revised Mar 2021.
    3. Zoričák, Martin & Gnip, Peter & Drotár, Peter & Gazda, Vladimír, 2020. "Bankruptcy prediction for small- and medium-sized companies using severely imbalanced datasets," Economic Modelling, Elsevier, vol. 84(C), pages 165-176.
    4. Hayashi, Yoichi, 2016. "Application of a rule extraction algorithm family based on the Re-RX algorithm to financial credit risk assessment from a Pareto optimal perspective," Operations Research Perspectives, Elsevier, vol. 3(C), pages 32-42.
    5. Wanke, Peter & Barros, Carlos Pestana, 2016. "Efficiency drivers in Brazilian insurance: A two-stage DEA meta frontier-data mining approach," Economic Modelling, Elsevier, vol. 53(C), pages 8-22.

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0146116. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.