跳到正文
Google Research·· 2026-09-16精选AI 评分41

Google Research 提出 Retrieve-for-Train:用离线强化学习蒸馏扩散模型加速搜索推理

Bypassing inference bottlenecks: Accelerating complex AI search with Retrieve-for-Train

AI 导读

Google Research 在 ICML 2026 论文中提出 Retrieve-for-Train 框架,先用离线强化学习让 Gemma3-4B 与 Qwen3-4B 生成满足多样性、对齐和 groundedness 的子查询,再蒸馏为一个 53.9M 参数的非自回归扩散模型。

推荐理由

用离线强化学习把检索策略蒸馏成轻量扩散模型,给搜索类推理加速提供了一条不依赖增大推理预算的迁移路径。

来源:Google Research · research.google