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

An empirical evaluation of sampling methods for the classification of imbalanced data

Author

Listed:
  • Misuk Kim
  • Kyu-Baek Hwang

Abstract

In numerous classification problems, class distribution is not balanced. For example, positive examples are rare in the fields of disease diagnosis and credit card fraud detection. General machine learning methods are known to be suboptimal for such imbalanced classification. One popular solution is to balance training data by oversampling the underrepresented (or undersampling the overrepresented) classes before applying machine learning algorithms. However, despite its popularity, the effectiveness of sampling has not been rigorously and comprehensively evaluated. This study assessed combinations of seven sampling methods and eight machine learning classifiers (56 varieties in total) using 31 datasets with varying degrees of imbalance. We used the areas under the precision-recall curve (AUPRC) and receiver operating characteristics curve (AUROC) as the performance measures. The AUPRC is known to be more informative for imbalanced classification than the AUROC. We observed that sampling significantly changed the performance of the classifier (paired t-tests P

Suggested Citation

  • Misuk Kim & Kyu-Baek Hwang, 2022. "An empirical evaluation of sampling methods for the classification of imbalanced data," PLOS ONE, Public Library of Science, vol. 17(7), pages 1-22, July.
  • Handle: RePEc:plo:pone00:0271260
    DOI: 10.1371/journal.pone.0271260
    as

    Download full text from publisher

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

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

    File URL: https://libkey.io/10.1371/journal.pone.0271260?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. Takaya Saito & Marc Rehmsmeier, 2015. "The Precision-Recall Plot Is More Informative than the ROC Plot When Evaluating Binary Classifiers on Imbalanced Datasets," PLOS ONE, Public Library of Science, vol. 10(3), pages 1-21, March.
    2. Zhongkai Liu & Howard D. Bondell, 2019. "Binormal Precision–Recall Curves for Optimal Classification of Imbalanced Data," Statistics in Biosciences, Springer;International Chinese Statistical Association, vol. 11(1), pages 141-161, April.
    Full references (including those not matched with items on IDEAS)

    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. Yifan Zhong & Chuang Cai & Tao Chen & Hao Gui & Jiajun Deng & Minglei Yang & Bentong Yu & Yongxiang Song & Tingting Wang & Xiwen Sun & Jingyun Shi & Yangchun Chen & Dong Xie & Chang Chen & Yunlang She, 2023. "PET/CT based cross-modal deep learning signature to predict occult nodal metastasis in lung cancer," Nature Communications, Nature, vol. 14(1), pages 1-14, December.
    2. Christopher J Greenwood & George J Youssef & Primrose Letcher & Jacqui A Macdonald & Lauryn J Hagg & Ann Sanson & Jenn Mcintosh & Delyse M Hutchinson & John W Toumbourou & Matthew Fuller-Tyszkiewicz &, 2020. "A comparison of penalised regression methods for informing the selection of predictive markers," PLOS ONE, Public Library of Science, vol. 15(11), pages 1-14, November.
    3. Jie-Huei Wang & Cheng-Yu Liu & You-Ruei Min & Zih-Han Wu & Po-Lin Hou, 2024. "Cancer Diagnosis by Gene-Environment Interactions via Combination of SMOTE-Tomek and Overlapped Group Screening Approaches with Application to Imbalanced TCGA Clinical and Genomic Data," Mathematics, MDPI, vol. 12(14), pages 1-24, July.
    4. Le, Hong Hanh & Viviani, Jean-Laurent, 2018. "Predicting bank failure: An improvement by implementing a machine-learning approach to classical financial ratios," Research in International Business and Finance, Elsevier, vol. 44(C), pages 16-25.
    5. João Chang Junior & Fábio Binuesa & Luiz Fernando Caneo & Aida Luiza Ribeiro Turquetto & Elisandra Cristina Trevisan Calvo Arita & Aline Cristina Barbosa & Alfredo Manoel da Silva Fernandes & Evelinda, 2020. "Improving preoperative risk-of-death prediction in surgery congenital heart defects using artificial intelligence model: A pilot study," PLOS ONE, Public Library of Science, vol. 15(9), pages 1-21, September.
    6. Arthur De Sá Ferreira & Ney Meziat-Filho & Ana Paula Antunes Ferreira, 2021. "Double threshold receiver operating characteristic plot for three-modal continuous predictors," Computational Statistics, Springer, vol. 36(3), pages 2231-2245, September.
    7. Fan, Xudong & Wang, Xiaowei & Zhang, Xijin & ASCE Xiong (Bill) Yu, P.E.F., 2022. "Machine learning based water pipe failure prediction: The effects of engineering, geology, climate and socio-economic factors," Reliability Engineering and System Safety, Elsevier, vol. 219(C).
    8. Zhang, Han, 2021. "How Using Machine Learning Classification as a Variable in Regression Leads to Attenuation Bias and What to Do About It," SocArXiv 453jk, Center for Open Science.
    9. Masabho P Milali & Samson S Kiware & Nicodem J Govella & Fredros Okumu & Naveen Bansal & Serdar Bozdag & Jacques D Charlwood & Marta F Maia & Sheila B Ogoma & Floyd E Dowell & George F Corliss & Maggy, 2020. "An autoencoder and artificial neural network-based method to estimate parity status of wild mosquitoes from near-infrared spectra," PLOS ONE, Public Library of Science, vol. 15(6), pages 1-16, June.
    10. Daniel R Jeske, 2018. "Metrics Used When Evaluating the Performance of Statistical Classifiers," Biostatistics and Biometrics Open Access Journal, Juniper Publishers Inc., vol. 8(1), pages 7-9, August.
    11. Juliet Chebet Moso & Stéphane Cormier & Cyril de Runz & Hacène Fouchal & John Mwangi Wandeto, 2021. "Anomaly Detection on Data Streams for Smart Agriculture," Agriculture, MDPI, vol. 11(11), pages 1-17, November.
    12. Kajal Lahiri & Cheng Yang, 2023. "ROC and PRC Approaches to Evaluate Recession Forecasts," Journal of Business Cycle Research, Springer;Centre for International Research on Economic Tendency Surveys (CIRET), vol. 19(2), pages 119-148, September.
    13. Tzu-Hsuan Lin & Jehn-Ruey Jiang, 2021. "Credit Card Fraud Detection with Autoencoder and Probabilistic Random Forest," Mathematics, MDPI, vol. 9(21), pages 1-16, October.
    14. Robert A. Blair & Nicholas Sambanis, 2021. "Is Theory Useful for Conflict Prediction? A Response to Beger, Morgan, and Ward," Journal of Conflict Resolution, Peace Science Society (International), vol. 65(7-8), pages 1427-1453, August.
    15. Mieke Deschepper & Willem Waegeman & Dirk Vogelaers & Kristof Eeckloo, 2020. "Using structured pathology data to predict hospital-wide mortality at admission," PLOS ONE, Public Library of Science, vol. 15(6), pages 1-11, June.
    16. Nicola Lazzarini & Avgoustinos Filippoupolitis & Pedro Manzione & Hariklia Eleftherohorinou, 2022. "A machine learning model on Real World Data for predicting progression to Acute Respiratory Distress Syndrome (ARDS) among COVID-19 patients," PLOS ONE, Public Library of Science, vol. 17(7), pages 1-16, July.
    17. Arijit Paladhi, 2025. "Predicting news deserts using supervised machine learning," Journal of Computational Social Science, Springer, vol. 8(2), pages 1-29, May.
    18. Félix Vandervorst & Wouter Verbeke & Tim Verdonck, 2024. "Claims fraud detection with uncertain labels," Advances in Data Analysis and Classification, Springer;German Classification Society - Gesellschaft für Klassifikation (GfKl);Japanese Classification Society (JCS);Classification and Data Analysis Group of the Italian Statistical Society (CLADAG);International Federation of Classification Societies (IFCS), vol. 18(1), pages 219-243, March.
    19. Apostolos Giannoulidis & Anastasios Gounaris & Athanasios Naskos & Nikodimos Nikolaidis & Daniel Caljouw, 2025. "Engineering and evaluating an unsupervised predictive maintenance solution: a cold-forming press case-study," Journal of Intelligent Manufacturing, Springer, vol. 36(3), pages 2121-2139, March.
    20. Dorian Knoblauch & Jürgen Großmann, 2023. "Towards a Risk-Based Continuous Auditing-Based Certification for Machine Learning," The Review of Socionetwork Strategies, Springer, vol. 17(2), pages 255-273, October.

    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:0271260. 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.