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

since
平成38年9月29日

← 検索結果に戻る

途中の解法を書かずに学ぶResa:モデル内部の特徴を使う推論調整

Resa / SAE-Tuning

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

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

原論文の図の説明を読む

Figure 2 : Two-Stage Pipeline of SAE-Tuning   The procedure begins with SAE training (Left), where an SAE is trained to capture reasoning features from a source model with a trigger dataset. During SAE-guided SFT (Right), the trained SAE is then frozen and inserted into a target model. An elicitation dataset is used to guide a SFT process to elicit the reasoning abilities in the target model. Notably, the trigger and elicitation datasets are usually the same CoT-free data. See Section 2.1 for a description of each component.

出典:Resa / SAE-Tuning / arXiv ↗

概要

  • Resaは、正解を確認した質問と最終回答からSAEを学び、対象モデルの調整に使う。解法の途中経過を記した学習データは使わない。
  • 著者らの6課題平均では、基礎モデルからSAEを学んだResa-STILL-v5が48.06%、強化学習済みTina-STILLが48.16%だった。
  • 実務利用を試すには正解ラベル付きの実務問題を用意し、通常の教師あり微調整と未使用問題で比較する必要がある。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Resa / SAE-Tuningの仕組み

左から右へ読む
  1. 01正解付き問答
  2. 02中間表現を収集
  3. 03SAEを学習
  4. 04対象モデルを調整
  5. 05回答を生成

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

研究の背景

Resaは、言語モデル内部の数値表現から推論に関わる特徴を捉え、モデルの調整に使う手法である。著者らによると、強化学習には計算資源がかかり、解法の途中経過を使う教師あり微調整では質の高い記録を集める負担がある。

成績が上がっても、内部表現がどう変わったかは調べにくい。そこで著者らは、正解を確認できる質問と最終回答だけを用い、学習費用を抑えながら内部の特徴も調べられる手順を検討した。

手法

まず、正解を確認した質問と最終回答を元モデルに入力し、指定した層の中間表現を集める。例えば数学の問題と正答を与えるが、解法の途中経過は与えない。SAE(疎なオートエンコーダー)は、この数値表現を少数の特徴の組み合わせで再現するモデルで、元の表現に近づくよう学習する。

次にSAEの重みを固定し、構造と規模が合う対象モデルの層に挿す。同種の問答を使い、低ランクアダプターという追加の小さな重みだけを学習して、SAEを挿した場合と挿さない場合の次の語の確率分布を近づける。調整後はSAEを外し、対象モデルから回答を得る。

新規性

著者らが提案するSAE-Tuningは、正解を確認した質問と最終回答だけを使い、元モデルの中間表現から学んだSAEを対象モデルの調整に利用する。解法を順に記した学習データを必要としない構成である。

基礎モデル自身を元モデルに選び、SAEも一から学習する構成を示した。SAEの特徴が個々の推論手順に一対一で対応すると証明したわけではない。

従来手法との違い

STILLを使う実験では、Resa-STILL-v1の6課題平均は47.28%、強化学習済みのTina-STILLは48.16%、同じ途中経過なしのデータで通常の教師あり微調整をしたモデルは39.00%だった。最後の比較は、SAEを使う調整手順の寄与を調べる対照実験である。

別構成のResa-STILL-v5は、基礎モデルを元モデルとしてSAEを一から学び、平均48.06%を得た。v1とv5はSAEの学習方法と元モデルが異なるため、数値だけで両者の差の原因を決めることはできない。

実験結果

著者らはAIME24・AIME25・AMC23・MATH500・GPQA Diamond・Minervaの6課題を、例題を示さないゼロショットのPass@1で評価した。Pass@1は一度の回答が正解する割合である。Resa-STILL-v5の6課題平均は48.06%、強化学習済みTina-STILLは48.16%だった。

同じ表でv5のAIME24は33.33%、AMC23は70.00%であり、課題ごとの成績には差がある。著者らの費用表では、v5のSAEと対象モデルを合わせた学習費用は約1ドル、評価費用は約4ドルと推定された。これらは掲載されたモデル、課題、計算条件での結果である。

応用の可能性

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

編集上の応用案:社内の定型的な数値照合で、文章の計算条件から回答候補を作る用途を試す。問題文を入力し、最終的な数値と根拠を確認するための元帳票の参照先を担当者に渡し、担当者が計算と数値を確定する。

適応には実務の問題文と検証済みの最終回答を組にした学習データが必要で、正解の数値をラベルとする。文書入力だけを扱う想定なので追加のセンサーは使わない。実務への効果は未検証であり、元モデルと対象モデルの選定後にSAE学習と調整を試す案である。

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

導入時の検証

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

導入時の検証案:実務の問題文と担当者が確認した最終回答を集め、学習用と未使用の評価用に分ける。最終回答を正解ラベルとし、文書以外のセンサー入力は使わない。元モデルと対象モデルの構造・規模が合うことを確かめ、SAE学習と対象モデルの調整を小規模に行う。

比較基準は、同じ問答で通常の教師あり微調整をしたモデルにする。未使用の問題で一度の回答の正答率を比べ、回答時間と学習費用も記録する。表記や計算条件が異なる帳票を含め、担当者が修正した誤答の種類を確認する。

限界と課題

著者らが明記する実装上の条件は、SAEを対象モデルの層へ直接挿すため、元モデルと対象モデルの基本構造と規模をそろえることだ。この版で同時に挿すSAEは最大一つで、学習後の評価時には取り外す。

編集上の検証事項は、数学・科学の課題での正答率が実務の数値照合にも当てはまるか、誤った正解ラベルや帳票の表記ゆれにどう反応するかである。確認できた本文の範囲では、こうした実務条件の結果は示されていない。費用の推定も著者らの実験条件に基づく。

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

出典

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

Resa: Transparent Reasoning Models via SAEs

How cost-effectively can we elicit strong reasoning in language models by leveraging their underlying representations? We answer this question with Resa, a family of 1.5B reasoning models trained via a novel and efficient sparse autoencoder tuning (SAE-Tuning) procedure. This method first trains an SAE to capture reasoning abilities from a source model, and then uses the trained SAE to guide a standard supervised fine-tuning process to elicit such abilities in a target model, all using verified question-answer data without any reasoning traces. Notably, when applied to certain base models before further RL post-training, SAE-Tuning retains >97% of its RL-trained counterpart's reasoning performance while reducing training costs by >2000x to roughly \$1 and training time by >450x to around 20 minutes. Furthermore, when applied to lightly RL-trained models (e.g., within 1 hour on 2 GPUs), it enables reasoning performance such as 43.33% Pass@1 on AIME24 and 90% Pass@1 on AMC23 for only around \$1 additional cost. Surprisingly, the reasoning abilities extracted via SAEs are potentially both generalizable and modular. Generality means abilities extracted from one dataset still elevate performance on a larger and overlapping corpus. Modularity means abilities extracted from Qwen or Qwen-Math can be attached to the R1-Distill model at test time, without any retraining, and yield comparable gains. Extensive ablations validate these findings and all artifacts are fully open-sourced.

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