统一排序框架,解决生成式重排中自回归与非自回归的权衡问题。
UniRank: Unified List-wise Reranking via Confidence-Ordered Denoising
- 通过置信度有序去噪机制,迭代填充最可信槽位,融合双向建模优势。
- 在多个数据集上超越现有基线,工业平台测试提升平均使用时长0.159%。
- 适合需要高精度排序且兼顾效率的推荐系统场景,如短视频推荐。
列表级重排将请求相关的候选项集合排列成一个有序列表,以最大化用户满意度。现有生成式重排方法分为两类:自回归(AR)重排器从左到右逐步构建列表,能捕捉项间依赖关系,但存在错误传播问题;非自回归(NAR)重排器并行预测所有槽位,避免错误传播,但在槽位独立假设下削弱了项间交互建模能力。这引出核心问题:是否存在一种统一架构,融合两者优势并实现更强的重排性能?本文提出UniRank,一种统一的列表级重排框架,其推理变体可恢复AR与NAR重排器作为特例。UniRank将双向列表建模融入迭代去噪过程,并在每一步填充最自信的槽位。为实现该框架,引入任务引导扩散接口(TGD),在项级别进行去噪,并将预测限制在请求相关的候选池内。TGD将每个项的语义令牌聚合为单一嵌入,并直接对候选池评分每个槽位。在Amazon Books、MovieLens-1M和一个工业级短视频数据集上的实验表明,UniRank持续优于最先进的基线。在线A/B测试在真实工业平台上进一步验证其有效性,分别带来+0.159%的用户平均应用时长和+1.016%的分享率提升。
原文摘要 · Abstract (English)
List-wise reranking arranges a request-specific pool of candidate items into an ordered slate that maximizes user satisfaction. Existing generative rerankers fall into two paradigms: Autoregressive (AR) rerankers construct the slate left to right and capture inter-item dependencies in the exposure list, but they suffer from error propagation because early mistakes affect subsequent slots. Non-autoregressive (NAR) rerankers predict all slots in parallel and avoid error propagation, but they weaken inter-item interaction modeling under a slot independence assumption. This raises a central question: is there a unified architecture that combines the strengths of both paradigms and delivers stronger reranking performance? We answer this question with UniRank, a unified list-wise reranking framework whose inference time variants recover AR and NAR rerankers as special cases. UniRank integrates bidirectional slate modeling into an iterative denoising process and fills the most confident slot at each step. To instantiate this framework for reranking, we introduce the Task Grounded Diffusion Interface (TGD), which performs denoising at the item level and restricts prediction to the request-specific candidate pool. TGD aggregates each item's semantic tokens into a single item embedding and scores each slot directly against the candidate pool. Experiments on Amazon Books, MovieLens-1M, and an industrial short video dataset show that UniRank consistently outperforms state-of-the-art baselines. Online A/B tests on a real-world industrial platform further validate its effectiveness, yielding significant improvements of +0.159% in user average app-time and +1.016% in share-rate.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。