arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.02136cs.LGcs.AIcs.CL
收藏

大语言模型与经典机器学习:在使用高维表格数据预测新冠死亡率中的表现

Large Language Models versus Classical Machine Learning: Performance in COVID-19 Mortality Prediction Using High-Dimensional Tabular Data

Research Institute for Gastroenterology and Liver Diseases, Shahid Beheshti University of Medical Sciences(沙希德·贝赫什提医科大学胃肠病与肝病研究所);Faculty of Medicine, Isfahan University of Medical Sciences(伊斯法罕医科大学医学院);Faculty of Business and Information Technology, Ontario Tech University(安大略理工大学商业与信息技术学院);School of Medicine, Shahid Beheshti University of Medical Sciences(沙希德·贝赫什提医科大学医学院);Digestive Disease Research Institute, Tehran University of Medical Sciences(德黑兰医科大学消化疾病研究所);Department of Surgery, The Johns Hopkins University(约翰霍普金斯大学外科学系);Student Research Committee, School of Nursing and Midwifery, Shahid Beheshti University of Medical Sciences(沙希德·贝赫什提医科大学护理与助产学院学生研究委员会);MPH department, Shiraz University of Medical Sciences(设拉子医科大学公共卫生硕士系);Department of Emergency Medicine, School of Medicine, Safety Promotion and Injury Prevention Research Center, Imam Hossein Hospital, Shahid Beheshti University of Medical Sciences(沙希德·贝赫什提医科大学伊玛目侯赛因医院医学院急诊医学系安全促进与伤害预防研究中心);Karsh Division of Gastroenterology and Hepatology, Cedars-Sinai Medical Center(西达赛奈医疗中心卡什胃肠病与肝病科);Department of Medicine, Stanford University(斯坦福大学医学系);Division of Data Driven and Digital Health (D3M), The Charles Bronfman Institute for Personalized Medicine, Icahn School of Medicine at Mount Sinai(西奈山伊坎医学院查尔斯·布朗夫曼个性化医学研究所数据驱动与数字健康部);Infectious Diseases and Tropical Medicine Research Center, Shahid Beheshti University of Medical Sciences(沙希德·贝赫什提医科大学传染病与热带医学研究中心);Department of Infectious Diseases, Loghman Hakim Hospital, Shahid Beheshti University of Medical Sciences(沙希德·贝赫什提医科大学洛格曼·哈基姆医院传染病科);National Institute for Health and Care Research (NIHR), Nottingham Biomedical Research Centre, Hearing Sciences, Mental Health and Clinical Neurosciences, School of Medicine, University of Nottingham(诺丁汉大学医学院国家健康与护理研究所诺丁汉生物医学研究中心听力科学、心理健康与临床神经科学)
Mohammadreza Ghaffarzadeh-Esfahani, Mahdi Ghaffarzadeh-Esfahani, Arian Salahi-Niri, Hossein Toreyhi, Zahra Atf, Amirali Mohsenzadeh-Kermani, Mahshad Sarikhani, … 展开作者

Mohammadreza Ghaffarzadeh-Esfahani, Mahdi Ghaffarzadeh-Esfahani, Arian Salahi-Niri, Hossein Toreyhi, Zahra Atf, Amirali Mohsenzadeh-Kermani, Mahshad Sarikhani, Zohreh Tajabadi, Fatemeh Shojaeian, Mohammad Hassan Bagheri, Aydin Feyzi, Mohammadamin Tarighatpayma, Narges Gazmeh, Fateme Heydari, Hossein Afshar, Amirreza Allahgholipour, Farid Alimardani, Ameneh Salehi, Naghmeh Asadimanesh, Mohammad Amin Khalafi, Hadis Shabanipour, Ali Moradi, Sajjad Hossein Zadeh, Omid Yazdani, Romina Esbati, Moozhan Maleki, Danial Samiei Nasr, Amirali Soheili, Hossein Majlesi, Saba Shahsavan, Alireza Soheilipour, Nooshin Goudarzi, Erfan Taherifard, Hamidreza Hatamabadi, Jamil S Samaan, Thomas Savage, Ankit Sakhuja, Ali Soroush, Girish Nadkarni, Ilad Alavi Darazam, Mohamad Amin Pourhoseingholi, Seyed Amir Ahmad Safavi-Naini

AI总结:

本文比较了经典特征机器学习模型与大语言模型在预测新冠死亡率中的性能,发现经典模型在处理高维表格数据方面仍占优势,但通过微调大语言模型可显著提升其效果。

摘要:

本研究比较了经典特征基于机器学习模型(CMLs)和大语言模型(LLMs)在使用来自四个医院9134名患者高维表格数据预测新冠死亡率中的性能。评估了七种CML模型,包括XGBoost和随机森林(RF),以及八种LLM,如GPT-4和Mistral-7b,这些模型在文本转换的结构数据上进行了零样本分类。此外,Mistral-7b使用QLoRA方法进行了微调。XGBoost和RF在CML中表现出色,分别在内部和外部验证中获得F1分数0.87和0.83。GPT-4在LLM类别中领先,F1分数为0.43,而微调Mistral-7b显著提高了其召回率从1%到79%,在外部验证中获得稳定的F1分数0.74。尽管LLM在零样本分类中表现一般,但微调显著提升了其效果,可能缩小与CML模型的差距。然而,CML在处理高维表格数据任务中仍占优势。本研究强调了CML和微调LLM在医学预测建模中的潜力,同时强调CML在结构数据分析中的当前优势。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