MedKPL:一种用于可迁移诊断的异构知识增强提示学习框架。

MedKPL: A heterogeneous knowledge enhanced prompt learning framework for transferable diagnosis.

机构信息

Department of Big Data and Biomedical AI, College of Future Technology, Peking University, Beijing 100091, China.

Department of Computer Science and Technology, Tsinghua University, Beijing 100084, China.

出版信息

J Biomed Inform. 2023 Jul;143:104417. doi: 10.1016/j.jbi.2023.104417. Epub 2023 Jun 12.

Abstract

Artificial Intelligence (AI) based diagnosis systems have emerged as powerful tools to reform traditional medical care. Each clinician now wants to have his own intelligent diagnostic partner to expand the range of services he can provide. However, the implementation of intelligent decision support systems based on clinical note has been hindered by the lack of extensibility of end-to-end AI diagnosis algorithms. When reading a clinical note, expert clinicians make inferences with relevant medical knowledge, which serve as prompts for making accurate diagnoses. Therefore, external medical knowledge is commonly employed as an augmentation for medical text classification tasks. Existing methods, however, cannot integrate knowledge from various knowledge sources as prompts nor can fully utilize explicit and implicit knowledge. To address these issues, we propose a Medical Knowledge-enhanced Prompt Learning (MedKPL) diagnostic framework for transferable clinical note classification. Firstly, to overcome the heterogeneity of knowledge sources, such as knowledge graphs or medical QA databases, MedKPL uniform the knowledge relevant to the disease into text sequences of fixed format. Then, MedKPL integrates medical knowledge into the prompt designed for context representation. Therefore, MedKPL can integrate knowledge into the models to enhance diagnostic performance and effectively transfer to new diseases by using relevant disease knowledge. The results of our experiments on two medical datasets demonstrate that our method yields superior medical text classification results and performs better in cross-departmental transfer tasks under few-shot or even zero-shot settings. These findings demonstrate that our MedKPL framework has the potential to improve the interpretability and transferability of current diagnostic systems.

摘要

人工智能(AI)为基础的诊断系统已经成为改革传统医疗保健的有力工具。现在,每位临床医生都希望拥有自己的智能诊断伙伴,以扩大他能够提供的服务范围。然而,基于临床记录的智能决策支持系统的实施受到端到端 AI 诊断算法缺乏可扩展性的阻碍。当阅读临床记录时,专家临床医生会利用相关医学知识进行推理,这些推理作为做出准确诊断的提示。因此,通常将外部医学知识作为医学文本分类任务的增强。然而,现有的方法既不能将来自各种知识源的知识集成作为提示,也不能充分利用显式和隐式知识。为了解决这些问题,我们提出了一种可转移临床记录分类的医学知识增强提示学习(MedKPL)诊断框架。首先,为了克服知识源(如知识图谱或医学 QA 数据库)的异构性,MedKPL 将与疾病相关的知识统一为固定格式的文本序列。然后,MedKPL 将医学知识集成到用于上下文表示的提示中。因此,MedKPL 可以将知识集成到模型中,以提高诊断性能,并通过使用相关疾病知识,有效地转移到新的疾病。我们在两个医疗数据集上的实验结果表明,我们的方法在医学文本分类方面取得了优越的结果,并在少样本甚至零样本设置下的跨科室转移任务中表现更好。这些发现表明,我们的 MedKPL 框架有可能提高当前诊断系统的可解释性和可转移性。

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索