跳到正文
原文
Google Research Blog·· 18 天前精选AI 评分66

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 Blog · research.google