Deep Interaction Machine: A Simple but Effective Model for High-order Feature Interactions

被引:16
作者
Yu, Feng [1 ,4 ,5 ]
Liu, Zhaocheng [2 ]
Liu, Qiang [2 ,3 ]
Zhang, Haoli [2 ]
Wu, Shu [4 ,5 ]
Wang, Liang [4 ,5 ]
机构
[1] Alibaba Grp, Hangzhou, Peoples R China
[2] RealAI, Beijing, Peoples R China
[3] Tsinghua Univ, Beijing, Peoples R China
[4] Chinese Acad Sci, Ctr Res Intelligent Percept & Comp, Inst Automat, Beijing, Peoples R China
[5] Univ Chinese Acad Sci, Sch Artificial Intelligence, Beijing, Peoples R China
来源
CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT | 2020年
基金
中国国家自然科学基金;
关键词
D O I
10.1145/3340531.3412077
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Click-Through Rate (CTR) prediction is a crucial task for various online applications, such as recommendation and online advertising. The task of CTR prediction is to predict the probability of users' clicking behaviors, with high-dimensional input features. To avoid heavy handcrafted feature engineering, the core topic of CTR prediction is the automatic interactions of the input features. Factorization Machine (FM) is an effective approach for modeling second-order feature interactions. Recently, FM has been extended for modeling higher-order feature interactions, such as xDeepFM and Higher-Order Factorization Machine (HOFM). However, these approaches are with either high complexity or iterative computation consuming much time and space. To overcome above problems, we express arbitrary-order FM in the form of power sums according to Newton's identities. Accordingly, we propose a novel Interaction Machine (IM) model. IM is an efficient and exact implementation of high-order FM, whose time complexity linearly grows with the order of interactions and the number of feature fields. Via IM, we can conduct arbitrary-order feature interactions in a very simple way. Moreover, we perform IM together with deep neural networks, and the resulted DeepIM model is more efficient than xDeepFM with comparable or even better performance. We conduct experiments on two real-world datasets, in which effectiveness and efficiency of both IM and DeepIM are strongly verified.
引用
收藏
页码:2285 / 2288
页数:4
相关论文
共 10 条
[1]  
Blondel M, 2016, ADV NEUR IN, V29
[2]  
Cheng WY, 2020, AAAI CONF ARTIF INTE, V34, P3609
[3]  
Guo HF, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1725
[4]   Fi-GNN: Modeling Feature Interactions via Graph Neural Networks for CTR Prediction [J].
Li, Zekun ;
Cui, Zeyu ;
Wu, Shu ;
Zhang, Xiaoyu ;
Wang, Liang .
PROCEEDINGS OF THE 28TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT (CIKM '19), 2019, :539-548
[5]   xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systems [J].
Lian, Jianxun ;
Zhou, Xiaohuan ;
Zhang, Fuzheng ;
Chen, Zhongxia ;
Xie, Xing ;
Sun, Guangzhong .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :1754-1763
[6]  
Liu Q., 2015, P 24 ACM INT C INFOR, P1743
[7]  
Mead DG, 1992, AM MATH MONTHLY
[8]  
Rendle Steffen, 2010, Proceedings 2010 10th IEEE International Conference on Data Mining (ICDM 2010), P995, DOI 10.1109/ICDM.2010.127
[9]  
Wu S, 2019, AAAI CONF ARTIF INTE, P346
[10]   TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation [J].
Yu, Feng ;
Zhu, Yanqiao ;
Liu, Qiang ;
Wu, Shu ;
Wang, Liang ;
Tan, Tieniu .
PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), 2020, :1921-1924