张大妈

Meta:一个技巧让长序列推荐推理快4倍

源自小红薯:每日ComputerScience

01-20 17:54

长序列用户行为对推荐效果至关重要,但Transformer模型随序列增长计算成本剧增。Meta推荐系统团队提出“个性化专家”机制,通过压缩历史行为,在几乎不影响效果的前提下,将推理成本降至1/4,为工业级应用提供了实用解决方案。

Meta:一个技巧让长序列推荐推理快4倍智能速览

  • 将长历史压缩为少量可学习token,显著降低计算量。

  • “压缩历史+最近行为”组合性能接近甚至优于完整长序列。

  • Transformer推理成本可降至原来的四分之一。

  • 该方法可直接套用到现有Transformer推荐模型,无需改动架构。

  • 压缩自早期历史的token在数百步行为后依然有效,长期兴趣稳定。

Meta:一个技巧让长序列推荐推理快4倍精华内容

这项名为“个性化专家”的机制,其核心思想并不复杂,但巧妙地平衡了效果与成本。下面将深入拆解其实现原理与关键优势。

核心压缩原理

该方法的核心是将用户的历史行为序列划分为多个segment。对于较早的历史行为,不再直接保留原始序列,而是通过少量可学习的token进行压缩,形成代表用户长期兴趣的“个性化专家”。在推理时,模型只需要处理这些压缩后的专家token和最近一段原始行为,从而大幅减少了计算量。

效果不降反升

实验数据表明,这种“压缩历史+最近行为”的组合方式在HSTU、HLLM等多个业界先进的模型上,其推荐效果几乎与使用完整1280或2000长序列的模型等价,甚至在部分场景下略有提升。这说明压缩过程并未丢失关键信息,反而可能起到了去噪和提炼核心兴趣的作用。

推理成本骤降

推理加速的实现依赖于segment级别的推理与KV cache复用技术。模型无需对整段历史进行全量attention计算,只需关注压缩后的专家token和近期行为。这使得Transformer的推理计算复杂度得到有效控制,最终实现了推理成本降至原先1/4的显著提升,直接解决了工业应用的核心瓶颈。

工程实践优势

该机制具备优秀的工程兼容性,不依赖于全新的模型架构,仅通过调整attention mask和插入可学习token,即可轻松集成到现有的Transformer推荐模型中。此外,实验还发现,压缩自早期历史的个性化专家token具有很好的稳定性,在用户产生数百个新行为后依然能保持有效性,不会快速失效。

Meta的“个性化专家”机制为长序列推荐的落地提供了极具价值的思路。它证明了通过智能压缩历史信息,完全可以在保证效果的同时,大幅优化工程成本。这一方案是否会成为未来推荐系统的标配,值得期待。

内容由AI生成
0
扫一下,分享更方便,购买更轻松
0评论

当前文章无评论,是时候发表评论了
提示信息

取消
确认
评论举报

最新文章 热门文章