UNREAL: Unifying Retrieval and Long-Context with a Single Model
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と初回トークン生成時間を削減しました。
論文の注目点
- UNREALは凍結されたデコーダー専用LLM上に軽量な検索ヘッドを学習させ、内部の残差ストリーム表現を活用して根拠チャンクのスコアリングと選択を行う。
- 本フレームワークは、2,100万チャンクのWikipediaに対する全コーパス検索と、プロンプト内での長文コンテキスト根拠選択を単一のモデルアーキテクチャ内に統合する。
- 実証評価において、外部リトリーバーパイプラインや全コンテキスト推論を上回る検索精度と長文QA性能を示し、長大なコンテキスト長において効率性も向上させた。
参考リンク
- arXiv: https://arxiv.org/abs/2610.08463v1
- Fugu-MT: https://fugumt.com/fugumt/paper_check/2610.08463v1
- Hugging Face Papers: https://huggingface.co/papers/2610.08463