arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2105.07752cs.AIcs.IRcs.LG
收藏

基于预训练图神经网络的显式语义交叉特征学习用于CTR预测

Explicit Semantic Cross Feature Learning via Pre-trained Graph Neural Networks for CTR Prediction

Alibaba Group(阿里巴巴集团)

Feng Li, Bencheng Yan, Qingqing Long, Pengjie Wang, Wei Lin, Jian Xu, Bo Zheng

摘要:

交叉特征在点击率(CTR)预测中起着重要作用。现有的方法大多采用基于DNN的模型以隐式方式捕获交叉特征。由于显式语义建模的限制,这些隐式方法可能导致次优化的性能。虽然传统的统计显式语义交叉特征可以解决这些隐式方法中的问题,但它仍然受到一些挑战,包括缺乏泛化能力和昂贵的内存成本。很少有人关注这些挑战。在本文中,我们在学习显式语义交叉特征方面迈出了第一步,并提出了预训练交叉特征学习图神经网络(PCF-GNN),这是一种基于GNN的预训练模型,旨在以显式方式生成交叉特征。在公共和工业数据集上进行了广泛的实验,其中PCF-GNN在各种任务中表现出性能和内存效率的能力。

英文摘要:

Cross features play an important role in click-through rate (CTR) prediction. Most of the existing methods adopt a DNN-based model to capture the cross features in an implicit manner. These implicit methods may lead to a sub-optimized performance due to the limitation in explicit semantic modeling. Although traditional statistical explicit semantic cross features can address the problem in these implicit methods, it still suffers from some challenges, including lack of generalization and expensive memory cost. Few works focus on tackling these challenges. In this paper, we take the first step in learning the explicit semantic cross features and propose Pre-trained Cross Feature learning Graph Neural Networks (PCF-GNN), a GNN based pre-trained model aiming at generating cross features in an explicit fashion. Extensive experiments are conducted on both public and industrial datasets, where PCF-GNN shows competence in both performance and memory-efficiency in various tasks.

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

↑