ENTRY

◀︎ AI NEWS TOPへ

グーグルリサーチのリトリーブ・フォー・トレイン。RLで得たファンアウトを拡散検索へ蒸留し、推論を12〜20倍速くしたとする

2026.09.24
グーグルリサーチのリトリーブ・フォー・トレイン。RLで得たファンアウトを拡散検索へ蒸留し、推論を12〜20倍速くしたとする

グーグルリサーチ(Google Research)のペンチェン・ジャン(Pengcheng Jiang)氏(Student Researcher)とジュディス・ユエ・リ(Judith Yue Li)氏(Senior Research Engineer)らは、検索の「ファンアウト」を速くする枠組み「リトリーブ・フォー・トレイン(Retrieve-for-Train、R4T)」を、2026年9月15日付の研究ブログで紹介しました。論文はアーカイブ(arXiv)2603.06397です。国際機械学習会議(ICML)2026のポスターでも発表されています。共著者にはジアウェイ・ハン(Jiawei Han)氏らがおり、ジャン氏とハン氏はイリノイ大学アーバナ・シャンペーン校(UIUC)所属を含みます。以下の数字は、いずれもブログと論文側の説明です。

何が起きたか

ファンアウト検索は、広い検索意図から、関連する候補を「セット」として返す処理です。ゼロショットの大規模言語モデル(LLM)だけに任せると、言い換えが似た文ばかりになる「言い換えの崩壊(paraphrastic collapse)」が起きます。文を一つずつ出す自己回帰も、待ち時間が長くなります。

R4Tは三段階です。第一に、強化学習(報酬で望ましい答えへ寄せる学習)でファンアウト言語モデル(広い問い合わせを複数の短い問い合わせに分けるモデル。以下FOLM)を訓練します。報酬は、実在データへの寄り(根拠)、セットのばらつき(ベンディ・スコア)、元の意図との一致を組み合わせます。学習の枠はソフト・ジーアールピーオーとソフト・ピーピーオー(どちらも、複数候補を比べて方針を少しずつ直す強化学習のやり方)です。ベースはジェマ3-4B(Gemma3-4B)とクウェン3-4B(Qwen3-4B)で、副問い合わせは毎回ちょうど10本です。

第二に、訓練したFOLMで「問い合わせ→目標セット」の組を、本番の外であらかじめ(オフライン)合成します。人手ラベルは不要です。第三に、約5390万(53.9M)パラメータの拡散型の検索モデルへ教え込み(蒸留)し、一度の処理で全方向を出します(文を一語ずつ出さない非自己回帰)。中間の副問い合わせがないため、R4T拡散では根拠への寄りは評価対象外です。

評価は、ファッションのポリボア(Polyvore。画像と文章を同じ空間で比べる仕組み=CLIP系)と、音楽プレイリスト(音楽と文章を同じ空間で比べる仕組み=MuLan、産業用の社内データ)です。開かれた抽象検索の品質(Table1・ジェマ3-4B)では、R4TのFOLMのポリボア平均は49.1です。比較相手のBest-of-N(同じやり方を何度も試し、いちばん良い結果を選ぶ)は40.9、ゼロショットは38.5です。多様性は76.8(ゼロショット56.0)です。音楽平均は58.1(Best-of-N 49.2)です。

待ち時間は、バッチ8で自己回帰が約1.46秒、拡散が0.07秒です。バッチ1024では約50秒対4.21秒です。12〜20倍速いです。多様性の報酬を外すと「line ending line ending」のような崩れた文字列へ寄ります。ベンディ・スコアが、その崩れを抑える錨になります。

何が変わったか

