计算机科学
特征(语言学)
嵌入
领域(数学)
图形
特征学习
人工智能
人工神经网络
推荐系统
理论计算机科学
机器学习
数学
哲学
语言学
纯数学
作者
Zekun Li,Zeyu Cui,Shu Wu,Xiaoyu Zhang,Liang Wang
出处
期刊:Cornell University - arXiv
日期:2019-11-03
被引量:53
标识
DOI:10.1145/3357384.3357951
摘要
Click-through rate (CTR) prediction is an essential task in web applications such as online advertising and recommender systems, whose features are usually in multi-field form. The key of this task is to model feature interactions among different feature fields. Recently proposed deep learning based models follow a general paradigm: raw sparse input multi-field features are first mapped into dense field embedding vectors, and then simply concatenated together to feed into deep neural networks (DNN) or other specifically designed networks to learn high-order feature interactions. However, the simple unstructured combination of feature fields will inevitably limit the capability to model sophisticated interactions among different fields in a sufficiently flexible and explicit fashion. In this work, we propose to represent the multi-field features in a graph structure intuitively, where each node corresponds to a feature field and different fields can interact through edges. The task of modeling feature interactions can be thus converted to modeling node interactions on the corresponding graph. To this end, we design a novel model Feature Interaction Graph Neural Networks (Fi-GNN). Taking advantage of the strong representative power of graphs, our proposed model can not only model sophisticated feature interactions in a flexible and explicit fashion, but also provide good model explanations for CTR prediction. Experimental results on two real-world datasets show its superiority over the state-of-the-arts.
科研通智能强力驱动
Strongly Powered by AbleSci AI