3.8亿参数模型在手机上跑医学问答,性能超执业医师
MedMobile: A mobile-sized language model with clinical capabilities
- 基于phi-3-mini精简优化,支持移动端运行
- 医学考试准确率达75.7%,超过执业医师及格线
- 最小突破75%的轻量级医疗语言模型,适合资源受限场景
语言模型在医学领域已展现专家级推理与记忆能力,但计算成本高和隐私问题阻碍了大规模应用。为此,我们提出一个精简版phi-3-mini模型——MedMobile,仅38亿参数,可在移动设备上运行,用于临床任务。通过细致的管道改进,我们发现思维链、集成与微调带来最大性能提升,而意外的是检索增强生成未显著改善效果。我们在MultiMedQA和MedBullets数据集上评估其效率。结果显示,MedMobile在MedQA(USMLE)上得分达75.7%,超过执业医师及格线(约60%),并媲美比其大100倍的模型表现。在完整MultiMedQA中,它是参数少于50亿的模型中最优,也是首个通过MedQA(USMLE)的最小模型。该模型有望降低医疗语言模型的使用门槛,实现低算力、快速推理,为临床相关模型发展迈出关键一步。
原文摘要 · Abstract (English)
Language models (LMs) have demonstrated expert-level reasoning and recall abilities in medicine. However, computational costs and privacy concerns are mounting barriers to wide-scale implementation. To address these significant limitations, we introduce a parsimonious adaptation of phi-3-mini, MedMobile, a 3.8 billion parameter LM capable of running on a mobile device, for medical applications. We perform a careful set of pipeline additions and demonstrate that chain of thought, ensembling, and fine-tuning lead to the greatest performance gains, while unexpectedly retrieval augmented generation fails to demonstrate significant improvements. We evaluate the efficiency of our pipeline on the MultiMedQA and MedBullets. We demonstrate that MedMobile scores 75.7% on the MedQA (USMLE), surpassing the passing mark for licensed physicians (~60%) and rivaling scores of models 100 times its size. Across the entirety of the MultiMedQA, MedMobile achieves SOTA performance for models with less than 5B parameters and represents the smallest model to pass the MedQA (USMLE). MedMobile holds promise to democratize access to language models in medicine, bolstering lower compute needs and fast inference speeds. With the ability to combat the biggest barriers to entry for language models in medicine, we hope that MedMobile is a critical step forward in developing clinically relevant language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。