公表されている説明 こちらで確認できていないこと
2026-09-15ブログ。アーカイブ2603.06397。ICML 2026ポスター 第三者による指標の再測定
強化学習でFOLMを訓練し、人手ラベルなしでデータ合成し、53.9M拡散へ蒸留(ブログ/論文) 訓練計算コストの独立再計測、学習レシピの完全再現手順
ソフト・ジーアールピーオー/ソフト・ピーピーオー。ジェマ3-4B/クウェン3-4B。副問い合わせは毎回10本(同) 他ベースモデルでの追試結果
開かれた抽象検索(Table1・ジェマ3-4B): ポリボア平均49.1(Best-of-N 40.9/ゼロショット38.5)。多様性76.8(ゼロショット56.0)。音楽平均58.1(Best-of-N 49.2)(同) 開かれた抽象検索の一部がLLM判定(ジェミニ系)であることの第三者検証
バッチ8で1.46秒→0.07秒。バッチ1024で約50秒→4.21秒。12〜20倍高速(同) 本番検索バー上の実測レイテンシ
多様性なしだと退化文字列。ベンディ・スコアが対策(同) 音楽データの外部再現(社内プロプライエタリ)
ICML 2026で発表 検索製品への搭載有無(ブログ・論文に明示なし)

音楽セットは社内データのため、外部からの同じ再実験はできません。

誰に関係するか

BtoBの部品カタログでは、規格の違う候補をセットで返し、待ち時間を短い拡散側に寄せます。自治体の手続き案内では、必要書類・窓口・期限をセットで先に出し、人が並びだけ直します。社内ヘルプデスクでは、権限・申請・FAQをセットで返し、対面に残す問いだけ残します。地域の観光案内では、宿・食事・移動の組み合わせを先にセットで作り、人が最終の並びだけ直します。

まとめ

広い意図をセットで返すとき、重い自己回帰のまま毎回考えさせず、報酬で一度そろえた振る舞いを軽い拡散側へ移します。ポリボア平均49.1や12〜20倍などの数字は、グーグルリサーチ側の報告です。

参考文献

Tags

タグ一覧へ ▶︎

Kamone
著者 Kamone

30歳で学者志望からITエンジニアに転身。通信・金融業界におけるインフラ領域を中心に、オンプレミスからクラウドまで幅広く経験。AWS学習コミュニティでは公認メンターとして教材制作や講師を務める。現在はリツアンSTCにて事業推進に携わり、AI活用とエンジニア支援の高度化に取り組んでいる。

Kamone
著者 Kamone

株式会社リツアンSTC社員。

1984年 東京都中野区生まれ。

大学中退後、工場や警備員などを経て、何もスキルが身についていないことに危機感を覚える。
2014年 何か手に職をつけなければと思い、電子専門学校からエンジニアになった弟を見て「コイツにできるなら俺もできる」と安易な考えで未経験からIT業界に飛び込む。
そこは「まともな案件に入れなければ営業としてこき使われるか中国に送られるらしい」と恐ろしい噂が飛び交う会社だった。
「とんでもないところに来てしまった…」と戦慄するも、運よく大手新聞社系のインフラチームから声がかかり、エンジニアとしてのスタートを切る。

その後、SESを転々としながら通信・金融業界のオンプレ・クラウド案件を中心にインフラエンジニアとして経験を積む。

2021年 AWS学習サービス「Cloudtech」に参加し、公認メンターとして書籍出版プロジェクトや教材制作・講師などのコミュニティ・サービス運営に関わる。

2023年 ゆとりーマンのYoutube動画で新興SESの闇とリツアンの存在を知り、転職。(当時所属していたSES企業が動画で紹介されていた特徴に当てはまっていて愕然)
その後、CloudTechとリツアンの橋渡し役を担い、合同プロジェクト「テラコヤテック」の運営に関わる。
2025年 リツアンの「エンジニアに一円でも高い報酬を」「会社は社員に使われるためにある」「出入り自由」といったスタンスに共鳴。エンジニアから転向し、リツアンの事業推進に参画。
現在に至る。

他人の文体を真似たり、他人が書いた文章を手直しするのが得意、という若干嫌な気持ちにさせる特技を持つ。
AI活用を鋭意研究中。
(リツアンいいとこ一度はおいで。)