With the explosive growth of medical data and the rapid development of artificial intelligence technology, precision medicine has emerged as a key to enhancing the quality and efficiency of healthcare services. In this context, Large Language Models (LLMs) play an increasingly vital role in medical knowledge acquisition and question-answering systems. To further improve the performance of these systems in the medical domain, we introduce an innovative method that jointly trains an Information Retrieval (IR) system and an LLM during the fine-tuning phase. This approach, which we call Joint Medical LLM and Retrieval Training (JMLR), is designed to overcome the challenges faced by traditional models in handling medical question-answering tasks. By employing a synchronized training mechanism, JMLR reduces the demand for computational resources and enhances the model's ability to leverage medical knowledge for reasoning and answering questions. Our experimental results demonstrate that JMLR-13B (81.2% on Amboos, 61.3% on MedQA) outperforms models using conventional pre-training and fine-tuning Meditron-70B (76.4% on AMBOSS, 60.3% on MedQA). For models of the same 7B scale, JMLR-7B(68.7% on Amboos, 51.7% on MedQA) significantly outperforms other public models (Meditron-7B: 50.1%, 47.9%), proving its superiority in terms of cost (our training time: 37 hours, traditional method: 144 hours), efficiency, and effectiveness in medical question-answering tasks. Through this work, we provide a new and efficient knowledge enhancement tool for healthcare, demonstrating the great potential of integrating IR and LLM training in precision medical information retrieval and question-answering systems.
翻译:随着医疗数据的爆炸式增长和人工智能技术的飞速发展,精准医学已成为提升医疗服务质量和效率的关键。在此背景下,大语言模型(LLMs)在医学知识获取与问答系统中发挥着日益重要的作用。为进一步提升这些系统在医学领域的性能,我们提出了一种创新方法——在微调阶段联合训练信息检索(IR)系统与大语言模型。该方法命名为联合医学大语言模型与检索训练(JMLR),旨在克服传统模型在处理医学问答任务时所面临的挑战。通过采用同步训练机制,JMLR降低了对计算资源的需求,并增强了模型利用医学知识进行推理和回答问题的能力。实验结果表明,JMLR-13B(Amboos数据集81.2%,MedQA数据集61.3%)优于采用传统预训练与微调的Meditron-70B(AMBOSS数据集76.4%,MedQA数据集60.3%)。在相同7B规模模型中,JMLR-7B(Amboos数据集68.7%,MedQA数据集51.7%)显著优于其他公开模型(Meditron-7B:50.1%,47.9%),在成本(我们的训练时间:37小时,传统方法:144小时)、效率及医学问答任务有效性方面均展现出其优越性。通过本研究,我们为医疗领域提供了一种新型高效的知识增强工具,证明了在精准医学信息检索与问答系统中整合IR与LLM训练的广阔潜力。