Google Research推出R4T扩散检索器

发生了什么

据MarkTechPost报道，Google Research推出了Retrieve-for-Train (R4T)框架，旨在解决搜索和推荐系统中查询扩展的问题。该框架使用强化学习离线学习良好的查询扇出行为，并将其蒸馏为小型扩散模型，以单次传递生成所有检索方向。相比自回归方法，R4T实现了12倍至20倍的速度提升，例如在批量大小为1024时，耗时从近50秒降至4.21秒。