Detecting Beneficial Feature Interactions for Recommender Systems

Authors

  • Yixin Su University of Melbourne
  • Rui Zhang Tsinghua University
  • Sarah Erfani University of Melbourne
  • Zhenghua Xu Hebei University of Technology

DOI:

https://doi.org/10.1609/aaai.v35i5.16561

Keywords:

Recommender Systems & Collaborative Filtering

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.

Downloads

Published

2021-05-18

How to Cite

Su, Y., Zhang, R., Erfani, S., & Xu, Z. (2021). Detecting Beneficial Feature Interactions for Recommender Systems. Proceedings of the AAAI Conference on Artificial Intelligence, 35(5), 4357-4365. https://doi.org/10.1609/aaai.v35i5.16561

Issue

Section

AAAI Technical Track on Data Mining and Knowledge Management