Detecting Beneficial Feature Interactions for Recommender Systems

Yixin Su, Rui Zhang, Sarah Erfani, Zhenghua Xu
8/2/2020
cs.LGcs.IRstat.ML

Official code: ruizhang-ai/SIGN-Detecting-Beneficial-Feature-Interactions-for-Recommender-Systems (34 stars, MIT, last commit 2021-08-13).

Abstract

Feature interactions are essential for achieving high accuracy in recommender systems. Many studies take into account the interaction between every pair of features. However, this is suboptimal because some feature interactions may not be that relevant to the recommendation result, and taking them into account may introduce noise and decrease recommendation accuracy. To make the best out of feature interactions, we propose a graph neural network approach to effectively model them, together with a novel technique to automatically detect those feature interactions that are beneficial in terms of recommendation accuracy. The automatic feature interaction detection is achieved via edge prediction with an L0 activation regularization. Our proposed model is proved to be effective through the information bottleneck principle and statistical interaction theory. Experimental results show that our model (i) outperforms existing baselines in terms of accuracy, and (ii) automatically identifies beneficial feature interactions.

View on arXivView PDF

Code Implementations(1)

MIT

Detecting Beneficial Feature Interactions for Recommender Systems, AAAI 2021

346PythonMay 18, 20215 years agoMIT
aaai2021feature-interactionsgraph-neural-networksinformation-bottleneck-theoryl0-regularization+1 more

Cite this paper

@article{su2020detecting,
  title  = {Detecting Beneficial Feature Interactions for Recommender Systems},
  author = {Yixin Su and Rui Zhang and Sarah Erfani and Zhenghua Xu},
  year   = {2020},
  eprint = {2008.00404},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2008.00404}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2008.00404.svg)](https://codeofpaper.com/paper/2008.00404)

Discussion