OneRank:面向多任务推荐的统一原生Transformer排序架构
摘要
OneRank提出了一种原生Transformer的多任务排序框架,该框架将特征编码与预测相结合,以减少任务间干扰并提升推荐系统中的排序性能。
查看缓存全文
缓存时间: 2026/06/16 11:33
Paper page - OneRank: 统一原生Transformer多任务推荐排序架构
来源:https://huggingface.co/papers/2606.16838 作者:
, , , , , , , , ,
摘要
OneRank 提出了一种原生 Transformer 的多任务学习框架,将特征编码与预测融为一体,旨在减少任务间干扰,提升推荐系统中的排序性能。
多任务学习 (https://huggingface.co/papers?q=Multi-task%20learning)(MTL)在推荐系统 (https://huggingface.co/papers?q=recommender%20systems) 中不可或缺,它能促进不同用户反馈间的互补学习。尽管现代工业实践已从 DNN 转向以 Transformer (https://huggingface.co/papers?q=Transformer) 为核心的架构来强化序列建模与扩展能力,但仍将特征编码与多任务预测分离,将 Transformer (https://huggingface.co/papers?q=Transformer) 视为与任务无关的编码器。这种设计从根本上限制了性能与可扩展性,原因是:(1) 在异质任务目标下形成信息瓶颈;(2) 引发梯度干扰,导致跷跷板现象;(3) 迫使数据流转换,使基于注意力的上下文自适应表示学习被转换为具有不兼容信息读写动态的静态前馈任务预测。我们提出 OneRank,一种原生 Transformer (https://huggingface.co/papers?q=Transformer) 的多任务排序框架,消除了编码器与预测器的分离,引入任务私有通道 (https://huggingface.co/papers?q=task-private%20channels) 用于前向表示学习与反向优化,在实现任务专门化学习的同时减少任务间干扰。在前向传播中,OneRank 通过任务条件信息选择 (https://huggingface.co/papers?q=task-conditioned%20information%20selection)、候选感知上下文化 (https://huggingface.co/papers?q=candidate-aware%20contextualization) 以及受控跨任务交互 (https://huggingface.co/papers?q=cross-task%20interaction) 自底向上学习任务特定表示。在反向传播中,跨任务梯度分离 (https://huggingface.co/papers?q=gradient%20detachment) 将任务私有的参数更新与共享知识提取模块隔离,防止负迁移 (https://huggingface.co/papers?q=negative%20transfer)。我们进一步用动态匹配评分 (https://huggingface.co/papers?q=dynamic%20matching-based%20scoring) 取代静态的任务特定 MLP 评分器,实现上下文感知的个性化排序。通过在 Transformer (https://huggingface.co/papers?q=Transformer) 堆栈内部完成多任务推理,OneRank 建立了一个统一且可扩展的架构范式。在大规模工业数据集上的离线与在线实验表明,OneRank 在保持计算效率的同时显著优于最先进的基线方法。
查看 arXiv 页面 (https://arxiv.org/abs/2606.16838) 查看 PDF (https://arxiv.org/pdf/2606.16838) 添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2606.16838)
在您的 agent 中获取这篇论文:
hf papers read 2606\.16838
没有最新的 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
引用本论文的模型0
没有模型链接到本论文
请在模型 README.md 中引用 arxiv.org/abs/2606.16838 以从此页面链接。
引用本论文的数据集0
没有数据集链接到本论文
请在数据集 README.md 中引用 arxiv.org/abs/2606.16838 以从此页面链接。
引用本论文的 Spaces0
没有 Space 链接到本论文
请在 Space README.md 中引用 arxiv.org/abs/2606.16838 以从此页面链接。
包含本论文的收藏集0
没有收藏集包含本论文
请将本论文添加到一个收藏集 (https://huggingface.co/new-collection) 中以从此页面链接。
相似文章
轻量级排序头:加速生产推荐系统中的多任务实验
本文介绍了轻量级排序头框架,通过启用动态任务注入,无需重新训练骨干模型,来加速生产推荐系统中的多任务实验,从而在YouTube规模下将迭代周期从数周缩短至数天。
扩展更多,收缩更少:为推荐系统中密集扩展塑造有效秩动态
本文提出RankElastor,一种新颖的架构,通过引入参数化全混合和GLU改进的P-FFN,缓解推荐模型密集扩展中的嵌入坍塌,实现鲁棒扩展并在大规模数据集上提升性能。
多模态重排序是一个System One决策
论文提出,多模态重排序可以被构建为一个由通用视觉-语言模型在单次前向传播中完成的System One单选问题,实验表明零样本的Qwen3.8-27B在多个检索基准上达到或超越了专门的有监督重排序模型。
Transformer的熵界:为何静态秩失效而注意力原生秩恢复
本文介绍了熵界(Entropic Bound),这是一种针对Transformer的任务内在容量的谱度量,证明了令牌混合算子的内在秩为所需模型容量提供了一个紧的下界。本文表明,虽然从线性注意力到真实注意力的简单迁移会失败,但注意力原生的内在秩能恢复完整的理论结构。
UMER:通过配对感知判别推理统一嵌入与排序以实现通用多模态检索
UMER 引入了一个统一的多模态检索框架,通过配对感知判别推理结合嵌入与排序,在 MMEB-V2 基准测试中达到了最先进的性能。