arXiv:2505.19722cs.CLcs.AI2025-05中稿 · ICIC 2025被引 3

用闭源大模型知识蒸馏出可本地部署的医学实体链接模型

Distilling Closed-Source LLM's Knowledge for Locally Stable and Economic Biomedical Entity Linking

  • 用闭源大模型生成标注数据,微调开源模型做候选重排序
  • 在数据少时提升准确率,中文英文双语数据上分别增0.019和0.036
  • 适合资源有限但需稳定可靠的医学信息处理场景

医学实体链接旨在将非标准实体映射到知识库中的标准实体。传统监督方法表现良好,但需大量标注数据,限制了其在低资源场景的应用。大语言模型(尤其是闭源模型)虽可缓解此问题,却存在稳定性差、经济成本高的缺陷:使用受商业公司限制,处理大规模数据时费用高昂。为此,我们提出RPDR框架,结合闭源与开源大模型,对少量数据微调的检索器返回的候选进行重排序。通过提示闭源模型从无标注数据生成训练数据,并用其微调开源模型实现知识蒸馏,最终得到可在本地部署的模型,避免了稳定性风险和高成本问题。我们在两个数据集上评估,包括一个真实世界数据集和一个包含中英文的公开数据集。当训练数据不足时,RPDR在Aier数据集和Ask A Patient数据集上分别实现Acc@1提升0.019和0.036,证明了该框架的优越性与泛化能力。

原文摘要 · Abstract (English)

Biomedical entity linking aims to map nonstandard entities to standard entities in a knowledge base. Traditional supervised methods perform well but require extensive annotated data to transfer, limiting their usage in low-resource scenarios. Large language models (LLMs), especially closed-source LLMs, can address these but risk stability issues and high economic costs: using these models is restricted by commercial companies and brings significant economic costs when dealing with large amounts of data. To address this, we propose ``RPDR'', a framework combining closed-source LLMs and open-source LLMs for re-ranking candidates retrieved by a retriever fine-tuned with a small amount of data. By prompting a closed-source LLM to generate training data from unannotated data and fine-tuning an open-source LLM for re-ranking, we effectively distill the knowledge to the open-source LLM that can be deployed locally, thus avoiding the stability issues and the problem of high economic costs. We evaluate RPDR on two datasets, including one real-world dataset and one publicly available dataset involving two languages: Chinese and English. RPDR achieves 0.019 Acc@1 improvement and 0.036 Acc@1 improvement on the Aier dataset and the Ask A Patient dataset when the amount of training data is not enough. The results demonstrate the superiority and generalizability of the proposed framework.

医学实体链接知识蒸馏大模型应用低资源

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