解决广告模型多轮训练过拟合问题,提升高基数特征学习效率。
The Evolution of Embedding Table Optimization and Multi-Epoch Training in Pinterest Ads Conversion
- 针对高基数类别特征设计频率自适应学习率优化嵌入表。
- 实验表明该方法比重置嵌入表更有效,尤其在标签稀疏任务中。
- 适用于大规模在线广告多目标预测场景,提升模型稳定性与性能。
深度学习在转化率预测中广泛应用,模型需联合预测点击、加购、下单等多种转化行为。为处理高基数类别特征(如广告主、用户、广告系列、产品ID),常使用嵌入表进行编码。然而,由于梯度稀疏、特征分布不均及标签稀疏等问题,训练过程易出现收敛慢和首轮后过拟合。以往方法通过强化特征哈希、过滤低频ID、正则化或每轮重初始化嵌入表缓解过拟合,但可能损害性能。本文分享了Pinterest广告转化模型中嵌入表优化与多轮训练的关键经验:提出基于频率自适应学习率的嵌入表优化策略,并在工业级大规模数据集上离线验证其有效性,结果表明该方法优于传统重初始化,在不同任务间对过拟合的抑制效果更具针对性。
原文摘要 · Abstract (English)
Deep learning for conversion prediction has found widespread applications in online advertising. These models have become more complex as they are trained to jointly predict multiple objectives such as click, add-to-cart, checkout and other conversion types. Additionally, the capacity and performance of these models can often be increased with the use of embedding tables that encode high cardinality categorical features such as advertiser, user, campaign, and product identifiers (IDs). These embedding tables can be pre-trained, but also learned end-to-end jointly with the model to directly optimize the model objectives. Training these large tables is challenging due to: gradient sparsity, the high cardinality of the categorical features, the non-uniform distribution of IDs and the very high label sparsity. These issues make training prone to both slow convergence and overfitting after the first epoch. Previous works addressed the multi-epoch overfitting issue by using: stronger feature hashing to reduce cardinality, filtering of low frequency IDs, regularization of the embedding tables, re-initialization of the embedding tables after each epoch, etc. Some of these techniques reduce overfitting at the expense of reduced model performance if used too aggressively. In this paper, we share key learnings from the development of embedding table optimization and multi-epoch training in Pinterest Ads Conversion models. We showcase how our Sparse Optimizer speeds up convergence, and how multi-epoch overfitting varies in severity between different objectives in a multi-task model depending on label sparsity. We propose a new approach to deal with multi-epoch overfitting: the use of a frequency-adaptive learning rate on the embedding tables and compare it to embedding re-initialization. We evaluate both methods offline using an industrial large-scale production dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。