HyperAIHyperAI

Command Palette

Search for a command to run...

記憶による推論:最近傍ノードを用いた知識グラフ埋め込み

Peng Wang Xin Xie Xiaohan Wang Ningyu Zhang

概要

従来の知識グラフ埋め込み手法は、エンティティを表現にマッピングし、スコア関数を用いてターゲットエンティティを予測するが、稀なまたは新規の未観測エンティティの推論においては通常、困難を抱えている。本論文では、事前学習済み言語モデルを活用した新たな知識グラフ埋め込み手法であるkNN-KGEを提案する。本手法は、エンティティの分布をk近傍(k-nearest neighbors)と線形補間することで、エンティティ表現の推論を改善する。近傍エンティティは、知識ストア内のエンティティ埋め込み空間における距離に基づいて計算される。このアプローチにより、稀なまたは新規のエンティティをモデルパラメータ内に暗黙的に記憶するのではなく、明示的に記憶することが可能となる。実験結果から、本手法が誘導的(inductive)および従属的(transductive)なリンク予測の性能を向上させ、わずか数個のトリプルしか存在しない低リソース環境においても優れた性能を発揮することが示された。これは、明示的な記憶によりより容易に推論が可能であることを示唆している。コードは、https://github.com/zjunlp/KNN-KG にて公開されている。


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています