Large language models versus classical machine learning performance in COVID-19 mortality prediction using high-dimensional tabular data
This study compared the performance of classical feature-based machine learning models (CMLs) and large language models (LLMs) in predicting COVID-19 mortality using high-dimensional tabular data from 9,134 patients across four hospitals. Seven CML models, including XGBoost and random forest (RF), were evaluated alongside eight LLMs, such as GPT-4 and Mistral-7b, which performed zero-shot classification on text-converted structured data. Additionally, Mistral-7b was fine-tuned using the QLoRA approach. XGBoost and RF demonstrated superior performance among CMLs, achieving F1 scores of 0.87 and 0.83 for internal and external validation, respectively. GPT-4 led the LLM category with an F1 score of 0.43, while fine-tuning Mistral-7b significantly improved its recall from 1% to 79%, yielding a stable F1 score of 0.74 during external validation. Although LLMs showed moderate performance in zero-shot classification, fine-tuning substantially enhanced their effectiveness, potentially bridging the gap with CML models. However, CMLs still outperformed LLMs in handling high-dimensional tabular data tasks. This study highlights the potential of both CMLs and fine-tuned LLMs in medical predictive modeling, while emphasizing the current superiority of CMLs for structured data analysis.
Authors
- Amirali Soheili (ORCID: https://orcid.org/0000-0002-7143-2818)
- Hamidreza Hatamabadi (ORCID: https://orcid.org/0000-0002-9085-8806)
- Saba Shahsavan (ORCID: https://orcid.org/0000-0002-2838-411X)
- Hossein Majlesi (ORCID: https://orcid.org/0000-0003-1060-9955)
- Zohreh Tajabadi
- Fatemeh Shojaeian (ORCID: https://orcid.org/0000-0001-5972-9953)
- Zahra Atf (ORCID: https://orcid.org/0000-0003-0642-4341)
- Hossein Toreyhi (ORCID: https://orcid.org/0000-0002-4856-6674)
- Omid Yazdani (ORCID: https://orcid.org/0009-0004-2187-3852)
- Ilad Alavi Darazam (ORCID: https://orcid.org/0000-0002-4440-335X)
- Danial Samiei Nasr (ORCID: https://orcid.org/0000-0002-2789-9226)
- Hossein Afshar
- Girish N. Nadkarni (ORCID: https://orcid.org/0000-0001-6319-4314)
- Jamil S. Samaan (ORCID: https://orcid.org/0000-0002-6191-2631)
- Fatemeh Heydari (ORCID: https://orcid.org/0000-0003-3171-7744)
- Mohamad Amin Pourhoseingholi (ORCID: https://orcid.org/0000-0002-0121-8031)
- Erfan Taherifard (ORCID: https://orcid.org/0000-0002-9101-0321)
- Naghmeh Asadimanesh (ORCID: https://orcid.org/0000-0002-1295-6574)
- Mahshad Sarikhani
- Amirreza Allahgholipour (ORCID: https://orcid.org/0000-0002-7842-9398)
- Ankit Sakhuja (ORCID: https://orcid.org/0000-0002-2045-518X)
- Thomas Savage (ORCID: https://orcid.org/0000-0001-8715-8369)
- Ali Soroush (ORCID: https://orcid.org/0000-0001-6900-5596)
- Mohammad Khalafi (ORCID: https://orcid.org/0000-0002-9447-5167)
- Seyed Amir Ahmad Safavi‐Naini (ORCID: https://orcid.org/0000-0001-9295-9283)
- Ali Moradi (ORCID: https://orcid.org/0000-0003-4169-0354)
- Romina Esbati (ORCID: https://orcid.org/0000-0002-7161-5948)
- Aryan Salahi‐Niri (ORCID: https://orcid.org/0000-0002-6772-0972)
- Mohammadreza Ghaffarzadeh-Esfahani
- Narges Gazmeh
- Mahdi Ghaffarzadeh-Esfahani
- Hadis Shabanipour
- Alireza Soheilipour
- Amirali Mohsenzadeh-Kermani
- Moozhan Maleki
- Ameneh Salehi
- Sajjad Hossein Zadeh
- Aydin Feyzi (ORCID: https://orcid.org/0000-0001-6152-1342)
- Mohammad Hassan Bagheri
- Farid Alimardani
- Nooshin Goudarzi
- Mohamadamin Tarighat-Payma
Institutions
- Isfahan University of Medical Sciences (IR)
- Cedars-Sinai Medical Center (US)
- Johns Hopkins University (US)
- Shiraz University of Medical Sciences (IR)
- Ontario Tech University (CA)
- Iranshahr University (IR)
- Research Institute for Endocrine Sciences (IR)
- Child Health and Development Institute (US)
- NIHR Nottingham Hearing Biomedical Research Unit (GB)
- Nottingham Biomedical Research Centre (GB)
- Imam Hossein Hospital (IR)
- Shahid Beheshti University of Medical Sciences (IR)
- Tehran University of Medical Sciences (IR)
- Stanford University (US)
- Icahn School of Medicine at Mount Sinai (US)
Publication Details
- Journal
- Scientific Reports
- Published
- 2025-11-28
- DOI
- https://doi.org/10.1038/s41598-025-26705-7
- Citations
- 1
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 1.36