アーカイブ / RAG / LLM
文書由来の知識グラフで質問の文脈を補い、関連文書を探すKG-CQR
KG-CQR
KG-CQRの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1: Overview of query expansion approaches for RAG systems: a) query decomposition; b) document generation; c) ours: KG-enhanced contextual generation
概要
- 文書から作った知識グラフの関係を選び、質問の検索用文脈を生成する。
- 著者らのRAGBench評価では、BGEのmAPが0.521から0.542になった。
- 社内文書への適用には、抽出した関係の正確さと更新作業の検証が必要である。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、KG-CQRの仕組み
- 01質問と文書群
- 02関係の三つ組を選択
- 03グラフ内の経路で補完
- 04文脈生成と文書検索
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
KG-CQRは、質問に関係する文書を検索する前に、その質問の文脈を補う方法である。検索拡張生成（RAG）は、外部の文書を探し、その内容を回答生成に利用する仕組みを指す。
短い質問と詳しい文書では、同じ話題でも使う言葉や説明の範囲が異なり、必要な文書を見つけにくい。著者らは、文書中の対象と関係を結んだ知識グラフを使い、特に複数の関係をたどる質問で、このずれを小さくすることを狙った。
手法
まず文書群から、対象・関係・対象を表す「三つ組」を抽出して知識グラフを作る。各三つ組には、抽出元の文書と関係を踏まえ、言語モデルが説明文を付ける。質問と説明文の意味の近さで候補を選び、言語モデルで無関係な候補を除く。「交換部品の確認先」を探すなら、部品と手順の関係などが候補になり得る。
次に、候補の対象同士を結ぶ経路をグラフ内で探し、質問との近さで既存の三つ組を追加する。追加後の関係から検索用の文章を生成し、それと元の質問をそれぞれ数値表現に変えて重み付きで合わせ、文書を検索する。この「補完」は既存グラフから関係を足す処理を指す。
新規性
KG-CQRの主張は、文書群から作った知識グラフを、質問側の文脈を補うために使う点にある。関係の三つ組に、抽出元の文書を踏まえた説明文を付け、質問との意味の近さで選ぶ。
選んだ三つ組をグラフ内の経路で補い、検索用の文章を生成する。著者らは、手法のための追加学習なしに、異なる規模の言語モデルと組み合わせられる構成としている。
従来手法との違い
著者らは、語の一致を使うBM25と、質問と文書を数値表現で照合するDPR、BGEを比較対象とした。質問拡張（QE）と、質問から仮の文書を生成するHyDEも評価したが、表にHyDEとBM25の組み合わせはない。
KG-CQRは文書由来の関係を選んで質問の文脈を生成し、元の質問と生成文の数値表現を重み付きで合わせる。単に質問語を増やすQEや、仮の文書を生成するHyDEとは、文脈を作る根拠が異なる。
実験結果
著者らはRAGBenchのテスト約11,000件とMultiHop-RAGの2,556問で文書検索を評価した。各データ集合の知識グラフ作成にはLLaMA-3.3-70Bを使い、報告値では元の質問の重みを0.7とした。mAPは関連文書の順位も考慮する指標、Recall@25は上位25件で関連文書を拾う度合いを示す。
RAGBenchでは、BGEのmAPが0.521、KG-CQR＋BGEが0.542で、Recall@25は0.657から0.675になった。MultiHop-RAGでは、BM25のRecall@25が0.486、KG-CQR＋BM25が0.532だった。いずれも著者らが示したデータ集合と設定での結果である。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：社内の保守手順書から「部品交換後の確認項目と異常時の連絡先」を探す。入力は質問と、部品、作業、担当部署の記述がある文書群。出力は、担当者が原文を確認できる関連文書の候補一覧とする。
元データは検索対象の文書で、手法のための追加学習用データやセンサーは想定しない。文書から関係を抽出し、原文との対応を点検する適用作業が必要になる。正解文書のラベルは評価用に人手で用意する。社内文書での効果は未検証である。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：対象を一部の社内保守文書に絞り、文書から対象と関係を抽出してグラフを作る。各関係の説明文と原文箇所の対応を人が点検する。追加センサーと手法のための学習用ラベルは不要と想定するが、文書に合わせた抽出結果の確認は必要である。
実際の質問を集め、正解文書のラベルを人手で付ける。同じ文書群でBM25単独とKG-CQR＋BM25を走らせ、Recall@25を比較する。誤った関係による検索の取り違え、文書更新後の修正作業、質問ごとの処理時間も記録する。
限界と課題
著者らは、対象分野の知識グラフを構築すること自体を課題と位置づける。KG-CQRの補完は、抽出済みの関係をグラフ内の経路で結び、既存グラフから三つ組を追加する処理であり、原文にない事実の正しさを保証するものではない。
編集上の検証事項は、関係抽出や説明文の誤りが検索に及ぼす影響、文書更新時のグラフ管理、処理時間と費用である。確認できた本文の範囲では、これらの運用値や社内保守文書への適用結果は示されていない。
出典
元のタイトル・要旨を確認する（英語）
KG-CQR: Leveraging Structured Relation Representations in Knowledge Graphs for Contextual Query Retrieval
The integration of knowledge graphs (KGs) with large language models (LLMs) offers significant potential to improve the retrieval phase of retrieval-augmented generation (RAG) systems. In this study, we propose KG-CQR, a novel framework for Contextual Query Retrieval (CQR) that enhances the retrieval phase by enriching the contextual representation of complex input queries using a corpus-centric KG. Unlike existing methods that primarily address corpus-level context loss, KG-CQR focuses on query enrichment through structured relation representations, extracting and completing relevant KG subgraphs to generate semantically rich query contexts. Comprising subgraph extraction, completion, and contextual generation modules, KG-CQR operates as a model-agnostic pipeline, ensuring scalability across LLMs of varying sizes without additional training. Experimental results on RAGBench and MultiHop-RAG datasets demonstrate KG-CQR's superior performance, achieving a 4-6% improvement in mAP and a 2-3% improvement in Recall@25 over strong baseline models. Furthermore, evaluations on challenging RAG tasks such as multi-hop question answering show that, by incorporating KG-CQR, the performance consistently outperforms the existing baseline in terms of retrieval effectiveness
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-10-01