阅读研究雷达投资体系
登录 / 注册
← 返回阅读
资讯专题 · 英伟达序列推荐加速1 分钟阅读·2026-10-15

NVIDIA工程师在PyTorch频道解析序列推荐模型加速:SASRec与HSTU的GPU优化实践

NVIDIA工程师Zan Huang和Jiayu Sun在PyTorch官方频道发布视频,介绍如何在PyTorch生态系统中加速序列推荐模型。演讲涵盖从TensorFlow到PyTorch的SASRec重写、HSTU注意力算子的结构差异及GPU内核融合等关键优化策略。

重要性局部证据E2 未复现写法快讯

2026-09-25 04:34

NVIDIA工程师Zan Huang和Jiayu Sun于2026年9月24日在PyTorch官方YouTube频道发布了题为“Accelerating Sequential Recommender Models in the PyTorch Ecosystem”的技术演讲。该视频聚焦于序列推荐模型与大语言模型(LLM)在自注意力机制上的联系,以及生产环境推荐系统所面临的独特挑战。

根据发布者提供的摘要,演讲首先回顾了早期基于Transformer解码器的序列推荐器SASRec,展示了其从TensorFlow到PyTorch的重写过程,并强调了PyTorch如何降低建模复杂性以及学术原型与生产系统之间的差距。随后,演讲深入探讨了HSTU这一大规模序列推荐注意力算子,分析了其标准Transformer注意力的结构性差异,并分享了关于内核融合、变长序列处理以及通过FBGEMM集成到PyTorch的关键GPU洞察。最后,演讲者从建模和系统角度反思了序列推荐与LLM的关系,分享了跨推荐工作负载的新兴趋势和优化策略观察。

来源

  • 原始出处youtube.com↗
专题

英伟达序列推荐加速

英伟达正通过全栈优化工具链,推动基于HSTU(Hierarchical Sequential Transduction Unit)架构的生成式推荐系统落地。该方案旨在解决传统深度学习推荐模型在长序列用户行为建模中的计算瓶颈,通过PyTorc…

5 节点1 篇更新于 2026-10-15 12:36

事情怎么走到这一步

  1. 2024-07-01Meta发布ICML论文《Actions Speak Louder than Words》,提出HSTU架构,证明其能超越传统Transformer并加速训练推理10x-1000x。
登录 / 注册
阅读研究雷达投资体系
  • 2025-06-19NVIDIA发布recsys-examples博客,介绍基于CUTLASS的HSTU高效算子及与TorchRec集成的动态Embedding方案,报告美团场景下训练吞吐提升85%。
  • 2026-09-08NVIDIA推出GR-Inference推理系统,针对SID生成式推荐的长Context、短Decode特点优化KV缓存与Beam Search,在小红书落地实现有效点击率提升0.2%。
  • 还没定论的部分

    • 生成式推荐在超大规模稀疏特征处理上,相比传统DLRM的成本效益拐点在哪里?
    • GR-Inference的Beam Width动态扩展机制在不同业务流量峰值下的稳定性表现如何?
    • 除美团、小红书外,其他主流平台是否已公开验证HSTU架构的Scaling Law有效性?
    读完整专题 →
    带入研究 →