.png){kind=icon}
Return
Bayesian feature interaction selection for factorization machines
.png){kind=icon}
PRE
AI
DOI：10.1016/j.artint.2021.103589.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
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.
Keywords:
Factorization machine
Bayesian variable selection
Feature interaction selection
Journal
IF:
13.9
Papers:
6.1K
Citations:
1.9W
Organization
View more organizations
Cited Papers
ShareSave
ShareSaveShareSave
Comparing the Areas under Two or More Correlated Receiver Operating Characteristic Curves: A Nonparametric Approach.png)
Biometrics
BiometricsIF0
PREAI
ShareSaveShareSave View more