arXiv:2502.18992cs.IR2025-02中稿 · as a workshop pape…被引 5

用知识图谱增强大模型,让医学编码映射更准更快。

OntologyRAG: Better and Faster Biomedical Code Mapping with Retrieval-Augmented Generation (RAG) Leveraging Ontology Knowledge Graphs and Large Language Models

  • 利用本体知识图谱为大模型提供上下文,实现无需重训练的智能映射
  • 生成带推理过程和匹配度评估的结果,提升专家决策效率
  • 适合需要频繁更新的医学本体映射任务,降低人工成本

生物医学本体系统全面定义了生物医学实体的概念与关系,对领域信息结构化至关重要。生物医学编码映射旨在识别不同本体间概念的相似性或等价性。高质量映射通常依赖领域微调的语言模型生成初步候选结果,再由具备丰富领域知识的编码专家手动筛选或修正。现有语言模型仅提供无推理依据的候选列表,专家仍需逐一核对本体源数据以确定最优匹配。由于本体持续更新以纳入新研究发现,该流程需反复重训练模型并人工干预,耗时且费力。本文提出OntologyRAG,一种基于本体知识图谱增强的检索增强生成方法,利用本体的归纳偏置支持大模型的上下文学习。该方法将大模型锚定于本体间的未精炼映射关系,通过生成包含推理逻辑与匹配度评估的可解释结果,实现无需重训练的映射更新。只需更新知识图谱即可同步本体变更。在自建黄金数据集上的评估表明,该方法能帮助编码专家更高效、更准确地完成编码映射任务。代码已开源:https://github.com/iqvianlp/ontologyRAG。

原文摘要 · Abstract (English)

Biomedical ontologies, which comprehensively define concepts and relations for biomedical entities, are crucial for structuring and formalizing domain-specific information representations. Biomedical code mapping identifies similarity or equivalence between concepts from different ontologies. Obtaining high-quality mapping usually relies on automatic generation of unrefined mapping with ontology domain fine-tuned language models (LMs), followed by manual selections or corrections by coding experts who have extensive domain expertise and familiarity with ontology schemas. The LMs usually provide unrefined code mapping suggestions as a list of candidates without reasoning or supporting evidence, hence coding experts still need to verify each suggested candidate against ontology sources to pick the best matches. This is also a recurring task as ontology sources are updated regularly to incorporate new research findings. Consequently, the need of regular LM retraining and manual refinement make code mapping time-consuming and labour intensive. In this work, we created OntologyRAG, an ontology-enhanced retrieval-augmented generation (RAG) method that leverages the inductive biases from ontological knowledge graphs for in-context-learning (ICL) in large language models (LLMs). Our solution grounds LLMs to knowledge graphs with unrefined mappings between ontologies and processes questions by generating an interpretable set of results that include prediction rational with mapping proximity assessment. Our solution doesn't require re-training LMs, as all ontology updates could be reflected by updating the knowledge graphs with a standard process. Evaluation results on a self-curated gold dataset show promises of using our method to enable coding experts to achieve better and faster code mapping. The code is available at https://github.com/iqvianlp/ontologyRAG.

知识图谱大模型医学编码RAG

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