引言:点击率模型的新难题
推荐系统的核心任务是猜中用户此刻最想要什么,而这取决于特征交互。近年来,自注意力机制被引入推荐排序模型,但计算复杂度高。为此,工业界开始采用Token-Mixing架构,用轻量级矩阵变换或MLP替代自注意力,降低计算开销。然而,推荐系统的特征token来自不同语义空间,直接混合效率低。HubMixer通过学习少量“隐枢纽”进行归纳、交互和读出,解决了这一问题。离线实验显示AUC全面领先,在线测试中简历提交转化率提升5.48%,已在快手上线。
方法概述:HubMixer整体架构与核心流程
HubMixer的整体架构如图1所示,分为四个阶段:Token化、枢纽归纳、枢纽交互、Token条件读出。
核心原理推导:归纳-交互-读出的数学表达
Token化后,输入表示为X = [x₁, x₂, ..., x_T] ∈ R^{T×d}。构造输入条件枢纽残差,得到全局摘要p。HubMixer通过交叉注意力和自注意力实现枢纽归纳和交互,最终通过Token条件读出将信息写回原始token。实验验证:离线AUC全面领先,在线转化率提升5.48%
HubMixer在快手短视频招聘业务上验证,覆盖4000万日活用户。任务为多目标排序,包含四个预测目标。离线实验中,HubMixer在四个目标上AUC最佳,参数量少于其他模型。在线A/B测试中,简历提交转化率提升5.48%,已全量上线。
深入分析:Token表示增强的证据与解释
HubMixer通过枢纽交互增强token表示。分析表明,HubMixer的token表示更新幅度大于TokenMixer,线性探测结果也更优,表明其特征表示更具任务相关性。
部署实践与未来展望:从快手招聘业务到更广阔的应用
HubMixer已在快手短视频招聘业务全量上线,在线A/B测试提升显著。其通用性适用于异构输入的交互建模,未来可扩展到搜索排序、广告点击率预估等场景。未来探索方向包括枢纽数量自适应选择、多层次枢纽结构和可解释性。龙迷三问
龙哥点评
论文创新性分数:★★★★☆
提出HubMixer架构,通过可学习的潜在中心(latent hubs)以归纳-交互-读出(induction-interaction-readout)范式组织异构特征交互,实现参数高效的特征交互建模。实验合理度:★★★★☆
离线使用AUC(四个任务分别报告),在线使用简历提交转化率。学术研究价值:★★★★☆
提出HubMixer架构,通过可学习的潜在中心(latent hubs)以归纳-交互-读出(induction-interaction-readout)范式组织异构特征交互,实现参数高效的特征交互建模;更关键的是问题定义是否可复用到同类任务。稳定性:★★★☆☆
现有材料未提供充分的极端条件、重复运行或扰动测试,稳定性暂按中性评价。适应性以及泛化能力:★★★☆☆
现有材料未完整展示跨数据集、跨场景或分布外实验,泛化能力仍需进一步验证。硬件需求及成本:★★★☆☆
交叉注意力的计算复杂度为O(TH),其中T为token数(32),H为中心数(16),远低于标准自注意力的O(T²)复杂度。整体计算量约为142.4M参数。复现难度:★★★☆☆
现有材料未确认完整代码、配置、数据处理脚本和权重是否齐备,复现难度暂按中性评价。产品化成熟度:★★★☆☆
论文验证以研究实验为主,真实部署中的时延、成本、维护和异常场景仍需补充验证。可能的问题:(1) 中心数量的选择需要调参;(2) 潜在中心的语义可解释性有限;(3) 对极端稀疏特征的处理能力有待进一步验证。
主要参考文献
推荐排序想涨点?
来龙哥读论文粉丝群,跟大家聊聊工业级的特征交互怎么玩~
扫描下方二维码或者添加龙哥助手微信号加群:kangjinlonghelper。一定要备注:研究方向+地点+学校/公司+昵称(如 推荐系统+北京+快手+龙哥),根据格式备注,可更快被通过且邀请进群。
『龙哥读论文』微信群目前包含:图像处理、大模型及智能体、自动驾驶及机器人、AI医疗及AI金融、推荐系统与广告,期待和你一起碰撞想法!
来龙哥读论文粉丝群,跟大家聊聊工业级的特征交互怎么玩~
扫描下方二维码或者添加龙哥助手微信号加群:kangjinlonghelper。一定要备注:研究方向+地点+学校/公司+昵称(如 推荐系统+北京+快手+龙哥),根据格式备注,可更快被通过且邀请进群。
『龙哥读论文』微信群目前包含:图像处理、大模型及智能体、自动驾驶及机器人、AI医疗及AI金融、推荐系统与广告,期待和你一起碰撞想法!
*本文仅代表个人理解及观点,不构成任何论文审核或者项目落地推荐意见,具体以相关组织评审结果为准。欢迎就论文内容交流探讨,理性发言哦~ 想了解更多原文细节的小伙伴,可以点击"阅读原文",查看更多原论文细节哦!