Integrating UMLS Knowledge into Large Language Models for Medical Question Answering

• Published in: arXiv.org
• Main Authors: Yang, Rui, Marrese-Taylor, Edison, Ke, Yuhe, Cheng, Lechao, Chen, Qingyu, Li, Irene
• Format: Paper Journal Article
• Language: English
• Published: Ithaca Cornell University Library, arXiv.org 13.10.2023
• Subjects: Chatbots; Completeness; Computer Science - Artificial Intelligence; Computer Science - Computation and Language; Health care; Large language models; Physicians; Questions; Test sets
• ISSN: 2331-8422
• DOI: 10.48550/arxiv.2310.02778

Large language models (LLMs) have demonstrated powerful text generation capabilities, bringing unprecedented innovation to the healthcare field. While LLMs hold immense promise for applications in healthcare, applying them to real clinical scenarios presents significant challenges, as these models may generate content that deviates from established medical facts and even exhibit potential biases. In our research, we develop an augmented LLM framework based on the Unified Medical Language System (UMLS), aiming to better serve the healthcare community. We employ LLaMa2-13b-chat and ChatGPT-3.5 as our benchmark models, and conduct automatic evaluations using the ROUGE Score and BERTScore on 104 questions from the LiveQA test set. Additionally, we establish criteria for physician-evaluation based on four dimensions: Factuality, Completeness, Readability and Relevancy. ChatGPT-3.5 is used for physician evaluation with 20 questions on the LiveQA test set. Multiple resident physicians conducted blind reviews to evaluate the generated content, and the results indicate that this framework effectively enhances the factuality, completeness, and relevance of generated content. Our research demonstrates the effectiveness of using UMLS-augmented LLMs and highlights the potential application value of LLMs in in medical question-answering.