arXiv:2506.03699cs.IR2025-06KDD被引 16

用生成式预训练解决推荐模型过拟合,让大模型更有效。

Scaling Transformers for Discriminative Recommendation via Generative Pretraining

  • 用生成模型参数初始化判别模型,缓解数据稀疏导致的过拟合。
  • 模型参数从13K扩到0.3B时性能持续提升,符合幂律规律。
  • 适合想用大模型做推荐、追求可扩展性的工业级系统开发者。

判别式推荐任务(如点击率、转化率预测)在大规模工业推荐系统排序阶段至关重要。然而,训练判别模型常因数据稀疏导致严重过拟合,且模型越大问题越严重,反而表现不如小模型。为此,我们提出GPSD(Generative Pretraining for Scalable Discriminative Recommendation)框架,借鉴生成式训练无明显过拟合的特性,利用预训练生成模型的参数初始化判别模型,并采用稀疏参数冻结策略。在工业级和公开数据集上的大量实验表明,GPSD性能优越,线上A/B测试中也取得显著提升。其优势在于:1)显著缩小泛化差距,提升测试表现;2)充分利用Transformer可扩展性,模型规模从13K增至0.3B时性能持续增长,紧密遵循幂律。该研究为统一推荐模型与语言模型架构铺路,使大语言模型成熟技术可直接应用于推荐系统。代码已开源:https://github.com/chqiwang/gpsd-rec。

原文摘要 · Abstract (English)

Discriminative recommendation tasks, such as CTR (click-through rate) and CVR (conversion rate) prediction, play critical roles in the ranking stage of large-scale industrial recommender systems. However, training a discriminative model encounters a significant overfitting issue induced by data sparsity. Moreover, this overfitting issue worsens with larger models, causing them to underperform smaller ones. To address the overfitting issue and enhance model scalability, we propose a framework named GPSD (\textbf{G}enerative \textbf{P}retraining for \textbf{S}calable \textbf{D}iscriminative Recommendation), drawing inspiration from generative training, which exhibits no evident signs of overfitting. GPSD leverages the parameters learned from a pretrained generative model to initialize a discriminative model, and subsequently applies a sparse parameter freezing strategy. Extensive experiments conducted on both industrial-scale and publicly available datasets demonstrate the superior performance of GPSD. Moreover, it delivers remarkable improvements in online A/B tests. GPSD offers two primary advantages: 1) it substantially narrows the generalization gap in model training, resulting in better test performance; and 2) it leverages the scalability of Transformers, delivering consistent performance gains as models are scaled up. Specifically, we observe consistent performance improvements as the model dense parameters scale from 13K to 0.3B, closely adhering to power laws. These findings pave the way for unifying the architectures of recommendation models and language models, enabling the direct application of techniques well-established in large language models to recommendation models. The code is available at https://github.com/chqiwang/gpsd-rec.

推荐系统Transformer生成预训练可扩展性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。