Google Research·· 23 天前精选AI 评分68
Google Research 提出 Retrieve-for-Train：利用强化学习蒸馏加速复杂 AI 搜索
Bypassing inference bottlenecks: Accelerating complex AI search with Retrieve-for-Train
AI 导读
Google Research 在 ICML 2026 发表论文提出 Retrieve-for-Train 框架，利用离线强化学习发现奖励对齐的查询扩展策略，并将其蒸馏至 53.9M 参数的扩散检索器中。该框架通过单次非自回归并行推理生成完整查询集，在保持多样性和对齐性的同时，相比自回归方法实现 12 至 20 倍的速度提升。
推荐理由
提出通过离线强化学习将复杂搜索策略蒸馏至轻量扩散模型，实现单次推理并大幅提升检索速度。
来源：Google Research · research.google