arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.04400cs.LGcs.IR
收藏

关于推荐模型扩展时嵌入坍缩现象的研究

On the Embedding Collapse when Scaling up Recommendation Models

Tencent Inc(腾讯公司)

Xingzhuo Guo, Junwei Pan, Ximei Wang, Baixu Chen, Jie Jiang, Mingsheng Long

AI总结:

本文识别出推荐模型扩展中的嵌入坍缩现象,揭示特征交互的双面效应,并提出多嵌入设计以增强嵌入多样性、缓解坍缩并提升模型可扩展性。

摘要:

基础模型的近期进展催生了一个有前景的趋势,即开发大型推荐模型以利用海量可用数据。然而,主流模型的规模仍然小得令人尴尬,简单扩大模型规模并不能带来足够的性能提升,这表明模型的可扩展性存在缺陷。本文中,我们将嵌入坍缩现象识别为可扩展性的抑制因素,即嵌入矩阵倾向于占据一个低维子空间。通过实证和理论分析,我们证明了推荐模型特有的特征交互具有“双面效应”。一方面,与坍缩的嵌入进行交互会限制嵌入学习并加剧坍缩问题。另一方面,交互对于抑制虚假特征的拟合至关重要,从而为可扩展性提供保障。基于我们的分析,我们提出了一种简单而有效的多嵌入设计,该设计结合了嵌入集特定的交互模块,以学习具有较大多样性的嵌入集,从而减少坍缩。大量实验表明,该设计为各种推荐模型提供了一致的可扩展性和有效的坍缩缓解。代码可在以下仓库获取:https://github.com/thuml/Multi-Embedding。

英文摘要:

Recent advances in foundation models have led to a promising trend of developing large recommendation models to leverage vast amounts of available data. Still, mainstream models remain embarrassingly small in size and naïve enlarging does not lead to sufficient performance gain, suggesting a deficiency in the model scalability. In this paper, we identify the embedding collapse phenomenon as the inhibition of scalability, wherein the embedding matrix tends to occupy a low-dimensional subspace. Through empirical and theoretical analysis, we demonstrate a \emph{two-sided effect} of feature interaction specific to recommendation models. On the one hand, interacting with collapsed embeddings restricts embedding learning and exacerbates the collapse issue. On the other hand, interaction is crucial in mitigating the fitting of spurious features as a scalability guarantee. Based on our analysis, we propose a simple yet effective multi-embedding design incorporating embedding-set-specific interaction modules to learn embedding sets with large diversity and thus reduce collapse. Extensive experiments demonstrate that this proposed design provides consistent scalability and effective collapse mitigation for various recommendation models. Code is available at this repository: https://github.com/thuml/Multi-Embedding.

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

↑