arXiv:2505.10717cs.CLcs.AI2025-05ACL被引 34

用合成数据训练小模型,实现医疗任务高精度表现。

A Modular Approach for Clinical SLMs Driven by Synthetic Data with Pre-Instruction Tuning, Model Merging, and Clinical-Tasks Alignment

  • 先预指令微调专家模型,再通过模型融合统一能力。
  • 在医疗实体识别等任务上提升超40%,超越GPT-4。
  • 适合医疗AI落地,尤其数据稀缺场景下使用。

大语言模型如GPT-4计算成本高、延迟大,限制其在临床环境部署。小型语言模型(SLMs)虽成本低,但需生物医学领域适配,且面临临床数据难获取、敏感性强的问题。为此,我们提出一种新框架,将SLMs转化为高性能临床模型。构建了3.8亿参数的MediPhi系列模型:在医学文献(PMC)、诊疗指南、MedWiki等语料上对专家模型进行预指令微调,结合模型融合与临床任务对齐。为覆盖多数临床任务,将CLUE基准扩展为双倍规模的CLUE+。无需任务特定微调,专家模型在该基准上相对基线提升:医疗实体识别64.3%、放射报告理解49.5%、ICD-10编码44%(优于GPT-4-0125达14%)。通过模型融合整合专家模型,保持跨基准性能优势。此外,构建了包含250万条高质量指令的MediFlow合成数据集,涵盖14项医疗自然语言处理任务、98种细粒度文档类型,并支持JSON格式。利用监督微调与直接偏好优化对MediPhi进行对齐,平均性能再提升18.9%。

原文摘要 · Abstract (English)

High computation costs and latency of large language models such as GPT-4 have limited their deployment in clinical settings. Small language models (SLMs) offer a cost-effective alternative, but their limited capacity requires biomedical domain adaptation, which remains challenging. An additional bottleneck is the unavailability and high sensitivity of clinical data. To address these challenges, we propose a novel framework for adapting SLMs into high-performing clinical models. We introduce the MediPhi collection of 3.8B-parameter SLMs developed with our novel framework: pre-instruction tuning of experts on relevant medical and clinical corpora (PMC, Medical Guideline, MedWiki, etc.), model merging, and clinical-tasks alignment. To cover most clinical tasks, we extended the CLUE benchmark to CLUE+, doubling its size. Our expert models deliver relative improvements on this benchmark over the base model without any task-specific fine-tuning: 64.3% on medical entities, 49.5% on radiology reports, and 44% on ICD-10 coding (outperforming GPT-4-0125 by 14%). We unify the expert models into MediPhi via model merging, preserving gains across benchmarks. Furthermore, we built the MediFlow collection, a synthetic dataset of 2.5 million high-quality instructions on 14 medical NLP tasks, 98 fine-grained document types, and JSON format support. Alignment of MediPhi using supervised fine-tuning and direct preference optimization achieves further gains of 18.9% on average.

小模型医疗AI合成数据模型融合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。