表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

LLMの正答に役立つ文書を選べるか――ICLERBによる検索評価とRLRAIFによる調整

ICLERB / RLRAIF

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

ICLERB / RLRAIFの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1: Retrieval performance for the task of ICL, as measured by nDCG@10 in the In-Context Learning Embedding and Reranker Benchmark (ICLERB), vs. model size in millions of parameters. cm-rerank-mxbai-rlaif-v0.1 (ours) is fine-tuned by RLRAIF and outperforms other approaches while being considerably smaller than most models.

出典:ICLERB / RLRAIF / arXiv ↗

概要

  • ICLERBは、文書を一つ加えたときのLLMの回答への効用を基準に、検索モデルの順位を評価する。
  • RLRAIFは、LLMによる効用評価を使い、問い合わせを抑えながら検索モデルを調整する提案である。
  • 評価時の質問は未見だが文書群は学習時と共通で、新規文書や複数文書の同時利用は評価範囲外である。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、ICLERB / RLRAIFの仕組み

左から右へ読む
  1. 01質問・正解・候補文書
  2. 02LLMで文書効用を採点
  3. 03順位を評価・学習
  4. 04質問別の文書順位

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

検索した文書を質問とともに大規模言語モデル(LLM)へ渡す検索拡張生成(RAG)では、文書の順位が回答に影響する。質問と話題が近い文書でも、正解に必要な情報を含むとは限らないため、意味の近さだけを測る評価では選択の良し悪しを捉えきれない。

この研究は、文書を加えることでLLMの課題遂行がどれだけ助けられるかに注目する。質問への回答に有用な文書を上位に置けるかを、検索モデルの評価と学習の両面から扱う。

手法

ICLERBは、質問、正解回答、候補文書から、検索モデルが付けた順位を評価する。各文書を質問に一つ加え、LLMが正解回答に割り当てる確率を、文書がない場合と比較する。その変化に基づくDPOという効用指標を使う。DPOは文脈に応じて不正解回答の確率も考慮でき、正解回答は採点用の情報であって、未知の質問を検索する際の入力ではない。

RLRAIFは、LLMによる文書の効用評価を使って検索モデルを調整する手法である。例えば、正解付きの規程に関する質問と候補文書を用いて調整し、新しい質問には役立つと予測した文書を順位付きで返す。著者らは、未評価の文書を調べる探索と有望な文書の活用を組み合わせると述べるが、確認できた本文の範囲では具体的な更新手順までは分からない。

新規性

ICLERBは、検索モデルが文書を質問との関連性だけで順位付けできるかではなく、その文書を加えた際のLLMの回答への効用に沿って順位付けできるかを評価する。埋め込みモデルと再順位付けモデルを同じ目的で比較する枠組みである。

RLRAIFは、LLMから得る効用の評価を使い、検索モデルを調整する提案である。著者らは、未評価の候補を調べる探索と、有望な候補を使う活用を両立させ、LLMへの問い合わせを抑えると説明している。

従来手法との違い

BEIRやMTEBなど、本文で挙げられた既存の評価は、主に質問と文書の関連性や意味の近さを扱う。ICLERBでは、文書がLLMの正答を助ける度合いに基づく順位と、検索モデルの順位を比べるため、評価する目的が異なる。

先行するEPRやUDRも、言語モデルの反応を使って役立つ例を選ぶ。本文によると、EPRは候補を採点して学習用の正例・負例を作り、UDRは候補の採点と学習を反復する。RLRAIFとの同一条件での問い合わせ数の差は、確認できた本文の範囲では示されていない。

実験結果

著者らは、複数のデータセットとLLMで検索モデルの順位を評価する。質問と正解回答を学習用80%、評価用20%に分け、分割と実験を繰り返して平均を報告する設計だ。評価時の質問は未見だが文書群は共通で、抽出した評価対象について、データセット当たり平均約100万組の質問・文書対を採点すると記す。

順位の指標は上位を重視するnDCGで、図ではnDCG@10を示す。著者らは、既存評価との違いに加え、RLRAIFで調整した小規模モデルが大型の検索モデルを上回ったと報告する。確認できた本文の範囲では、個々のデータセット名、モデル別の得点、改善幅を確認できない。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:社内手順への質問に対し、回答担当者が参照する文書を順位付きで提示する。入力は社員の質問と承認済み手順書、出力は候補文書の順位とし、担当者が本文を確認して回答する。業務での改善は未検証の仮説である。

適用には、過去の質問、確認済みの正解回答、候補となる手順書を用意する。正解回答を採点用ラベルとして文書の効用を調べ、その評価で検索モデルを調整する想定だ。入力はテキストなのでセンサーは不要であり、文書形式と質問分布への適応は別途検証する。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:過去の社内質問と確認済み回答を集め、回答を正解ラベル、現行の手順書を候補文書とする。入力はテキストで、センサーは不要である。質問を学習用と評価用に分け、現行の検索順位を比較基準として保存する。

次に、正解回答の確率を取得できるLLMで文書の効用を採点し、その評価で検索モデルを調整する。未見の質問について、現行方式と調整後の方式の上位文書の順位指標、回答の正解率、採点に要したLLMへの問い合わせ数を測る。手順書を更新した場合は、更新後の文書でも別に確かめる。

限界と課題

著者らが明示するICLERBの評価範囲は、質問に文書を一つ加える場合である。評価時の質問は未見だが、文書群は学習時と共通で、新しく追加された文書は対象外となる。効用の採点には、回答の確率を取得できるLLMが必要である。

編集上の検証事項は、複数文書を同時に提示したときや、手順書の更新後にも順位が有効かどうかである。確認できた本文の範囲では、RLRAIFの問い合わせ予算、データセット別の改善幅、実務での運用費用は確認できない。導入先で効果と採点の手間を測る必要がある。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

ICLERB: In-Context Learning Embedding and Reranker Benchmark

In-Context Learning (ICL) enables Large Language Models (LLMs) to perform new tasks by conditioning on prompts with relevant information. Retrieval-Augmented Generation (RAG) enhances ICL by incorporating retrieved documents into the LLM's context at query time. However, traditional retrieval methods focus on semantic relevance, treating retrieval as a search problem. In this paper, we propose reframing retrieval for ICL as a recommendation problem, aiming to select documents that maximize utility in ICL tasks. We introduce the In-Context Learning Embedding and Reranker Benchmark (ICLERB), a novel evaluation framework that compares retrievers based on their ability to enhance LLM accuracy in ICL settings. Additionally, we propose a novel Reinforcement Learning-to-Rank from AI Feedback (RLRAIF) algorithm, designed to fine-tune retrieval models using minimal feedback from the LLM. Our experimental results reveal notable differences between ICLERB and existing benchmarks, and demonstrate that small models fine-tuned with our RLRAIF algorithm outperform large state-of-the-art retrieval models. These findings highlight the limitations of existing evaluation methods and the need for specialized benchmarks and training strategies adapted to ICL.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01