用强化学习生成高质量多结果检索,速度提升十倍
Efficient, Property-Aligned Fan-Out Retrieval via RL-Compiled Diffusion
- 先用强化学习训练生成器,再用其合成训练数据
- 扩散模型仅需单次前向计算,实现高效多结果生成
- 适合需要快速生成多样推荐的电商与音乐场景
许多现代检索任务为集合式目标:给定广泛意图,系统需返回一组优化高阶属性(如多样性、覆盖度、互补性)的结果,同时保持与固定数据库的一致性。现有监督数据集仅关注单结果召回,无法捕捉集合级目标。为此,常采用扇出检索生成多样化子查询。尽管强化学习可通过交互优化集合目标,但部署强化学习调优的大模型在推理时成本过高。相反,基于扩散的生成检索可在嵌入空间中实现高效单次扇出,但需目标对齐的训练目标。本文提出R4T(Retrieve-for-Train),通过三步流程:(i) 以复合集合奖励训练扇出大模型;(ii) 合成目标一致的训练对;(iii) 训练轻量级扩散检索器以建模集合输出的条件分布。在包含精心策划物品集的大规模时尚与音乐基准上,R4T相较强基线显著提升检索质量,同时将查询时扇出延迟降低一个数量级。
原文摘要 · Abstract (English)
Many modern retrieval problems are set-valued: given a broad intent, the system must return a collection of results that optimizes higher-order properties (e.g., diversity, coverage, complementarity, coherence) while remaining grounded with respect to a fixed database. Set-valued objectives are typically non-decomposable and are not captured by existing supervised (query, content) datasets which only prioritize top-1 retrieval. Consequently, fan-out retrieval is often employed to generate diverse subqueries to retrieve item sets. While reinforcement learning (RL) can optimize set-level objectives via interaction, deploying an RL-tuned LLM for fan-out retrieval is prohibitively expensive at inference time. Conversely, diffusion-based generative retrieval enables efficient single-pass fan-out in embedding space, but requires objective-aligned training targets. To address these issues, we propose R4T (Retrieve-for-Train), which uses RL once as an objective transducer in a three-step process: (i) train a fan-out LLM with composite set-level rewards, (ii) synthesize objective-consistent training pairs, and (iii) train a lightweight diffusion retriever to model the conditional distribution of set-valued outputs. Across large-scale fashion and music benchmarks consisting of curated item sets, we show that R4T improves retrieval quality relative to strong baselines while reducing query-time fan-out latency by an order of magnitude.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。