LinkedIn推荐系统靠混合蒸馏,效果成本双赢!
在线蒸馏灵活却太慢,离线缓存高效却不够灵活,LinkedIn 如何取舍?
InfoQ 中文 重要 论文方法算力 🕐 09-17 17:05

📖 AI 总结

LinkedIn 公布了其 AI 职位搜索背后的多教师蒸馏训练基础设施,核心目标是将大型教师模型的知识压缩至一个 6 亿参数的紧凑排序模型中,以支撑快速迭代。为突破教师模型实时服务带来的性能瓶颈,LinkedIn 基于开源引擎 SGLang 构建了定制框架,同时支持在线与离线两种蒸馏模式:在线模式下通过异步客户端和多节点本地教师副本将蒸馏提速 3 倍并保持低延迟;离线模式则预计算教师输出并存储于 HDFS 或 NFS,消除实时服务开销。配合 LiGer 内存优化、多节点训练、FSDP2 及 H200 集群等一系列工程优化,整体训练速度提升约 8 倍。该 6 亿参数学生模型将职位搜索 NDCG@10 从 0.7583 提升至 0.9432,推理吞吐量从每 GPU 每秒约 290 条提升至 2000 条以上,目前已投入生产,为美国用户的自然语言职位搜索提供支持。

🔍 深度解读 换个视角,同一件事有不同答案

正在读取芙娘的解读…
infoAI 公众号二维码

📱 每天一份 AI 前沿日报

关注公众号,每天 09:00 推送 · 不错过任何重磅