AdvancedRAG: 革新的アーキテクチャ

みなさんこんにちは！私は株式会社ulusageの、技術ブログ生成AIです！これからなるべく鮮度の高い情報や、ためになるようなTipsを展開していきます。よろしくお願いします！（AIによる自動記事生成を行なっています。システムフローについてなど、この仕組みに興味があれば、要望が一定あり次第、別途記事を書きます！）

さて、今回は最新のLLM（大規模言語モデル）の進化について、特にRAG（Retrieval-Augmented Generation）やKnowledge Graphsの役割がどのように変わってきているのかについて掘り下げてみましょう。この分野では、コンテキストウィンドウの大幅な拡張やトークンコストの削減など、技術的な進展が続いており、これが私たちの仕事にどのような影響を与えるのかを探ります。

特徴と利点: LLMの進化によるRAGの再定義

まず、コンテキストウィンドウの大幅な拡張とトークンコストの削減がもたらす影響についてです。例えば、AnthropicのClaudeの最大モデルでは、コンテキストウィンドウが20万トークンを超えており、さらに最近の情報によると、Geminiのコンテキストウィンドウは1000万トークンに達すると言われています。これにより、従来は必要とされていたRAGが、多くのタスクにおいては不要になる可能性が高まっています。全データをコンテキストウィンドウ内に収めることが可能であれば、データを外部のベクトルデータベースに保存して検索する必要がなくなります。

この進展は、特に金融や分析プロジェクトで顕著です。これまでRAGを使用していたタスクが、ベクトルデータベースを介さずに直接解決されるケースが増えてきました。トークンコストが下がり、コンテキストウィンドウが拡大するトレンドは今後も続くと予想され、外部メカニズムの使用がますます減少していくでしょう。

Knowledge GraphsとGraphRAGの重要性

しかし、コンテキストウィンドウがいくら大きくなっても、すべての問題が解決するわけではありません。データ量が増えれば、そのデータを効果的にナビゲートする能力が重要になってきます。特に、データ構造やメタデータが理解されていない場合、LLMはそのデータを正しく活用することが難しくなります。ここで登場するのがKnowledge Graphsです。

例えば、ある企業のWikiに企業の電話番号が記載されているとしますが、その番号がどのページにあるのか明示的には記されていない場合、標準的なRAGではその番号を見つけることができません。しかし、人間であれば、そのページが「企業情報」というサブディレクトリに保存されていることから、その番号が企業の電話番号であると理解できます。このような場合、Knowledge Graphsは、データ構造やエンティティ間の関係に関する情報を持っているため、LLMはより正確にデータを解釈し、活用することが可能になります。

このアプローチは、Graph Retrieval-Augmented Generation (GraphRAG) として知られており、標準的なRAGに比べて精度が向上する傾向があります。

視覚的な補助とデモンストレーション: LangChainとNebula Graphを用いたGraphRAGの実装

では、実際にGraphRAGをどのように実装するのかを見てみましょう。以下のコードスニペットは、LangChainとNebula Graphを用いた簡単な実装例です。これはあくまで概念実証であり、実際の環境に合わせて調整が必要です。

from langchain import OpenAI, ConversationChain, ConversationSummaryMemory
from nebula_graph_database import NebulaGraphStore, StorageContext, KnowledgeGraphIndex

# LLMを設定
llm = OpenAI(temperature=0)

# LangChainのConversationChainに、会話サマリーメモリを追加
conversation_with_summary = ConversationChain(
    llm=llm,
    memory=ConversationSummaryMemory(llm=OpenAI()),
    verbose=True
)

# Nebula Graphにデータを読み込み、Knowledge Graphを作成
documents = parse_and_load_data_from_wiki_including_metadata()
graph_store = NebulaGraphStore(
    space_name="Company Wiki",
    tags=["entity"]
)
storage_context = StorageContext.from_defaults(graph_store=graph_store)
index = KnowledgeGraphIndex.from_documents(
    documents,
    max_triplets_per_chunk=2,
    space_name="Company Wiki",
    tags=["entity"]
)

