ITPub博客

首页 > 人工智能 > 人工智能 > 搜推精排2025代表工作汇总——序列&特征的Token化建模

搜推精排2025代表工作汇总——序列&特征的Token化建模

原创 人工智能 作者:圆圆的算法笔记 时间:2026-06-09 16:04:28 0 删除 编辑
点关注,不迷路,用心整理每一篇算法干货~
后台回复“星球”加入圆圆算法知识星球~
(已有1100+同学加入学习)


最近一年,搜推中的精排模型迭代逐渐走向大模型方向,通过参数scaling law、序列和特征交叉建模等方式,不断释放算力对搜推模型效果的提升。从统一视角看,这些方法有很多相似性。它们的底层围绕特征和序列token化展开,上层通过多层的Transformer或MLP以及一些从LLM迁移的优化手段提升模型复杂度,验证推荐系统中的scaling law。今天这篇文章,就给大家汇总一下2025年字节在精排模型结构优化的代表工作,以及这些工作之间背后的关联。


1

特征交叉—Rankmixer

字节提出的Rankmixer重点进行特征交叉层面的优化。其前身是谷歌DeepMind提出的Hiformer。Hiformer的核心思路是使用Transformer的结构进行特征交叉的计算,同时区别于NLP中每个token语义空间相同的应用场景,对每个特征使用独立的QKV映射矩阵进行映射,作为后续attention的输入表征。

Rankmixer在这个思路的基础上,借鉴了MLP-Mixer的建模思路,将Transformer结构替换成MLP结构。其核心思路是将attention这种两两相似度计算的方式,升级为MLP-Mixer这种非参数化的计算方式。

具体计算逻辑为,对于底层输入特征,将其分成多组,每组特征使用MLP映射成多个token表征,每组特征使用单独的MLP进行映射。接下来,借鉴MLP-Mixer的建模思路,通过token矩阵的转置操作,直接将部分各个token表征进行混合。

这种方式相当于强行让不同token之间不同维度表征进行互换,也起到了token间信息交互的目的。在MLP-Mixer原文中,也验证了其和Transformer效果差异不大,但是计算效率有明显提升。此外,文中也引入了Sparse-MoE等结构进一步对参数量进行scaling up。


2

长序列建模—Longer

长序列建模一直是推荐领域的核心话题,如何在计算复杂度提升量可控的前提下,建模更长周期的用户行为序列,是长序列建模研究的关键问题。在之前,业内通常采用两阶段进行训练,如MARM的全序列预训练大模型+表征引入精排模型的方式,或者阿里的ESU、GSU两阶段方法,以及快手的TWIN方式。这些方法虽然引入了更长的序列,但两阶段并不是最优解。

字节的Longer给出了一种端到端长序列建模方案。模型的输入包括global tokne和用户序列token两部分。Global token是用户属性信息、候选item信息、交叉特征等组成的多个token,其实就是模型中的其他基础特征。序列部分文中将长序列分成多组,每组多个行为,组内的行为子序列使用内部的一个transformer进行融合。通过这种方式,大幅缩短了输入序列的长度,组内由于子序列较短,attention计算的复杂度也不高。

在此基础上,后续采用cross attention和self attention相结合的方式进行建模。首先使用一层cross attention,采用global token,以及行为序列中最近的k个token作为query,检索序列和特征中的相关信息。接下来,使用多层self-attention中,则在上一层输出的token内部计算,进一步提取高阶的交叉信息。通过这种方式,以较小的代价首先将完整的长序列信息通过cross-attention吸收到较少的token中,再通过这些token的self-attention进一步提取高阶信息。


3

特征&序列统一建模—OneTrans

字节的OneTrans可以看成是字节前序的rankmixer、longer的融合版本,将特征交叉和序列建模放在一起进行优化。从OneTrans的结构中,可以看到很多rankmixer和longer的建模方法。

首先,OneTrans将序列和基础特征放到一起建模,对于非序列特征使用类似rankmixer的方法进行分组(人工分组或自动分组)得到多个token;对于序列特征将各种类型行为拼在一起,并用[SEP]特殊token进行不同类型行为的区分。接下来是多层的Transfomrer模块,这里对于序列特征使用共享参数进行QKV和FFN的映射,对于非序列特征,每个token使用单独的参数进行映射,类似rankmixer和hiformer中的操作。

最后,类似Longer的思路,每一层只选择最近的几个序列token进入下一层,减少计算量,同时也让模型将最关键的信息逐层汇集到少量的token上。


4

总结

从字节的这三篇代表工作可以看出,基础特征建模和序列建模正在走向统一融合的过程,借鉴NLP、CV领域达模型的经验,将底层特征统一token化,再使用Transformer建模,配合序列压缩的方法,实现高效的精排模型Scaling Up,是未来精排进一步迭代的发展方向。


END




后台留言”交流“,加入圆圆算法交流群~
知识星球提供一文贯通笔记、经典代码解析、问答服务、新人入门,已有1100+小伙伴加入,价格随人数增加和内容丰富上涨,感兴趣的同学尽早加入~
【历史干货算法笔记】
2024时序预测都有哪些经典工作——汇总篇
生成式模型入门:一文讲懂3大类生成式模型
Sptial-Temporal时空预测总结:建模思路、优化方法梳理
时序预测顶会论文数据集、数据处理方法、训练方法汇总
时间序列预测实战方法概述:从数据到模型
Informer模型结构和代码解析

如果觉得有帮助麻烦分享在看点赞~  

来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/70050176/viewspace-3173379/ ,如需转载,请注明出处,否则将追究法律责任。

下一篇: 没有了~
请登录后发表评论 登录
全部评论

注册时间:2026-05-20

  • 博文量
    4
  • 访问量
    5729