arrow
返回

Bayesian feature interaction selection for factorization machines

delete2022-01-01
delete23
PRE
AI
Y
Yifan Chen
阳旺 (Yang Wang) *
P
Pengjie Ren
王萌 (Meng Wang)
M
Maarten de Rijke
DOI:10.1016/j.artint.2021.103589delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Factorization machines are a generic supervised method for a wide range of tasks in the field of artificial intelligence, such as prediction, inference, etc., which can effectively model feature interactions. However, handling combinations of features is expensive due to the exponential growth of feature interactions with the order. In nature, not all feature interactions are equally useful for prediction. Recently, a large number of methods that perform feature interaction selection have attracted great attention because of their effectiveness at filtering out useless feature interactions. Current feature interaction selection methods suffered from the following limitations: (1) they assume that all users share the same feature interactions; and (2) they select pairwise feature interactions only. In this paper, we propose novel Bayesian variable selection methods, targeting feature interaction selection for factorization machines, which effectively reduce the number of interactions. We study personalized feature interaction selection to account for individual preferences, and further extend the model to investigate higher-order feature interaction selection on higher-order factorization machines. We provide empirical evidence for the advantages of the proposed Bayesian feature interaction selection methods using different prediction tasks. (C) 2021 Elsevier B.V. All rights reserved.
Keyword:
Factorization machine
Bayesian variable selection
Feature interaction selection

期刊

Artificial Intelligence Review 封面图
Artificial Intelligence Review
IF:
13.9
论文数:
6.1K
被引数:
1.9W

机构

H
hefei university of technology
学者数:
2.5W
论文数: 1.7W
被引数: 35
U
university of amsterdam
学者数:
6.0W
论文数: 5.1W
被引数: 94
N
national university of defense technology - china
学者数:
1.8W
论文数: 1.4W
被引数: 9
学者 查看更多机构
引用论文

引用论文

Product-Based Neural Networks for User Response Prediction over Multi-Field Categorical Data
err2018-10-30
err132
errOAAI
errQu, Yanru; Fang, Bohui; Zhang, Weinan; Tang, Ruiming; Niu, Minzhe; Guo, Huifeng; Yu, Yong; He, Xiuqiang
err分享
err收藏
err分享
err收藏
Efficient Mining of Frequent Patterns on Uncertain Graphs
err2019-02-01
err36
PREAI
errChen, Yifan; Zhao, Xiang; Lin, Xuemin; Wang, Yang; Guo, Deke
err分享
err收藏
AutoHash: Learning Higher-Order Feature Interactions for Deep CTR Prediction
err2020-01-01
err10
PREAI
errXue, Niannan; Liu, Bin; Guo, Huifeng; Tang, Ruiming; Zhou, Fengwei; Zafeiriou, Stefanos; Zhang, Yuzhou; Wang, Jun; Li, Zhenguo
err分享
err收藏
学者 查看更多内容