1. 问题与动机
CTR 预估模型面临一个独特的训练困境:单轮过拟合现象(one-epoch overfitting)。与 CV/NLP 领域模型通常需要多轮训练才能收敛不同,CTR 模型在第二轮训练开始时性能即显著下降。
根本原因在于 embedding 层过拟合:CTR 模型处理数十亿稀疏分类特征(如用户 ID、物品 ID),大量低频特征的 embedding 向量仅有极少训练样本,重复暴露导致 embedding 记忆训练数据而非学习泛化表征。作者通过控制实验验证:固定 embedding 层时 MLP 几乎不过拟合;固定 MLP 后 embedding 过拟合仍然出现。
工业意义:多轮训练对于以下场景至关重要——(1) 数据量有限但模型容量大;(2) 需要无监督域适应或标签噪声修正;(3) 计算资源充足但训练时间有限。快手实际部署中,MEDA 使基线 2 周训练效果仅需 1 周达到。
2. 方法详解
核心思想:解耦 Embedding 与 MLP
MEDA 发现 MLP 层学习的是 embedding 间的相对关系(方向)而非绝对位置(数值)。因此可通过在每轮训练时重新初始化 embedding 来实现隐式数据增强——MLP 被迫在不同的 embedding 空间中学习稳健的匹配函数。
Non-Continual MEDA (Algorithm 1)
- 随机初始化 MLP 参数 θ₀
- 对每轮 r (1 到 k):
- 随机重新初始化 embedding 层 Eᵣ(Glorot/均匀分布)
- 使用上轮 MLP θ_{r-1} 作为初始化(保留 MLP 知识)
- 在完整数据集上训练一轮
- 同时更新 θ 和 E
关键洞察:embedding 重新初始化 = 隐式数据增强。每轮的 embedding 空间不同,MLP 看到的是"同一数据的不同表示",等效于在特征空间做增强。
Continual MEDA (Algorithm 2)
适用于连续数据流(如日增量训练):维护 k 个独立初始化的 embedding 组:
- 初始化共享 MLP θ^c 和 k 个 embedding 层
- 对每个数据集 t (日增量):对每轮 r 选择 embedding Eᵣ^{t-1},初始化 MLP 为 θ^c,训练一轮
训练细节
- 损失函数:BCELoss(分类)/ AUC scoring(回归/LTV)
- 优化器:Adam (lr=0.001) 公开数据集;Adagrad (lr=0.01) 业务数据集
- Embedding 初始化:Glorot(公开);uniform [-0.01, 0.01](业务)
- 验证 MLP 收敛性:跨轮 MLP 参数的 cosine similarity 单调递增
3. 核心技术贡献
- 首次系统诊断 CTR 模型单轮过拟合根因:通过控制实验明确归因于 embedding 层的高维稀疏性过拟合。推翻了此前模糊的"整体模型过拟合"认知。
- 通用多轮训练策略:MEDA 可无缝集成到 DNN、DIN、DIEN、MIMN、ADFM 等不同架构中,无需修改模型结构。2 轮训练即带来 +0.73% 到 +5.17% AUC 提升。
- 数据效率翻倍:在 Taobao 数据上,50% 数据 + 2 轮训练即达到基线 100% 数据单轮的性能,节省一半数据采集成本。
- MLP 学习相对关系的发现:余弦相似度分析证明 MLP 参数方向跨轮收敛,为 embedding 重新初始化策略提供理论基础。
4. 实验结果分析
主实验 AUC 提升(2轮 vs 单轮)
公开数据集:
| 模型 | Amazon | Taobao |
|---|---|---|
| DNN | +0.73% | +1.82% |
| DIN | +1.30% | +1.06% |
| DIEN | +1.75% | +1.98% |
| MIMN | +1.41% | +1.61% |
| ADFM | +0.90% | +5.17% |
业务数据集:
| 模型 | SVO (快手短视频, 17.5亿样本) | SVSL (快手社交, 3亿样本) |
|---|---|---|
| DNN | +0.24% | +0.49% |
| DIN | +0.33% | +0.64% |
Taobao 提升显著高于 Amazon,因 Taobao 数据稀疏度更低。
消融实验
- Fix embedding(不重新初始化):无增强效果
- Fix MLP after epoch 1:embedding 过拟合仍发生
- Embedding same init:减轻但不消除过拟合
- Embedding reinit(MEDA):成功消除过拟合
数据效率
Taobao:MEDA 50% 数据 + 2 轮 = 基线 100% 数据单轮
ADFM with 3 epochs on 1/8 data > 基线全数据单轮
在线 A/B 测试(快手视频广告,10% 流量 9 天)
- Test AUC: +0.14%
- 用户留存: +6.6%
- 整体收入: +0.32%;预期收入: +0.91%
关键成果:MEDA 使 2 周训练效果仅需 1 周达到,训练效率翻倍。
5. 局限与未来方向
作者承认:
- 信息损失:随机重新初始化丢弃了高频特征积累的有价值 embedding。建议未来借鉴差分隐私思想,根据特征频率设计差异化噪声水平。
- 在线学习时效性:多轮方法牺牲时效性,但 batch+online 混合方案可容纳。
未被作者提及的局限:(1) 十亿级特征表重新初始化的显存和 IO 开销;(2) 高频特征的冷启动效应。
6. 工业实践意义
直接可采纳:
- 零代码改动提升 CTR 模型效果:仅需在训练脚本中加入 embedding 重新初始化逻辑
- 在数据量有限的垂类场景(如 UGC 社区搜索排序),多轮训练增益更显著
- 数据效率翻倍意味着可用更短数据窗口训练出等效模型
部署考量:
- 训练时间增加 k 倍,但可并行化
- 在线 serving 无额外开销:最终只保留一套 embedding + MLP
- 与增量训练兼容:Continual MEDA 专为日更场景设计
对 UGC 搜索排序的适用性:社区搜索的 CTR 模型通常面临样本稀疏(长尾 query/帖子组合多)、特征高维稀疏特点,正是 MEDA 增益最大的场景。建议在现有 CTR 模型上尝试 2-3 轮 MEDA 训练,预期 AUC +0.3-1.0%。