用稀疏专家模型提升医疗大模型效率,训练成本更低。
SparseDoctor: Towards Efficient Chat Doctor with Mixture of Experts Enhanced Large Language Models
- 采用对比学习增强的低秩适配专家混合架构,实现高效参数更新。
- 在三个医学基准上优于华佗系列模型,性能更优且资源消耗更低。
- 适合需要低成本部署的医疗AI应用,如智能问诊系统。
大型语言模型在医学问答和临床决策中表现卓越,推动了个性化虚拟医生的普及。然而,传统微调策略需更新数十亿参数,显著增加训练成本。为提升医疗大模型的效率与表征能力,本文提出一种新型稀疏医疗大模型SparseDoctor,其基于对比学习增强的LoRA-MoE(低秩适配-专家混合)架构。通过自动路由机制,科学分配计算资源至不同受对比学习监督的LoRA专家。此外,引入专家记忆队列机制,进一步提升框架效率并防止训练过程中的内存溢出。在CMB、CMExam和CMMLU-Med三个典型医学基准上进行综合评估,实验结果表明,该模型持续优于HuatuoGPT系列等强基线模型。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved great success in medical question answering and clinical decision-making, promoting the efficiency and popularization of the personalized virtual doctor in society. However, the traditional fine-tuning strategies on LLM require the updates of billions of parameters, substantially increasing the training cost, including the training time and utility cost. To enhance the efficiency and effectiveness of the current medical LLMs and explore the boundary of the representation capability of the LLMs on the medical domain, apart from the traditional fine-tuning strategies from the data perspective (i.e., supervised fine-tuning or reinforcement learning from human feedback), we instead craft a novel sparse medical LLM named SparseDoctor armed with contrastive learning enhanced LoRA-MoE (low rank adaptation-mixture of experts) architecture. To this end, the crafted automatic routing mechanism can scientifically allocate the computational resources among different LoRA experts supervised by the contrastive learning. Additionally, we also introduce a novel expert memory queue mechanism to further boost the efficiency of the overall framework and prevent the memory overflow during training. We conduct comprehensive evaluations on three typical medical benchmarks: CMB, CMExam, and CMMLU-Med. Experimental results demonstrate that the proposed LLM can consistently outperform the strong baselines such as the HuatuoGPT series.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。