跳转至

Zero-shot Cross-domain KD:YouTube 到 YouTube Music

Fidelity: 核心机制复现。源域大 teacher、目标域小 student、目标样本上的 zero-shot logits 和 non-serving auxiliary-task distillation 均执行;私有跨产品 schema 未复刻。

论文信息

项目 内容
论文链接 arXiv 2603.28994
公司/机构 Google / YouTube
首次公开日期 2026-03-30(arXiv v1)
原文开源代码 否:论文未提供官方/作者代码(核查日期:2026-08-09)
Adapter cross-domain-kd
本地复现代码 src/auto_research/reproductions/cross_domain_kd/

原始论文总结

背景与主要改动

低流量产品无力训练大型专用 teacher。论文直接把数据丰富的 YouTube 多任务 teacher 应用于 YouTube Music 样本,把 serving task 与辅助 task logits 蒸馏到小模型,不要求源域数据或匹配的界面特征。

flowchart LR
 A["YouTube source data"] --> B["large multi-task teacher"]
 C["Music target examples"] --> B
 B --> D["serving + auxiliary logits"]
 C --> E["small Music student"]
 D --> E

原论文关键图

Zero-shot Cross-domain KD:YouTube 到 YouTube Music 原论文 Figure 1

原论文 Figure 1(关键图):展示原论文方法的总体设计和关键组成。图片来自原论文,版权归原作者所有;点击图片可查看来源。

核心公式

\(L=L_{target}+\lambda T^2KL(softmax(z_T/T)\Vert softmax(z_S/T))+\gamma\|a_T-a_S\|^2\)

论文离线与线上效果

CTR AUC 79.34→79.55,trail engagement 0.312→0.320;线上 discovery +1.12%、new releases engagement +11.39%

本地复现

本地对照口径:跨模型基线是在相同 target split 上训练的 DIN,实验组 KD student 的 NDCG@10 相对 DIN -68.46%;内部基线 target-only student 加蒸馏后 +3.02%。后者只说明 KD 对弱 student 的增量,不能表述成优于 DIN。 180 source users、60 target users,teacher 19,091 参数、student 7,171 参数。DIN 与 KD 在完全相同 target split 上评估:DIN Hit/NDCG 0.1056/0.05518,KD student 0.0444/0.01741,NDCG 相对 DIN -68.46%。KD 相对 target-only student 仍为 +3.02%,但远弱于 DIN 且 DIN head share 更高。指标见 metrics/movielens-100k-seeds42-44.json