知識圖譜表示學習(一) TransE,TransH,TransR,CTransR,TransD

【自然語言處理】【知識圖譜】知識圖譜表示學習(一)：TransE、TransH、TransR、CTransR、TransD
【自然語言處理】【知識圖譜】知識圖譜表示學習(二)：TranSparse、PTransE、TransA、KG2E、TransG
【自然語言處理】【知識圖譜】知識圖譜表示學習(三)：SE、SME、LFM、RESCAL、HOLE
【自然語言處理】【知識圖譜】知識圖譜表示學習(四)：【RotatE】基于復數空間關系旋轉的知識圖譜嵌入
【自然語言處理】【知識圖譜】知識圖譜表示學習(五)：【PairRE】基于成對關系向量的知識圖譜嵌入

一、簡介

• 知識圖譜及挑戰

  知識圖譜以三元組的方式來描繪整個世界，大型知識圖譜Freebase、DBpedia和YAGO等已經在各種應用程序中被廣泛使用。但是，隨著知識圖譜規模的增加，面臨著兩個挑戰：

  • 數據稀疏

    大型網絡中節點太多而邊太少。

  • 計算低效

    圖譜尺寸增加，計算效率必然降低。

• 解決方案

  為了解決上面的挑戰，表示學習被引入至知識表示中。具體來說，就是將知識圖譜中的實體和關系投影至低維連續空間中，從而獲得一個分布式表示。相比于傳統的one-hot表示，分布式表示的維度更低且計算復雜度也低。此外，通過低維向量間的距離計算還可以精準的展示實體間的相似性。

二、符號

• 使用$G=(E,R,T)$來表示完整的知識圖譜，其中$E=\{e_1,e_2,\dots ,e_{|E|}\}$表示實體集合，$R=\{r_1,r_2,\dots ,r_{|R|}\}$表示關系集合，$T$表示三元組集合，$|E|$和$|R|$表示實體和關系的數量。
• 知識圖譜以三元組$?h,r,t?$的形式表示，其中$h\in E$表示頭實體，$t\in E$表示尾實體，$r\in R$表示$h$和$t$間的關系。

三、TransE

1. 動機

• 將關系看作實體間的翻譯

  直覺上，將實體投影至低維向量空間時，具有相似語義的實體應該被投影至相同的簇，而含義不同的實體則屬于不同的簇。例如$\text{William?Shakespeare}$和$\text{Jane?Austen}$應該投影至作者簇，而$\text{Romeo?and?Juliet}$和$\text{Pride?and?Prejudice}$則被投影至書籍簇中。它們之間的共同點就是共享相同的關系$\text{works\_written}$，通過關系的翻譯可以將作者和書籍投影至不同的簇。

• $\text{word2vec}$的突破

  $\text{word2vec}$通過上下文來學習詞表示，得到的詞向量存在一個有趣的現象：若兩個詞共現相似的語義或者句法關系，那么對應的詞向量也相似。舉例來說，
  $$\text{w}(king)?\text{w}(man)\approx \text{w}(queen)?\text{w}(woman)$$
  上式表示在隱空間中king和man關系類似于queen和woman的關系。此外，除了在語義中發現這種近似關系，句法中也有類似關系，即
  $$\text{w}(bigger)?\text{w}(big)\approx \text{w}(smaller)?\text{w}(small)$$

• 計算復雜度

  首先，模型復雜度增加會導致高的計算代價和模型解釋性差；

  其次，復雜模型會過擬合；

  最后，實驗表明，具有相對較大關系量的知識圖譜應用，簡單模型和復雜模型表現相當；

  因此，基于翻譯的這種假設能夠很好的平衡準確率和效率；

2. 方法

? $\text{TransE}$會將實體和關系投影至低維向量空間${\mathbb{R}}^d$，其中$d$是嵌入向量維度的超參數。基于翻譯的假設，對于每個三元組$?h,r,t?\in T$，均期望嵌入向量和$\text{h+r}$與尾實體嵌入向量$\text{t}$接近。

? 具體來說，$\text{TransE}$會先定義一個評分函數
$$\mathcal{E}(h,r,t)=\parallel \text{h+r-t}\parallel$$
然后，基于該評分函數構建了一個基于邊界的損失函數
$$\mathcal{L}=\sum _{?h,r,t?\in T}\sum _{?h^{\prime },r^{\prime },t^{\prime }?\in T^{?}}max([\gamma +\mathcal{E}(h,r,t)?\mathcal{E}(h^{\prime },r^{\prime },t^{\prime })],0)$$
其中，$\mathcal{E}(h,r,t)$是正樣本的評分函數，$\mathcal{E}(h^{\prime },r^{\prime },t^{\prime })$是負樣本的評分函數，$\gamma >0$是邊界超參數，$T^{?}$是與$T$對應的負樣本三元組。簡單分析該損失函數，其會最小化$\mathcal{E}(h,r,t)$并最大化$\mathcal{E}(h^{\prime },r^{\prime },t^{\prime })$，但是兩者的差距不會大于$\gamma$。

