アーカイブ / LLM / RAG
誤情報が混じる検索結果に備えるRAGの微調整手法「RbFT」
RbFT
RbFTの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1 . Overview of our RbFT. Specifically, RbFT consists of two sub-tasks: Defects Detection and Utility Extraction, which aim to identify the types of retrieval defects and generate the final answer with limited useful information, respectively. In the figure, green text indicates relevant information, while red text represents incorrect counterfactual information.
概要
- RbFTは、検索文書の欠陥判定と、欠陥を含む文書からの回答生成を学習させる。
- 著者らは3種類の質問応答データで評価し、全検索文書を置換した混合条件で通常のRAGを上回る平均EMを報告した。
- 社内文書への適用には、正解付き質問と文書ごとのラベルを用意し、旧版や誤情報への反応を検証する必要がある。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、RbFTの仕組み
- 01質問・正解・検索文書
- 02欠陥文書へ置換
- 03判定と回答を学習
- 04微調整済みモデル
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
RbFTは、検索結果に欠陥があっても質問に答えられるよう、検索拡張生成の生成モデルを微調整する手法である。検索拡張生成(RAG)は外部文書を探して回答に使うため、検索器や文書集合の質が回答に影響する。
著者らが扱う欠陥は、話題は近いが答えがない文書、質問と無関係な文書、誤情報を含む文書の三種類である。実務でも古い版や別件の文書が検索結果に混じり得るため、取得した文書を生成モデルがどう扱うかが問題になる。
手法
RbFTは、学習用の検索文書を一定の確率で欠陥文書に置き換え、二つの課題で生成モデルを微調整する。話題だけ近い文書は検索順位の低い結果から、無関係な文書は文書集合から採る。誤情報を含む文書は、正解と異なる答えを生成し、それに合わせて元の文書を書き換える。
一つ目の課題では、質問と文書一覧から各文書の有用性と欠陥種別を判定する。二つ目では、欠陥を含む一覧から正解を出すよう学習する。例えば曲の収録アルバムを尋ねる質問では、バンドの一般説明や誤ったアルバム名が混じる入力を使う。微調整には、モデルの重みの変更を小さな追加分として学習するLoRAを用いる。
新規性
RbFTの提案は、検索結果を受け取る生成モデルそのものに、文書の欠陥判定と、残る有用な情報からの回答生成を学習させる点にある。判定では文書一覧を一度に入力し、各文書が回答に役立つか、役立たない場合はどの欠陥に当たるかを出力させる。
回答の学習では、欠陥文書を含む一覧から正解を出力させる。著者らはこの二つの課題を組み合わせ、検索結果への過度な依存を減らすことを狙う。
従来手法との違い
通常の検索拡張生成(RAG)は、質問に関連する文書を検索し、生成モデルへ渡して回答を作る。RbFTは検索後の生成モデルを微調整し、話題だけ近い文書、無関係な文書、誤情報を含む文書が混じる条件を学習させる。
著者らは検索なし、通常のRAG、RobustRAG、CRAG、InstructRAG、AstuteRAGと比較した。RobustRAGは文書ごとに回答してから集約する方式である。比較時のCRAGでは追加知識を得る大規模ウェブ検索機能を無効にしているため、その条件での結果として読む必要がある。
実験結果
著者らはNatural Questions、HotpotQA、TriviaQAの質問応答で評価した。三つの訓練用集合から計2万件を抽出し、その10%を検証用とした。評価には前二者のうちNatural QuestionsとTriviaQAのテスト集合、HotpotQAの検証集合から各1000問を使い、回答の完全一致率(EM)と単語単位のF1を測った。
表の3集合平均では、全検索文書を欠陥文書に置換する混合条件で、LlamaのRbFT、通常のRAG、AstuteRAGのEMは順に31.9、11.4、21.1だった。Qwenでは順に24.0、11.0、16.4だった。これは著者らが示したモデルと置換条件での結果であり、社内文書での性能を示す値ではない。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:社内規程の質問応答で、社員の質問と社内検索で見つかった規程の抜粋を入力し、短い回答を出す。担当者が回答と参照文書を確認してから利用する運用を想定する。旧版の規程や別部署の手順が混じる場合に、回答がどう変わるかを調べる。
適応には、利用できる現行の規程本文、正解付きの質問、検索文書ごとの有用性と欠陥種別のラベルが要る。正解は担当者が現行規程に照らして確定する。入力は文書と質問なので追加センサーは想定せず、対象規程に合わせた学習例と評価用の質問を作る。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:対象を一種類の社内規程に絞り、現行の規程本文を検索用データにする。担当者が正解を確認した質問を集め、検索文書ごとに「有用」「話題のみ近い」「無関係」「誤り」のラベルを付ける。旧版や別件を含む学習例を用意し、対象領域に合わせてモデルを適応させる。追加センサーは想定しない。
同じ検索器と文書集合を使う通常のRAGを比較基準にし、欠陥のない入力と欠陥を混ぜた入力を分けて測る。学習に使わない質問で完全一致率を比較し、担当者が根拠文書の適切さ、誤情報を採用した件数、回答時間も記録する。
限界と課題
確認できた本文の範囲では、著者らによる本手法固有の限界の明示は確認できない。報告された評価は質問応答データと、検索文書を置換して作った欠陥に基づく。社内規程の版違いや、実際の文書間の矛盾でも同じ結果になるとは示されていない。
編集上の検証事項は、有用な文書が一件もない場合、誤情報が多数を占める場合、誤った文書を有用と判定した場合の回答である。学習時には元の検索文書を正例として扱うため、その文書自体が不十分な場合のラベルの妥当性も確認したい。
出典
元のタイトル・要旨を確認する(英語)
RbFT: Robust Fine-tuning for Retrieval-Augmented Generation against Retrieval Defects
Retrieval-augmented generation (RAG) enhances large language models (LLMs) by integrating external knowledge retrieved from a knowledge base. However, its effectiveness is fundamentally constrained by the reliability of both the retriever and the knowledge base. In real-world scenarios, imperfections in these components often lead to the retrieval of noisy, irrelevant, or misleading counterfactual information, ultimately undermining the trustworthiness of RAG systems. To address this challenge, we propose Robust Fine-Tuning (RbFT), a method designed to enhance the resilience of LLMs against retrieval defects through two targeted fine-tuning tasks. Experimental results demonstrate that RbFT significantly improves the robustness of RAG systems across diverse retrieval conditions, surpassing existing methods while maintaining high inference efficiency and compatibility with other robustness techniques.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01




