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

since
平成38年9月29日

← 検索結果に戻る

文書を検索せずに答えるモデルを、知識注入・回答調整・能力回復で育てる

IAR

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

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

原論文の図の説明を読む

Figure 1: The overview of IAR . Vanilla SFT learns from generated QA pairs and covers only the facts selected by those questions. CPT+SFT models full document token streams before a separate QA alignment stage. IAR injects document knowledge through continuation, rewrite, and instruction-conditioned reconstruction objectives, aligns the injected model with answer-only QA supervision, and recovers general ability via post-hoc model merging. The final model is selected by balancing retrieval-free domain QA performance and general capability retention.

出典:IAR / arXiv ↗

概要

  • IARは文書内容の学習、質問回答への調整、一般能力の回復を三段階に分ける。
  • 著者らの主比較では、八条件中七条件で質問回答のみの追加学習を四指標すべてで上回った。
  • 学習後の知識の正確さに加え、一般的な指示へ対応する能力も評価する必要がある。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、IARの仕組み

左から右へ読む
  1. 01固定文書と質問回答
  2. 02文書内容を注入
  3. 03回答形式に調整
  4. 04統合済み回答モデル

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

研究の背景

固定された文書群について、質問時に本文を検索して渡せない場合、モデルは追加学習で覚えた内容を使って答える必要がある。質問と回答だけを学ばせると、学習用の質問に選ばれなかった事実へ触れる機会が少ない。

一方、文書を多く学ばせても、質問への答え方や一般的な指示への対応が弱まることがある。専門文書の正答率と、通常の仕事で使う応答能力を併せて見る課題である。

手法

IARのInjectでは、文書の前半から続きを出す、要約や骨子から本文を書き直す、短い読解指示から本文を再構成する課題を作り、文書内容を学ばせる。Alignでは、文書由来の質問を入力し、回答部分だけを学習対象にして、知識を質問回答の形で使えるようにする。

Recoverでは、追加学習後のモデルと元の指示調整済みモデルの重みを、複数の既存手法で統合する。候補は検証用の質問と一般課題で選び、最後に取り分けた評価用質問で測る。例えば社内規程の文書を学んだ後、規程への回答と通常の文章指示への対応を両方点検する流れである。

新規性

IARの新規性は、新しい文書再構成の損失関数やモデル統合法そのものではない。著者らは、文書を広く学ぶInject、質問に答えるAlign、一般能力を戻すRecoverを分け、各段階の効果を検証した。

最終モデルは、専門文書の質問での正答率を主に見ながら、指示への追従や一般的な課題の成績も確認して選ぶ。専門知識の獲得と能力低下を別々に測る構成である。

従来手法との違い

基準のVanilla SFTは、元の指示調整済みモデルを文書由来の質問と回答だけで追加学習する。IARは、その前に文書の続きや書き直しを学び、最後に元モデルとの統合を試す。学習量だけの影響を見るため、質問回答の学習量を合わせたBudgetMatchとも比べる。

本文はCPTと質問回答学習の組み合わせも示すが、その主な比較行は別のBaseモデルから始まる。したがって、その差だけからInjectの方式単独の効果は判断できないと著者ら自身が説明する。

実験結果

著者らはCommon Corpus由来のCCで学習用1万4258問、評価用750問、CCIで学習用1万926問、評価用575問を用いた。Llama、Phi、Qwen、SmolLMの四系列を比較し、本文なしの専門質問の正答率とIFEval、MMLU、MSBenchを測った。

Vanilla SFTに対し、八つのデータ・モデル条件のうち七つで四指標すべてが改善したと報告する。平均差は専門質問で3.6ポイント、一般三指標の平均で12.1ポイント。例としてQwen3-4BのCC正答率は42.4%から50.5%になった。

応用の可能性

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

編集上の応用案:社内の固定された業務手順集について、検索を伴わずに短い質問へ答える専用モデルを試す。入力は版を固定した手順書と、そこから作った学習用の質問回答で、出力は質問に対する回答にする。

文書の内容を使ってInjectし、よくある問い合わせでAlignした後、一般的な指示への応答が落ちていない候補をRecoverで選ぶ。回答を手順書の正本へ照合し、更新時に再学習が必要かも確認する。

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

導入時の検証

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

導入時の検証案:版を固定した少量の文書から質問を作り、学習用と評価用を分ける。元モデル、質問回答だけの追加学習、IARの三条件を同じ評価質問で比べ、文書の節ごとに正答率を記録する。

同時に、通常の指示への対応を測る別の課題を用意し、誤答や回答不能を人が点検する。文書を更新した後も同じ問いを再評価し、古い記述を答える頻度が許容範囲か確認する。

限界と課題

著者らは、最終モデルが全ての比較法や指標を常に上回るとは主張していない。CCIのPhiでは、Vanilla SFTに比べて専門質問の正答率とMMLUがわずかに下がる。大きいQwenモデルでも、一般能力の回復は元モデルの水準まで一様には達しない。

編集上の確認事項は、文書更新後の古い回答、誤答を自信ありげに返す場合、評価者モデルによる採点の揺れである。論文の専門質問はモデルが本文を見ずに答える設定であり、回答の出典提示は別途検証が要る。

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

出典

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

Inject, Align, Recover: Staged Post-Training for Retrieval-Free Document Knowledge Internalization

Large language models often fail to answer questions about a bounded document collection when the source documents are not retrieved at inference time. We study this setting as document knowledge internalization: converting a fixed corpus into usable parametric knowledge for retrieval-free question answering. We propose IAR (Inject, Align, and Recover), a three-stage post-training framework that separates structured document knowledge injection, QA behavior alignment, and general ability recovery. Unlike conventional continued pretraining, Inject converts source documents into continuation, rewrite, and instruction-conditioned reconstruction objectives. Align then adapts the injected model with answer-only QA supervision, while Recover merges the domain-adapted model with the base instruction model to recover general capabilities. Across Common Corpus (CC) and CCI, and across Llama, Phi, Qwen, and SmolLM model families, IAR improves the domain-primary domain-general frontier for retrieval-free document internalization. In the main comparison, IAR improves over Vanilla SFT on all four reported metrics in 7 of 8 dataset-model settings, with average gains of 3.6 percentage points in domain QA accuracy and 12.1 percentage points in mean general performance across IFEval, MMLU, and MSBench. Extended CC baselines show that LoRA and FAPM can win individual general metrics, but among methods that also reach leading or near-leading domain internalization, IAR retains one of the strongest general profiles.

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