? 知識圖譜中并沒有顯式的負樣本三元組，因此按如下定義構造負樣本三元組$T^{?}$
$$T^{?}=\{?h^{\prime },r,t?|h^{\prime }\in E\}\cup \{?h,r^{\prime },t?|r^{\prime }\in R\}\cup \{?h,r,t^{\prime }?|t^{\prime }\in E\},?h,r,t?\in T$$
上式的一個直觀解釋，隨機使用其他三元組替換原始三元組$?h,r,t?$的頭實體、尾實體或者關系。此外，若生成的三元組已經在$T$，那么就不會被加入$T^{?}$。

• 知識補全

  給定三元組中的任意兩個元素，預測第3個元素被稱為知識補全任務。該任務用于評估學習到的知識表示。

3. 缺點與挑戰

? $\text{TransE}$雖然簡單有效，但是仍然存在一些缺點和挑戰。

3.1 復雜關系

? 在知識補全任務中，給定三元組中的兩個元素，可能會存在多個答案。例如，給定頭實體$\text{William?Shakespeare}$和關系$\text{works\_written}$，將會得到一個代表作的列表$\text{Romeo?and?Juliet}$、$\text{Hamlet}$和$\text{A?Midsummer?Night’s?Dream}$。這些代表中共現相同的作者信息，但是在主題、背景、角色上不同。但是，$\text{TransE}$僅能為一個實體參數一個嵌入向量，這極大的限制了$\text{TransE}$的能力。在許多文獻中，將關系分為四類：1-to-1、1-to-Many、Many-to-1和Many-to-Many，而更加統計表明1-to-Many、Many-to-1和Many-to-Many類型的關系最多，但是$\text{TransE}$僅能處理1-to-1的關系。

? 此外，$\text{TransE}$在處理自反關系時也存在困難。

3.2 一跳關系

? $\text{TransE}$僅考慮了一條關系，忽略了長距離的隱關系。

3.3 效果和效率

? $\text{TransE}$出于效率的原因，其函數和表示都被過度簡化，導致其不足以建模知識圖譜中的復雜實體和關系。如果評估效果和效率，仍然是非常大的挑戰。

四、TransH

? $\text{TransH}$主要是解決1-to-Many、Many-to-1和Many-to-Many問題的。

• 一個例子

  在談及國籍時，$\text{William?Shakespeare}$應該和$\text{Isaac?Newton}$接近；而談及職業時，則應該和$\text{Mark?Twain}$接近。

• 方法

$\text{TransH}$通過為具有不同關系的相同實體賦予不同的向量表示來解決這個問題的。如上圖所示，$\text{TransH}$為每個關系賦予一個超平面${\text{w}}_r$，然后判斷超平面上的相似性，而不是原始的實體向量。給定三元組$?h,r,t?$，$\text{TransH}$將會原始向量表示$\text{h}$和$\text{t}$投影至超平面${\text{w}}_r$上來獲得投影向量${\text{h}}_{\perp }$和${\text{t}}_{\perp }$。翻譯向量$\text{r}$用于連接超平面${\text{h}}_{\perp }$和${\text{t}}_{\perp }$。評分函數定義為
$$\mathcal{E}(h,r,t)=\parallel {\text{h}}_{\perp }+\text{r}?{\text{t}}_{\perp }\parallel$$
其中，
$${\text{h}}_{\perp }=\text{h}?{\text{w}}_r^T\text{h}{\text{w}}_r,{\text{t}}_{\perp }=\text{t}?{\text{w}}_r^T\text{t}{\text{w}}_r$$

其中，${\text{w}}_r$是向量且約束$\parallel {\text{w}}_r{\parallel }_2$為1。

• 訓練

  損失函數和訓練方式同$\text{TransE}$。

五、TransR

• 動機

  $\text{TransH}$通過使用超平面的方式為具有多個關系的實體賦予不同的向量表示，但是實體和關系仍然在相同的語義空間中，這限制了建模實體和關系的能力。$\text{TransR}$假設實體和關系處于不同的語義空間中。

• 方法

