谷歌优化广告推荐训练系统,提升效率并降本18%。
Scalable Machine Learning Training Infrastructure for Online Ads Recommendation and Auction Scoring Modeling at Google
- 共享输入生成降低多模型计算开销
- 嵌入表分片与请求合并提升处理效率
- 预占通知机制减少资源浪费,适合大规模部署
谷歌大规模广告推荐与竞价评分模型训练需海量算力。尽管专用硬件如TPU提升了线性代数运算效率,但系统仍存在三大瓶颈:(1)原始特征(如“搜索词”)到数值输入的高效转换与流式传输至TPU;(2)稀疏特征转为神经网络可用的稠密浮点向量的嵌入表优化;(3)在共享数据中心中应对中断与错误,避免资源浪费。为此,提出共享输入生成技术,通过跨模型摊销成本降低计算负载;设计嵌入操作的分片、流水线与远程过程调用(RPC)合并策略;引入新型预占通知与训练保持机制,实现资源高效利用与故障快速响应。这些技术在谷歌生产环境中显著提升性能,代表性模型实现116%性能提升和18%训练成本下降。
原文摘要 · Abstract (English)
Large-scale Ads recommendation and auction scoring models at Google scale demand immense computational resources. While specialized hardware like TPUs have improved linear algebra computations, bottlenecks persist in large-scale systems. This paper proposes solutions for three critical challenges that must be addressed for efficient end-to-end execution in a widely used production infrastructure: (1) Input Generation and Ingestion Pipeline: Efficiently transforming raw features (e.g., "search query") into numerical inputs and streaming them to TPUs; (2) Large Embedding Tables: Optimizing conversion of sparse features into dense floating-point vectors for neural network consumption; (3) Interruptions and Error Handling: Minimizing resource wastage in large-scale shared datacenters. To tackle these challenges, we propose a shared input generation technique to reduce computational load of input generation by amortizing costs across many models. Furthermore, we propose partitioning, pipelining, and RPC (Remote Procedure Call) coalescing software techniques to optimize embedding operations. To maintain efficiency at scale, we describe novel preemption notice and training hold mechanisms that minimize resource wastage, and ensure prompt error resolution. These techniques have demonstrated significant improvement in Google production, achieving a 116% performance boost and an 18% reduction in training costs across representative models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。