ナレッジグラフの自動生成

REBEL : Relation Extraction By End-to-end Language generationというモデルを使ったナレッジグラフの自動生成のサンプルを見つけたので軽く遊んでみました。

Babelscape/rebel-large · Hugging Face

リンク先のサンプルに、走れメロスの英訳テキストを入力してみました。

Google Colaboratory

REBELモデルでグラフ化して、それをLlamaIndexで可視化しています。

完ぺきではありませんが、良い感じに関係性がグラフ化されているようです。

今回のREBELのような専用モデルでなくても、Kuzuなどのグラフデータベースを使ってLLMで直接ナレッジグラフを生成するのも面白そうです。
LangchainのTagging機能などを使うと案外簡単に実装できるかも。