跳到正文
Google Research·· 24 天前AI 评分36

Google Research 提出 Retrieve-for-Train:用 RL 编译扩散检索器绕过推理瓶颈

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

AI 导读

Google Research 提出 Retrieve-for-Train 框架,用离线强化学习发现奖励对齐的查询扇出并编译为监督信号,再蒸馏进一个 53.9M 参数的扩散检索器,实现推理时单次非自回归的查询扇出,无需消耗 CoT 推理 token。

来源:Google Research · research.google