MedKGEval團隊 投稿
量子位 | 公眾號 QbitAI

醫(yī)療大模型知識覆蓋度首次被精準量化!

在醫(yī)療領(lǐng)域,大語言模型(LLM)的潛力令人振奮,但其知識儲備是否足夠可靠?騰訊優(yōu)圖實驗室天衍研究中心的最新研究給出了答案。

他們提出的MedKGEval框架,首次通過醫(yī)療知識圖譜(KG)的多層級評估,系統(tǒng)揭示了GPT-4o等主流模型的醫(yī)學(xué)知識覆蓋度。

該研究已被WWW 2025會議Web4Good Track錄用為口頭報告(oral)。目前,WWW 2025正在悉尼舉行,會議時間從4月28日持續(xù)至5月2日。

打開網(wǎng)易新聞 查看精彩圖片

背景

背景

大語言模型(LLM)在醫(yī)療領(lǐng)域的快速發(fā)展凸顯了其知識存儲與處理的潛力,但其臨床部署前的可靠性驗證亟需更系統(tǒng)化的評估框架。

當(dāng)前主流的Prompt-CBLUE、Medbench和MedJourney等評估體系雖通過醫(yī)學(xué)問答基準測試LLM的任務(wù)執(zhí)行能力,卻存在三個明顯的局限:

1)其長尾數(shù)據(jù)分布導(dǎo)致罕見病癥覆蓋不足,評測結(jié)果存在偏差;

2)任務(wù)導(dǎo)向的設(shè)計聚焦疾病預(yù)測、用藥咨詢等單一場景,難以量化模型內(nèi)在醫(yī)學(xué)知識儲量;

3)傳統(tǒng)問答形式局限于表面對錯判斷,無法捕捉醫(yī)學(xué)概念間的復(fù)雜拓撲關(guān)聯(lián)。

為解決這些問題,本文提出基于醫(yī)療知識圖譜(KG)的多層級評估框架MedKGEval。

醫(yī)療KG通過結(jié)構(gòu)化存儲復(fù)雜實體關(guān)系網(wǎng)絡(luò),為評估提供天然基準??蚣軇?chuàng)新性地設(shè)計三級評估體系:實體層評估醫(yī)學(xué)概念理解,關(guān)系層檢驗醫(yī)學(xué)關(guān)聯(lián)區(qū)分能力,子圖層驗證結(jié)構(gòu)化推理水平。

通過真?zhèn)闻袛嗪投噙x題形式,同時實現(xiàn)任務(wù)導(dǎo)向(task-oriented)的粗粒度性能評估與知識導(dǎo)向(knowledge-oriented)的細粒度三重覆蓋度測量(實體/關(guān)系/知識三元組)。

醫(yī)療知識覆蓋度評估框架MedKGEval

醫(yī)療知識覆蓋度評估框架MedKGEval

在MedKGEval中研究團隊設(shè)計了多層級的任務(wù)體系,其中包含3個層級的9項核心任務(wù),通過真?zhèn)闻袛啵═FQ)與多選題(MCQ)任務(wù)形式,實現(xiàn)任務(wù)導(dǎo)向與知識導(dǎo)向的雙重評測。

具體評估流程框架見下圖。

打開網(wǎng)易新聞 查看精彩圖片

任務(wù)架構(gòu)設(shè)計

任務(wù)架構(gòu)設(shè)計

基于醫(yī)療知識圖譜的實體、關(guān)系、三元組結(jié)構(gòu),構(gòu)建三級評估體系:

實體層面(3項任務(wù)):驗證醫(yī)學(xué)概念理解

  • 實體類型標(biāo)注(ET):通過多選題識別“糖尿病”等實體的分類標(biāo)簽(如疾病/癥狀)
  • 實體聚類(EC):從5個實體中辨識類型異常項(如混入癥狀類別的藥物實體)
  • 實體消歧(ED):判斷兩個實體是否等價,比如“阿司匹林”與“乙酰水楊酸”是否為等價實體

關(guān)系層面(3項任務(wù)):檢驗醫(yī)學(xué)關(guān)聯(lián)認知

  • 關(guān)系類型標(biāo)注(RT):選擇“并發(fā)癥”關(guān)系可連接的實體類型對(如疾病→疾?。?/li>
  • 事實核驗(FC):判斷三元組的真?zhèn)危热纭安悸宸?治療-偏頭痛”
  • 關(guān)系預(yù)測(RP):補全實體之間缺失的關(guān)系,比如“冠狀動脈硬化→(?)→心肌梗死”

子圖層面(3項任務(wù)):評估結(jié)構(gòu)化推理

  • 錯誤識別(ER):從5個三元組中檢測異常項(如錯誤藥物禁忌關(guān)系)
  • 子圖推理1(R1):基于多跳關(guān)系推理,比如基于“高血壓→并發(fā)癥→腦出血→影像檢查→CT”路徑,推斷“高血壓→影像檢查→CT”是否成立
  • 子圖推理2(R2):在相同推理鏈中,從候選關(guān)系中選擇正確關(guān)聯(lián)

打開網(wǎng)易新聞 查看精彩圖片

