Fugu-MT: arxivの論文翻訳

UNREAL: Unifying Retrieval and Long-Context with a Single Model

著者 Edan Kinderman, Elad Hoffer, Yochai Blau, Brian Chmiel, Ron Banner, Daniel Soudry, Boris Ginsburg
所属 NVIDIA / Technion – Israel Institute of Technology
カテゴリ Method / Retrieval / Unified retrieval with frozen LLM, Task / Long-Context Reasoning / Evidence selection over differing scales, Application / RAG Enhancement / Combining retrieval and long-context reasoning
ライセンス CC BY 4.0

Abstractの概要

UNREALは、単一の凍結されたデコーダー専用LLMをコーパス検索と長文コンテキスト推論の双方に用いる、モデルネイティブな根拠選択フレームワークです。この手法は候補チャンクをLLMでエンコードし、内部の残差ストリーム状態から検索クエリを導出するとともに、検索トークンと層混合の重みのみを学習することで、追加される学習可能パラメータを50万未満に抑えています。本論文では、30億トークン・2,100万チャンクのインデックスに対するWikipedia全コーパス検索と、多数の妨害情報の中に適切な根拠がスパースに存在する長文コンテキストベンチマークにおいてUNREALを評価しています。これらの設定を通じて著者らは、検索と長文コンテキスト推論が異なるスケールにおける同一の根拠選択問題として統合できることを実証しています。

新規性

主な新規性は、単一の凍結されたLLMが自身の内部表現を用いて、コーパス規模の検索と長文コンテキストプロンプトの双方で明示的に根拠を選択する点にあります。外部のリトリーバーに依存する標準的なRAGパイプラインとは異なり、UNREALは内部の残差状態から直接検索クエリを導出し、バックボーンのファインチューニングを行うことなく、Dense、線形アテンション、状態空間ハイブリッドの各バックボーンに一様に適用できます。

成果

2,100万チャンクのWikipediaインデックスにおいて、評価された4つのUNREALバックボーンすべてが既存のDenseおよびハイブリッドな検索・再ランキングのベースラインを上回り、recall@10をHotpotQAで49.1%から73.2%、2WikiMultiHopQAで31.7%から60.1%、MuSiQueで8.8%から14.4%へと向上させました。長文コンテキストベンチマークにおいても同一メカニズムにより、128Kトークン時のNoLiMa精度を1.0%から24.83%へ、256K時のLV-Eval F1を49.97%から54.66%へと向上させ、約32Kトークン以降では全コンテキスト推論と比較してFLOPsと初回トークン生成時間を削減しました。

論文の注目点

  1. UNREALは凍結されたデコーダー専用LLM上に軽量な検索ヘッドを学習させ、内部の残差ストリーム表現を活用して根拠チャンクのスコアリングと選択を行う。
  2. 本フレームワークは、2,100万チャンクのWikipediaに対する全コーパス検索と、プロンプト内での長文コンテキスト根拠選択を単一のモデルアーキテクチャ内に統合する。
  3. 実証評価において、外部リトリーバーパイプラインや全コンテキスト推論を上回る検索精度と長文QA性能を示し、長大なコンテキスト長において効率性も向上させた。

参考リンク

このページはGPT-5、Claude Opus 4、Gemini 3、Gemini 3.1 Flash Image 及びその上位バージョンなどの生成AIを用いて作成されています。内容の保証は一切できません。