如上圖所示，三元組$?h,r,t?$中$\text{h,t}\in {\mathbb{R}}^k$且$\text{t}\in {\mathbb{R}}^d$。$\text{TransR}$首先會將$\text{h}$和$\text{t}$從實體空間投影至關系空間。也就是說，每個實體在每個關系上都有一個表示。$\text{TransR}$的評分函數為
$$\mathcal{E}(h,r,t)=\parallel {\text{h}}_r+\text{r}?{\text{t}}_r\parallel$$
其中，${\text{h}}_r$和${\text{t}}_r$是實體向量$\text{h}$和$\text{t}$在$r$對應向量空間的表示，具體的投影過程為
$${\text{h}}_r={\text{hM}}_r,{\text{t}}_r={\text{tM}}_r$$
其中，${\text{M}}_r\in {\mathbb{R}}^{k\times d}$是將實體映射至$r$關系空間的投影矩陣。$\text{TransR}$會約束嵌入向量的范數，具有$\parallel \text{h}{\parallel }_2\le 1$、$\parallel \text{t}{\parallel }_2\le 1$、$\parallel \text{r}{\parallel }_2\le 1$、$\parallel {\text{h}}_r{\parallel }_2\le 1$、$\parallel {\text{t}}_r{\parallel }_2\le 1$。

• 訓練

  $\text{TransR}$的訓練方式同$\text{TransE}$

六、CTransR

• 動機

  知識圖譜中的一些關系可以被劃分為子關系，從而給出更加準確的信息。這些子關系間的不同，可以通過實體對來學習。舉例來說，關系$\text{location\_contains}$可以被認為包含子關系$\text{city-street}$、$\text{country-city}$、$\text{country-university}$。隨著考慮子關系，實體會被投影至語義向量空間中更加準確的位置。

• 方法

  $\text{CTransR(Cluster-based?TransR)}$可以看作是考慮子關系情況下的$\text{TransR}$增強版本。具體來說，對于每個關系$\text{r}$，所有的實體對$\text{(h,t)}$會被聚類至若干個組。聚類主要是依據$\text{t-h}$，其中$\text{h}$和$\text{t}$是通過$\text{TransE}$預訓練得到的。隨后，通過每個聚類簇中的實體對可以學習到子關系向量${\text{r}}_c$，那么最終的評分函數
  $$\mathcal{E}(h,r,t)=\parallel {\text{h}}_r+{\text{r}}_c?{\text{t}}_r\parallel +\alpha \parallel {\text{r}}_c?\text{r}\parallel$$
  其中，$\parallel {\text{r}}_c?\text{r}\parallel$是希望向量${\text{r}}_c$與$\text{r}$的差距不要太大

七、TransD

• 動機

  $\text{TransH}$和$\text{TransR}$主要是解決不同關系的多實體表示問題，從而改善知識補全和三元組分類的效果。然而，這兩個方法僅是根據關系來投影實體，忽略了實體的多樣性。此外，使用矩陣-向量乘法實現投影操作將會導致高的時間復雜度，這導致在大規模圖中的時間消耗太多。因此，$\text{TransD}$提出了一種基于實體和關系的動態映射矩陣，其能同時考慮實體和關系的多樣性。

• 方法

$\text{TransD}$會為每個實體或者關系定義兩個向量，一個是使用$\text{TransE}$、$\text{TransH}$和$\text{TransR}$獲得的原始向量，另一個是被用于構造投影句子的投影向量。

$\text{TransD}$使用$\text{h,r,t}$表示原始向量，${\text{h}}_p,{\text{t}}_p,{\text{r}}_p$來表示投影向量。使用兩個矩陣${\text{M}}_{rh},{\text{M}}_{rt}\in {\mathbb{R}}^{m\times n}$完成實體空間至關系空間的映射，這兩個投影句子的動態計算為
$${\text{M}}_{rh}={\text{r}}_p{\text{h}}_p^{?}+{\text{I}}_{m\times n}{\text{M}}_{rt}={\text{r}}_p{\text{t}}_p^{?}+{\text{I}}_{m\times n}$$
上式意味著實體和關系的投影向量被合并來決定最終的投影矩陣。評分函數定義為
$$\mathcal{E}(h,r,t)=\parallel {\text{M}}_{rh}\text{h}+\text{r}?{\text{M}}_{rt}\text{t}\parallel$$
投影矩陣被初始化為單位矩陣。

$\text{TransD}$提出了一種動態的方法來構造投影矩陣，從而能夠同時考慮實體和關系的多樣性，從而實現了更優的表述。此外，相較于$\text{TransR}$，其具有更低的時間和空間復雜度。

引用文獻

[1]. Zhiyuan Liu, Yankai Lin and Maosong SUn. Representation Learning for Natural Language Processing.

總結

以上是生活随笔為你收集整理的【自然语言处理】【知识图谱】知识图谱表示学习(一)：TransE、TransH、TransR、CTransR、TransD的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。