Sign In

SKG-LLM: Developing a Mathematical Model for Stroke Knowledge Graph Construction Using Large Language Models

Created by
  • Haebom
Category
Empty

저자

Ali Sarabadani, Kheirolah Rahsepar Fard, Hamid Dalvand

개요

본 연구는 뇌졸중 관련 논문을 이용하여 지식 그래프(KG)를 구축하는 SKG-LLM 모델을 제시합니다. 수학적 모델과 대규모 언어 모델(LLM)인 GPT-4를 활용하여 생물 의학 문헌에서 복잡한 관계를 추출하고 구성하여 뇌졸중 연구에서 KG의 정확성과 심도를 높입니다. GPT-4를 데이터 전처리 및 임베딩 추출에 사용하여 KG를 구축하였으며, 정밀도와 재현율을 평가 지표로 성능을 측정했습니다. Wikidata와 WN18RR과 비교하여, 특히 정밀도와 재현율에서 우수한 성능을 보였으며, 전문가 검토를 거쳐 정밀도 0.923, 재현율 0.918을 달성했습니다. 구축된 지식 그래프는 13가지 유형의 노드 2692개와 24가지 유형의 에지 5012개로 구성됩니다.

시사점, 한계점

시사점:
GPT-4를 활용한 KG 구축으로 뇌졸중 연구의 정확성과 심도 향상에 기여.
기존 KG 모델(Wikidata, WN18RR) 대비 높은 정밀도와 재현율 달성.
전문가 검토를 통한 성능 개선 가능성 제시.
뇌졸중 연구에 특화된 방대한 지식 그래프 구축.
한계점:
GPT-4에 대한 의존도가 높음. (GPT-4의 성능에 결과가 크게 영향받을 수 있음)
사용된 데이터셋의 한계 (뇌졸중 관련 논문에 국한됨)
모델의 일반화 성능에 대한 추가적인 검증 필요.
다른 유형의 질병이나 의학 분야에 대한 적용 가능성에 대한 추가 연구 필요.
👍