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

Suboptimal capability of individual machine learning algorithms in modeling small-scale imbalanced clinical data of local hospital

Author

Listed:
  • Gang Li
  • Chenbi Li
  • Chengli Wang
  • Zeheng Wang

Abstract

In recent years, artificial intelligence (AI) has shown promising applications in various scientific domains, including biochemical analysis research. However, the effectiveness of AI in modeling small-scale, imbalanced datasets remains an open question in such fields. This study explores the capabilities of eight basic AI algorithms, including ridge regression, logistic regression, random forest regression, and others, in modeling a small, imbalanced clinical dataset (total n = 387, class 0 = 27, class 1 = 360) related to the records of the biochemical blood tests from the patients with multiple wasp stings (MWS). Through rigorous evaluation using k-fold cross-validation and comprehensive scoring, we found that none of the models could effectively model the data. Even after fine-tuning the hyperparameters of the best-performing models, the results remained below acceptable thresholds. The study highlights the challenges of applying AI to small-scale datasets with imbalanced groups in biochemical or clinical research and emphasizes the need for novel algorithms tailored to small-scale data. The findings also call for further exploration into techniques such as transfer learning and data augmentation, and they underline the importance of understanding the minimum dataset scale required for effective AI modeling in biochemical contexts.

Suggested Citation

  • Gang Li & Chenbi Li & Chengli Wang & Zeheng Wang, 2024. "Suboptimal capability of individual machine learning algorithms in modeling small-scale imbalanced clinical data of local hospital," PLOS ONE, Public Library of Science, vol. 19(2), pages 1-13, February.
  • Handle: RePEc:plo:pone00:0298328
    DOI: 10.1371/journal.pone.0298328
    as

    Download full text from publisher

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

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

    File URL: https://libkey.io/10.1371/journal.pone.0298328?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. Kathryn Tunyasuvunakool & Jonas Adler & Zachary Wu & Tim Green & Michal Zielinski & Augustin Žídek & Alex Bridgland & Andrew Cowie & Clemens Meyer & Agata Laydon & Sameer Velankar & Gerard J. Kleywegt, 2021. "Highly accurate protein structure prediction for the human proteome," Nature, Nature, vol. 596(7873), pages 590-596, August.
    2. JungHo Kong & Doyeon Ha & Juhun Lee & Inhae Kim & Minhyuk Park & Sin-Hyeog Im & Kunyoo Shin & Sanguk Kim, 2022. "Network-based machine learning approach to predict immunotherapy response in cancer patients," Nature Communications, Nature, vol. 13(1), pages 1-15, December.
    3. John Jumper & Richard Evans & Alexander Pritzel & Tim Green & Michael Figurnov & Olaf Ronneberger & Kathryn Tunyasuvunakool & Russ Bates & Augustin Žídek & Anna Potapenko & Alex Bridgland & Clemens Me, 2021. "Highly accurate protein structure prediction with AlphaFold," Nature, Nature, vol. 596(7873), pages 583-589, August.
    4. Andrew W. Senior & Richard Evans & John Jumper & James Kirkpatrick & Laurent Sifre & Tim Green & Chongli Qin & Augustin Žídek & Alexander W. R. Nelson & Alex Bridgland & Hugo Penedones & Stig Petersen, 2020. "Improved protein structure prediction using potentials from deep learning," Nature, Nature, vol. 577(7792), pages 706-710, January.
    5. Lin Huang & Lin Wang & Xiaomeng Hu & Sen Chen & Yunwen Tao & Haiyang Su & Jing Yang & Wei Xu & Vadanasundari Vedarethinam & Shu Wu & Bin Liu & Xinze Wan & Jiatao Lou & Qian Wang & Kun Qian, 2020. "Machine learning of serum metabolic patterns encodes early-stage lung adenocarcinoma," Nature Communications, Nature, vol. 11(1), pages 1-11, December.
    6. Li, Yanting & Jiang, Wenbo & Zhang, Guangyao & Shu, Lianjie, 2021. "Wind turbine fault diagnosis based on transfer learning and convolutional autoencoder with small-scale data," Renewable Energy, Elsevier, vol. 171(C), pages 103-115.
    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. Zachary C. Drake & Justin T. Seffernick & Steffen Lindert, 2022. "Protein complex prediction using Rosetta, AlphaFold, and mass spectrometry covalent labeling," Nature Communications, Nature, vol. 13(1), pages 1-9, December.
    2. Nicolae Sapoval & Amirali Aghazadeh & Michael G. Nute & Dinler A. Antunes & Advait Balaji & Richard Baraniuk & C. J. Barberan & Ruth Dannenfelser & Chen Dun & Mohammadamin Edrisi & R. A. Leo Elworth &, 2022. "Current progress and open challenges for applying deep learning across the biosciences," Nature Communications, Nature, vol. 13(1), pages 1-12, December.
    3. Niklas W. A. Gebauer & Michael Gastegger & Stefaan S. P. Hessmann & Klaus-Robert Müller & Kristof T. Schütt, 2022. "Inverse design of 3d molecular structures with conditional generative neural networks," Nature Communications, Nature, vol. 13(1), pages 1-11, December.
    4. Hajkowicz, Stefan & Naughtin, Claire & Sanderson, Conrad & Schleiger, Emma & Karimi, Sarvnaz & Bratanova, Alexandra & Bednarz, Tomasz, 2022. "Artificial intelligence for science – adoption trends and future development pathways," MPRA Paper 115464, University Library of Munich, Germany.
    5. Qiufen Chen & Yuanzhao Guo & Jiuhong Jiang & Jing Qu & Li Zhang & Han Wang, 2023. "The Relative Distance Prediction of Transmembrane Protein Surface Residue Based on Improved Residual Networks," Mathematics, MDPI, vol. 11(3), pages 1-16, January.
    6. Agnese I. Curatolo & Ofer Kimchi & Carl P. Goodrich & Ryan K. Krueger & Michael P. Brenner, 2023. "A computational toolbox for the assembly yield of complex and heterogeneous structures," Nature Communications, Nature, vol. 14(1), pages 1-13, December.
    7. Noelia Ferruz & Steffen Schmidt & Birte Höcker, 2022. "ProtGPT2 is a deep unsupervised language model for protein design," Nature Communications, Nature, vol. 13(1), pages 1-10, December.
    8. Aaron Gupta & Kevin S. Kao & Rachel Yamin & Deena A. Oren & Yehuda Goldgur & Jonathan Du & Pete Lollar & Eric J. Sundberg & Jeffrey V. Ravetch, 2023. "Mechanism of glycoform specificity and in vivo protection by an anti-afucosylated IgG nanobody," Nature Communications, Nature, vol. 14(1), pages 1-11, December.
    9. Lei Wang & Jiangguo Zhang & Dali Wang & Chen Song, 2022. "Membrane contact probability: An essential and predictive character for the structural and functional studies of membrane proteins," PLOS Computational Biology, Public Library of Science, vol. 18(3), pages 1-27, March.
    10. Gustavo Arango-Argoty & Elly Kipkogei & Ross Stewart & Gerald J. Sun & Arijit Patra & Ioannis Kagiampakis & Etai Jacob, 2025. "Pretrained transformers applied to clinical studies improve predictions of treatment efficacy and associated biomarkers," Nature Communications, Nature, vol. 16(1), pages 1-18, December.
    11. Jong Woo Bae & Sangtae Kim & V. Narry Kim & Jong-Seo Kim, 2021. "Photoactivatable ribonucleosides mark base-specific RNA-binding sites," Nature Communications, Nature, vol. 12(1), pages 1-10, December.
    12. Erika Erickson & Japheth E. Gado & Luisana Avilán & Felicia Bratti & Richard K. Brizendine & Paul A. Cox & Raj Gill & Rosie Graham & Dong-Jin Kim & Gerhard König & William E. Michener & Saroj Poudel &, 2022. "Sourcing thermotolerant poly(ethylene terephthalate) hydrolase scaffolds from natural diversity," Nature Communications, Nature, vol. 13(1), pages 1-15, December.
    13. Zhiye Guo & Jian Liu & Jeffrey Skolnick & Jianlin Cheng, 2022. "Prediction of inter-chain distance maps of protein complexes with 2D attention-based deep neural networks," Nature Communications, Nature, vol. 13(1), pages 1-10, December.
    14. Nicolas Renaud & Cunliang Geng & Sonja Georgievska & Francesco Ambrosetti & Lars Ridder & Dario F. Marzella & Manon F. Réau & Alexandre M. J. J. Bonvin & Li C. Xue, 2021. "DeepRank: a deep learning framework for data mining 3D protein-protein interfaces," Nature Communications, Nature, vol. 12(1), pages 1-8, December.
    15. Ye Yuan & Lei Chen & Kexu Song & Miaomiao Cheng & Ling Fang & Lingfei Kong & Lanlan Yu & Ruonan Wang & Zhendong Fu & Minmin Sun & Qian Wang & Chengjun Cui & Haojue Wang & Jiuyang He & Xiaonan Wang & Y, 2024. "Stable peptide-assembled nanozyme mimicking dual antifungal actions," Nature Communications, Nature, vol. 15(1), pages 1-17, December.
    16. Ivica Odorčić & Mohamed Belal Hamed & Sam Lismont & Lucía Chávez-Gutiérrez & Rouslan G. Efremov, 2024. "Apo and Aβ46-bound γ-secretase structures provide insights into amyloid-β processing by the APH-1B isoform," Nature Communications, Nature, vol. 15(1), pages 1-14, December.
    17. Surabhi Kokane & Ashutosh Gulati & Pascal F. Meier & Rei Matsuoka & Tanadet Pipatpolkai & Giuseppe Albano & Tin Manh Ho & Lucie Delemotte & Daniel Fuster & David Drew, 2025. "PIP2-mediated oligomerization of the endosomal sodium/proton exchanger NHE9," Nature Communications, Nature, vol. 16(1), pages 1-17, December.
    18. Stella Vitt & Simone Prinz & Martin Eisinger & Ulrich Ermler & Wolfgang Buckel, 2022. "Purification and structural characterization of the Na+-translocating ferredoxin: NAD+ reductase (Rnf) complex of Clostridium tetanomorphum," Nature Communications, Nature, vol. 13(1), pages 1-11, December.
    19. Pierre Azoulay & Joshua Krieger & Abhishek Nagaraj, 2024. "Old Moats for New Models: Openness, Control, and Competition in Generative Artificial Intelligence," NBER Chapters, in: Entrepreneurship and Innovation Policy and the Economy, volume 4, pages 7-46, National Bureau of Economic Research, Inc.
    20. Riya Shah & Thomas C. Panagiotou & Gregory B. Cole & Trevor F. Moraes & Brigitte D. Lavoie & Christopher A. McCulloch & Andrew Wilde, 2024. "The DIAPH3 linker specifies a β-actin network that maintains RhoA and Myosin-II at the cytokinetic furrow," Nature Communications, Nature, vol. 15(1), pages 1-17, December.

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