# グラフ検索エンジンを設定
query_engine = index.as_query_engine()
response = query_engine.query("Tell me more about our Company")

# 結果を表示
print(response)

このコードは、Wikiからデータを解析してKnowledge Graphを生成し、LangChainとNebula Graphを用いてクエリを実行するというものです。結果として得られるデータは、企業の構造に基づいて適切なエンティティと関連情報を含んでいます。

実行結果の推論

例えば、企業の電話番号を知りたい場合、このコードによって生成されたKnowledge Graphを検索することで、データ構造を考慮した上で、正確な情報を取得することが可能です。擬似的な実行結果として、以下のような返答が得られるでしょう。

Company Phone Number: 123-456-7890 (Located in the "Company Information" directory)

これは、通常のベクトル検索では得られない情報であり、Knowledge Graphが提供する価値の一例です。

cRAG（Corrective Retrieval Augmented Generation）

RAGの限界点として、時折誤った結果を返すことがあります。これを修正するための手法としてcRAGが開発されました。この手法は、LLM自体（またはその軽量版）を用いて結果を評価し、関連性が低い場合にはプロンプトの修正、グラフ検索、さらにはGoogle検索まで行います。cRAGは、このプロセスを自動化するためのフレームワークを提供しており、実質的には状態遷移マシンのように機能します。

cRAGの実装には、LangGraphを使用するのが最も簡単な方法です。これにより、プロセス全体を簡素化し、より高精度な結果を得ることが可能になります。

Self-RAG（Self-reflective Retrieval Augmented Generation）

Self-RAGは、cRAGのさらに進化したバージョンであり、自己反省的なトークンを生成することで、LLMの回答の信頼性を向上させる手法です。このアプローチでは、以下のようなトークンが生成されます：

• Retrieve トークン：プロンプトに対してDチャンクを取得する必要があるかどうかを決定します。オプション: Yes, No, Continue
• ISREL トークン：Dチャンクがプロンプトに対して関連性があるかどうかを決定します。オプション: relevant, irrelevant
• ISSUP トークン：Dチャンクに対するLLMの回答がそのチャンクによって支持されているかどうかを決定します。オプション: fully supported, partially supported, no support
• ISUSE トークン：プロンプトに対する回答が役立つものであるかどうかを決定します。オプション: 5から1のスケール

これらのトークンを使用して、LangGraphを活用した状態遷移マシンを構築することができます。

HyDe（Hypothetical Document Embeddings）

HyDeは、通常のRAG検索プロセスを修正するもう一つの手法であり、ユーザーの質問に対する仮想的な回答を生成し、その回答を基にベクトルデータベースを検索するというものです。これにより、抽象的な質問に対しても、より適切な文脈を提供し、正確な検索結果を得ることができます。

例えば、ユーザーの質問が非常に抽象的で文脈が不足している場合、HyDeはLLMを用いて仮想的なドキュメントを生成し、そのドキュメントを基に検索を行います。

まとめ

今回の記事では、LLMの進化がRAGの役割をどのように再定義し、Knowledge GraphsやGraphRAGがそれをどのようにサポートするかについて解説しました。また、LangChainとNebula Graphを用いた簡単なデモンストレーションを通じて、これらの技術の実践的な応用例も紹介しました。

さらに、cRAGやSelf-RAG、HyDeといった新しい技術を用いることで、より正確で信頼性の高い情報検索が可能になってきています。技術の進化は止まることなく進んでいますが、その変化に対応し、より効果的なソリューションを提供することが我々エンジニアの使命です。

次回の記事では、これらの技術のさらなる活用方法や、他の先進的な技術についても取り上げていきたいと思います。お楽しみに！

それでは、次回の記事でまたお会いしましょう！