arXiv:2503.04930cs.CLcs.AI2025-03

用医学知识库和大模型生成合成数据,提升生物医学命名实体识别效果

HILGEN: Hierarchically-Informed Data Generation for Biomedical NER Using Knowledgebases and Large Language Models

  • 结合UMLS知识库层级结构与GPT-3.5生成相关实体的合成样本
  • 在四个数据集上使BERT-Large模型平均F1提升42.29%、DANN提升25.03%
  • 无需人工标注,适合少样本场景下的生物医学信息提取任务

我们提出HILGEN,一种基于层次化知识引导的数据生成方法,将统一医学语言系统(UMLS)的领域知识与大语言模型(如GPT-3.5)生成的合成数据相结合。该方法利用UMLS的层级结构扩展训练数据,通过针对性提示词从LLMs中自动生成稀有实体的上下文实例。在MIMIC III、BC5CDR、NCBI-Disease和Med-Mentions四个生物医学命名实体识别数据集上评估,采用BERT-Large和DANN模型进行实验。BERT-Large模型下,仅使用UMLS可实现平均F1提升40.36%,仅使用GPT-3.5达40.52%,最佳集成策略提升42.29%;DANN模型对应结果分别为22.74%、21.53%和25.03%。实验表明,结合历史医学知识与生成式模型可有效提升少样本场景下的命名实体识别性能。

原文摘要 · Abstract (English)

We present HILGEN, a Hierarchically-Informed Data Generation approach that combines domain knowledge from the Unified Medical Language System (UMLS) with synthetic data generated by large language models (LLMs), specifically GPT-3.5. Our approach leverages UMLS's hierarchical structure to expand training data with related concepts, while incorporating contextual information from LLMs through targeted prompts aimed at automatically generating synthetic examples for sparsely occurring named entities. The performance of the HILGEN approach was evaluated across four biomedical NER datasets (MIMIC III, BC5CDR, NCBI-Disease, and Med-Mentions) using BERT-Large and DANN (Data Augmentation with Nearest Neighbor Classifier) models, applying various data generation strategies, including UMLS, GPT-3.5, and their best ensemble. For the BERT-Large model, incorporating UMLS led to an average F1 score improvement of 40.36%, while using GPT-3.5 resulted in a comparable average increase of 40.52%. The Best-Ensemble approach using BERT-Large achieved the highest improvement, with an average increase of 42.29%. DANN model's F1 score improved by 22.74% on average using the UMLS-only approach. The GPT-3.5-based method resulted in a 21.53% increase, and the Best-Ensemble DANN model showed a more notable improvement, with an average increase of 25.03%. Our proposed HILGEN approach improves NER performance in few-shot settings without requiring additional manually annotated data. Our experiments demonstrate that an effective strategy for optimizing biomedical NER is to combine biomedical knowledge curated in the past, such as the UMLS, and generative LLMs to create synthetic training instances. Our future research will focus on exploring additional innovative synthetic data generation strategies for further improving NER performance.

生物医学NER知识增强合成数据大模型应用

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