Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Knowledge Graph Augmented Large Language Models for Next-Visit Disease Prediction

Created by
  • Haebom
Category
Empty

저자

Ruiyu Wang, Tuan Vinh, Ran Xu, Yuyin Zhou, Jiaying Lu, Carl Yang, Francisco Pasquel

개요

본 논문은 전자 건강 기록(EHR) 기반의 질병 예측 모델에서 환자 수준의 의사 결정을 지원하기 위해, 지식 그래프(KG) 기반의 Chain-of-Thought(CoT) 프레임워크를 제안합니다. MIMIC-III 데이터를 활용하여 질병 관련 노드와 추론 경로를 추출하고, 이를 기반으로 임상적으로 타당하고 시간적으로 일관된 설명을 생성합니다. LLaMA-3.1-Instruct-8B와 Gemma-7B 모델을 미세 조정하여 기존 모델 대비 높은 성능을 보였으며, CRADLE 코호트에도 제로샷으로 적용 가능함을 입증했습니다. 또한, 의료진 평가를 통해 제안하는 KG 기반 CoT 설명의 명확성, 관련성, 임상적 정확성이 높음을 확인했습니다.

시사점, 한계점

시사점:
KG를 활용하여 EHR 기반 예측 모델의 설명 가능성을 향상시킴.
Chain-of-Thought 방식을 통해 환자 수준의 의사 결정에 기여할 수 있는 임상적 근거를 제공.
다양한 모델 및 데이터셋에서 우수한 성능을 보임.
의료진 평가를 통해 제안 모델의 설명의 질을 검증.
제로샷 학습을 통해 다른 데이터셋에도 적용 가능성을 제시.
한계점:
AUROC 0.66-0.70, macro-AUPR 0.40-0.47으로, 아직 개선의 여지가 있음.
훈련 데이터의 크기가 제한적(400, 1000 케이스).
모델의 일반화 성능에 대한 추가적인 연구 필요.
PrimeKG에 의존적인 면모가 있음.
👍