arXivDaily arXiv每日学术速递 周一至周五更新
稍安勿躁,官方尚未更新
arXiv 2302.09178cs.LG
收藏

提升推荐系统中多任务排序模型的训练稳定性

Improving Training Stability for Multitask Ranking Models in Recommender Systems

Google Deepmind(谷歌DeepMind);Google Research(谷歌研究院);Google Inc(谷歌公司)

Jiaxi Tang, Yoel Drori, Daryl Chang, Maheswaran Sathiamoorthy, Justin Gilmer, Li Wei, Xinyang Yi, Lichan Hong, Ed H. Chi

AI总结:

针对推荐系统大模型训练易损失发散的稳定性问题,本文分析YouTube多任务排序模型不稳定的特性与成因,提出新算法缓解现有方案缺陷,实验证实其在生产数据集上可稳训练且不影响收敛。

摘要:

推荐系统在众多内容平台中发挥着重要作用。尽管大多数推荐研究致力于设计更优模型以改善用户体验,但我们发现,针对此类模型训练稳定性的研究仍严重不足。随着推荐模型规模愈发庞大、结构愈发复杂,它们更易出现训练不稳定问题,即损失发散,这会导致模型无法使用、浪费大量资源并阻碍模型开发。本文中,我们分享了在提升YouTube推荐所用真实多任务排序模型的训练稳定性方面的发现与最佳实践。我们展示了该模型导致训练不稳定的若干特性,并对成因进行了推测。此外,基于对训练不稳定点附近训练动态的观察,我们假设了现有解决方案失效的原因,并提出一种新算法以缓解现有方案的局限性。我们在YouTube生产数据集上开展的实验表明,与几种常用基线方法相比,所提算法能显著提升训练稳定性,同时不损害收敛性能。

英文摘要:

Recommender systems play an important role in many content platforms. While most recommendation research is dedicated to designing better models to improve user experience, we found that research on stabilizing the training for such models is severely under-explored. As recommendation models become larger and more sophisticated, they are more susceptible to training instability issues, i.e., loss divergence, which can make the model unusable, waste significant resources and block model developments. In this paper, we share our findings and best practices we learned for improving the training stability of a real-world multitask ranking model for YouTube recommendations. We show some properties of the model that lead to unstable training and conjecture on the causes. Furthermore, based on our observations of training dynamics near the point of training instability, we hypothesize why existing solutions would fail, and propose a new algorithm to mitigate the limitations of existing solutions. Our experiments on YouTube production dataset show the proposed algorithm can significantly improve training stability while not compromising convergence, comparing with several commonly used baseline methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