arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.05625cs.IRstat.ML
收藏

CAN:用于点击率预测的特征协同作用

CAN: Feature Co-Action for Click-Through Rate Prediction

Weijie Bian, Kailun Wu, Lejian Ren, Qi Pi, Yujing Zhang, Can Xiao, Xiang-Rong Sheng, Yong-Nan Zhu, Zhangming Chan, Na Mou, Xinchen Luo, Shiming Xiang, Guorui Zh… 展开作者

Weijie Bian, Kailun Wu, Lejian Ren, Qi Pi, Yujing Zhang, Can Xiao, Xiang-Rong Sheng, Yong-Nan Zhu, Zhangming Chan, Na Mou, Xinchen Luo, Shiming Xiang, Guorui Zhou, Xiaoqiang Zhu, Hongbo Deng

摘要:

特征交互是机器学习中的一个重要问题,也是点击率(CTR)预测的关键。近年来,深度神经网络(DNN)能够从原始稀疏特征中自动学习隐含的非线性相互作用,因此在工业CTR预测任务中得到了广泛应用。然而,在DNN中学习的隐式特征交互不能完全保留原始和经验特征交互的完整表示能力(例如,产品)而没有损失。例如,学习特征A和特征B <A,B>的组合作为新特征的显式卡耐基乘积表示的简单尝试可以优于先前的隐式特征交互模型,包括基于因子分解机(FM)的模型及其变型。在本文中,我们提出了一个共同行动网络(CAN)来近似显式的两两特征相互作用,而不引入太多的额外参数。更具体地说,给定特征A及其相关联的特征B,它们的特征交互通过学习两组参数来建模:1)特征A的嵌入,以及2)用于表示特征B的多层感知器(MLP)。通过将特征A的嵌入通过特征B的MLP网络,可以获得近似的特征交互。我们将这种成对特征交互称为特征协同作用,并且这种协同作用网络单元可以提供非常强大的能力来拟合复杂的特征交互。在公共数据集和工业数据集上的实验结果表明,CAN的性能优于最先进的CTR模型和Carnival乘积方法。此外,CAN已经在阿里巴巴的展示广告系统中部署,CTR提高了12%,RPM提高了8%,这对业务来说是一个很大的提升。

英文摘要:

Feature interaction has been recognized as an important problem in machine learning, which is also very essential for click-through rate (CTR) prediction tasks. In recent years, Deep Neural Networks (DNNs) can automatically learn implicit nonlinear interactions from original sparse features, and therefore have been widely used in industrial CTR prediction tasks. However, the implicit feature interactions learned in DNNs cannot fully retain the complete representation capacity of the original and empirical feature interactions (e.g., cartesian product) without loss. For example, a simple attempt to learn the combination of feature A and feature B <A, B> as the explicit cartesian product representation of new features can outperform previous implicit feature interaction models including factorization machine (FM)-based models and their variations. In this paper, we propose a Co-Action Network (CAN) to approximate the explicit pairwise feature interactions without introducing too many additional parameters. More specifically, giving feature A and its associated feature B, their feature interaction is modeled by learning two sets of parameters: 1) the embedding of feature A, and 2) a Multi-Layer Perceptron (MLP) to represent feature B. The approximated feature interaction can be obtained by passing the embedding of feature A through the MLP network of feature B. We refer to such pairwise feature interaction as feature co-action, and such a Co-Action Network unit can provide a very powerful capacity to fitting complex feature interactions. Experimental results on public and industrial datasets show that CAN outperforms state-of-the-art CTR models and the cartesian product method. Moreover, CAN has been deployed in the display advertisement system in Alibaba, obtaining 12\% improvement on CTR and 8\% on Revenue Per Mille (RPM), which is a great improvement to the business.

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

↑