Q-KGR精讀筆記

原文:https://arxiv.org/abs/2410.01401

知識(shí)圖譜問(wèn)答(KGQA)通常先從大規(guī)模的知識(shí)圖譜中檢索出一個(gè)與目標(biāo)問(wèn)題相關(guān)的子圖作為處理查詢的基礎(chǔ)。然而這個(gè)子圖往往包含不相關(guān)的噪聲信息。本文提出了一種知識(shí)圖譜重評(píng)分(Q-KGR)的方法,旨在消除噪聲路徑,使模型能夠更加專注于相關(guān)的知識(shí)。除此之外,本文還引入了一種參數(shù)高效的方法Knowformer,用于將重評(píng)分后的知識(shí)圖譜注入到大模型中,以增強(qiáng)其事實(shí)推理能力。

1 方法

1.1 問(wèn)題導(dǎo)向的知識(shí)圖譜重評(píng)分

評(píng)分主要計(jì)算子圖每條邊和問(wèn)題的相似度,方法如下:

  • 對(duì)子圖的節(jié)點(diǎn)對(duì)通過(guò)PLM(pre-trained language model)編碼成e_se_o。
  • 將問(wèn)題通過(guò)同一個(gè)PLM進(jìn)行編碼,然后通過(guò)一個(gè)平均池化層聚合成e_q。
  • 最后通過(guò)一個(gè)雙線性層輸出每條邊的相關(guān)性分?jǐn)?shù)
    \eta=Normalize(Bilinear([e_x,e_o],e_q))
1.2 重評(píng)分知識(shí)圖譜建模

增加了一個(gè)連接所有主題節(jié)點(diǎn)的全局節(jié)點(diǎn)(global node),以此拓展了圖注意力網(wǎng)絡(luò)(GAT)。節(jié)點(diǎn)\mathcal{V}_t層間更新公式如下:
h^{(l+1)}_t=f_m(\sum_{s\in N_t \cup\{t\}}\alpha_{st}m_{st})+h^{l}_t

  • f_m:兩層MLP。
  • N_t\mathcal{V}_t的鄰居節(jié)點(diǎn)。
  • \alpha_{st}:權(quán)重,實(shí)際計(jì)算方式為注意力分?jǐn)?shù)和前面計(jì)算的邊相關(guān)性分?jǐn)?shù)的乘積。
  • m_{st}:節(jié)點(diǎn)信息。

其中,全局節(jié)點(diǎn)的初始化表示為0。

1.3 Knowformer

用Knowformer層替換最接近LLM輸出的M個(gè)Transformer層,將結(jié)構(gòu)化知識(shí)信息注入到LLM中。注入方法是將圖結(jié)構(gòu)信息拼接到key-value矩陣上:
FFN(x)=(x ·[key:\phi_k])·[value:\phi_v]
其中\phi_k=Pr_kh,\phi_v=Pr_vh,Pr_k,Pr_v是兩個(gè)參數(shù)矩陣。

2 實(shí)驗(yàn)

實(shí)驗(yàn)采用編碼器-解碼器架構(gòu)的大模型,F(xiàn)LAN-T5-XL(3B)和FLAN-T5-XXL(11B)。采用數(shù)據(jù)集為:

  • OpenBookQA(OBQA):選擇題,新型問(wèn)答數(shù)據(jù)集,回答問(wèn)題需要額外的常識(shí)和豐富的文本理解。
  • AI2 Reasoning Challenge(ARC):用于評(píng)估常識(shí)推理。包含7,787個(gè)科學(xué)考試選擇題。
  • PIQA:物理類選擇題。
  • RiddleSense (Riddle):謎語(yǔ)風(fēng)格的選擇題。

采用的知識(shí)圖譜為結(jié)構(gòu)化存儲(chǔ)形式的ConceptNet。



為了探索模型泛化能力,分別在僅編碼器型和僅解碼器型架構(gòu)的大模型上做了實(shí)驗(yàn)。


本文還針對(duì)三個(gè)主要步驟做了一些消融實(shí)驗(yàn)

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容