隨著利用的KG信息增多,任務(wù)難度也在逐漸升高,這樣階梯式、多層級的評估更有利用全面了解LLMs的性能。

任務(wù)導(dǎo)向和知識導(dǎo)向的評估機制

任務(wù)導(dǎo)向和知識導(dǎo)向的評估機制

在每項任務(wù)中均配備評估核心實體/關(guān)系映射(如上圖 core E and R),實現(xiàn)細粒度知識覆蓋分析:

任務(wù)導(dǎo)向評估:計算準確率指標(biāo)

知識導(dǎo)向評估:

  • 實體覆蓋率:實體正確率均值(CovAvg-E)、引入節(jié)點中心度加權(quán)(CovDeg-E)
  • 關(guān)系覆蓋率:關(guān)系正確率均值(CovAvg-R)、按關(guān)系出現(xiàn)頻次加權(quán)(CovDeg-R)
  • 三元組覆蓋率Cov-T:反映知識單元整體掌握度
實驗及評估結(jié)果

實驗及評估結(jié)果

MedKGEval選用中文醫(yī)療領(lǐng)域主流知識圖譜CPubMedKG和CMeKG作為基準,經(jīng)下采樣構(gòu)建實驗數(shù)據(jù)集。

評估模型涵蓋三大類:1)開源通用模型;2)醫(yī)療垂類模型;3)閉源模型。

下表展示了11個LLM的任務(wù)導(dǎo)向評估結(jié)果,可以看到:GPT-4o以70.65%平均準確率領(lǐng)先;同架構(gòu)LLM參數(shù)量翻倍帶來3-5%準確率提升;大多LLM在實體層面任務(wù)上表現(xiàn)優(yōu)于關(guān)系和子圖層面;通用模型性能超越醫(yī)療垂類模型(歸因分析:垂類模型微調(diào)數(shù)據(jù)側(cè)重具體任務(wù)(如用藥咨詢、醫(yī)患對話摘要),導(dǎo)致醫(yī)學(xué)知識廣度受限)。

打開網(wǎng)易新聞 查看精彩圖片

下表展示了11個LLM的知識導(dǎo)向評估結(jié)果,可以看到:GPT-4o在CPubMedKG (small)上覆蓋了65.66%的實體、55.60%的關(guān)系、62.31%的三元組;更大的參數(shù)量通常會帶來更高的知識覆蓋度;CovAvg 和 CovDeg 的對比體現(xiàn)出了LLM對高關(guān)聯(lián)度實體(如糖尿病)和高頻關(guān)系(如鑒別診斷)的偏好性:CovAvg < CovDeg 說明 LLM 在高關(guān)聯(lián)度實體的上表現(xiàn)更好、反之說明 LLM 在低關(guān)聯(lián)度實體上表現(xiàn)更好。

打開網(wǎng)易新聞 查看精彩圖片

接下來,研究團隊使用MedKGEval評估框架對四個示例LLM在關(guān)聯(lián)度最高的15個實體和最高頻的15個關(guān)系上的知識覆蓋情況進行分析。

以常用臨床實體“超聲”為例,可以看到GPT-4o以94.16%正確率領(lǐng)先,Qwen2-7B(88.83%)、WiNGPT2(85.41%)次之。

打開網(wǎng)易新聞 查看精彩圖片

在醫(yī)學(xué)關(guān)系覆蓋度上,4個LLM也表現(xiàn)出了類似的特點。

打開網(wǎng)易新聞 查看精彩圖片

分析結(jié)果表明,MedKGEval能有效定位LLM在特定醫(yī)學(xué)知識領(lǐng)域的認知缺陷。

這些發(fā)現(xiàn)對模型優(yōu)化具有重要指導(dǎo)價值:如上圖所示,WiNGPT在“肺結(jié)核”實體相關(guān)問答中表現(xiàn)欠佳、Baichuan2-13B在“相關(guān)(轉(zhuǎn)換)”關(guān)系中存在明顯短板。

因此,在下輪微調(diào)中建議針對性補充結(jié)核病診療指南和病理轉(zhuǎn)化機制相關(guān)數(shù)據(jù),通過基于知識缺陷診斷的定向增強策略,可顯著提升醫(yī)療領(lǐng)域LLM的整體性能。

總結(jié)

總結(jié)

本文提出的MedKGEval框架通過醫(yī)療KG視角,構(gòu)建了評估LLM醫(yī)學(xué)知識覆蓋度的多維度體系。

該框架在實體、關(guān)系和子圖三個層級展開評估,系統(tǒng)揭示了當(dāng)前大語言模型在醫(yī)學(xué)知識存儲與推理能力方面的優(yōu)勢與局限。

研究團隊提出的的任務(wù)導(dǎo)向與知識導(dǎo)向雙軌評估機制,不僅能夠精準定位模型的知識薄弱環(huán)節(jié),更為提升醫(yī)療領(lǐng)域LLM的可靠性和臨床應(yīng)用價值提供了量化依據(jù)。


論文地址:https://dl.acm.org/doi/10.1145/3696410.3714535
代碼地址:https://github.com/ZihengZZH/MedKGEval