用一个生成模型同时完成推荐的检索与排序,提升效果且不增加计算量。
Killing Two Birds with One Stone: Unifying Retrieval and Ranking with a Single Generative Recommendation Model
- 将检索和排序统一为序列生成任务,实现信息无缝共享。
- 在多个数据集上显著超越现有模型,性能提升明显。
- 适合希望简化推荐系统架构的研究者与工程师。
推荐系统中传统的多阶段范式(包括检索与排序)常因阶段间信息丢失而影响性能。受自然语言处理中生成模型的启发,本文提出统一生成推荐框架(UniGRF),将检索与排序整合为单一生成模型。通过将两个阶段均视为序列生成任务,UniGRF 实现了无需额外计算成本的信息充分共享,且保持模型无关性。为增强阶段间协作,引入基于排序精度的增强模块,构建反馈循环;同时采用梯度引导的自适应加权器,动态平衡检索与排序的优化,实现同步性能提升。大量实验表明,UniGRF 在基准数据集上显著优于现有模型,验证了其在促进信息传递方面的有效性。消融实验进一步揭示,该框架不仅能高效协同两阶段,还能实现同步优化。UniGRF 为生成式推荐系统提供了一个高效、可扩展且兼容性强的解决方案。
原文摘要 · Abstract (English)
In recommendation systems, the traditional multi-stage paradigm, which includes retrieval and ranking, often suffers from information loss between stages and diminishes performance. Recent advances in generative models, inspired by natural language processing, suggest the potential for unifying these stages to mitigate such loss. This paper presents the Unified Generative Recommendation Framework (UniGRF), a novel approach that integrates retrieval and ranking into a single generative model. By treating both stages as sequence generation tasks, UniGRF enables sufficient information sharing without additional computational costs, while remaining model-agnostic. To enhance inter-stage collaboration, UniGRF introduces a ranking-driven enhancer module that leverages the precision of the ranking stage to refine retrieval processes, creating an enhancement loop. Besides, a gradient-guided adaptive weighter is incorporated to dynamically balance the optimization of retrieval and ranking, ensuring synchronized performance improvements. Extensive experiments demonstrate that UniGRF significantly outperforms existing models on benchmark datasets, confirming its effectiveness in facilitating information transfer. Ablation studies and further experiments reveal that UniGRF not only promotes efficient collaboration between stages but also achieves synchronized optimization. UniGRF provides an effective, scalable, and compatible framework for generative recommendation systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。