本文提出了一種新的知識圖譜嵌入模型OrthogonalE,主要包括以下三個方面:
使用矩陣表示實體,可以通過調整矩陣的列數m來靈活控制實體維度,避免不必要地擴大關係矩陣的大小。
使用分塊對角正交矩陣表示關係,可以調整分塊矩陣的維度d來適應不同複雜度的數據集,提高模型的一般性。同時,這種關係表示方式也能捕捉對稱、反對稱、反轉和非交換組合等多種關係模式。
對關係矩陣使用黎曼優化,而對實體矩陣使用隨機梯度下降,可以更有效地優化模型參數。
實驗結果表明,OrthogonalE模型在WN18RR和FB15K-237數據集上的鏈接預測性能顯著優於現有的最先進模型,同時大幅減少了關係參數的數量。此外,通過分析關係嵌入的直方圖,驗證了OrthogonalE模型能夠同時捕捉多種關係模式。
Egy másik nyelvre
a forrásanyagból
arxiv.org
Mélyebb kérdések