• Home
  • Search
  • Causal factorization machine for robust recommendation
  • Open Access IconOpen Access
  • Cite Icon12
  • https://doi.org/10.1145/3529372.3530921Copy DOI Icon

Causal factorization machine for robust recommendation

  • Jun 20, 2022
  • Yunqi Li +3 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Factorization Machines (FMs) are widely used for the collaborative recommendation because of their effectiveness and flexibility in feature interaction modeling. Previous FM-based works have claimed the importance of selecting useful features since incorporating unnecessary features will introduce noise and reduce the recommendation performance. However, previous feature selection algorithms for FMs are proposed based on the i.i.d. hypothesis and select features according to their importance to the predictive accuracy on training data. However, the i.i.d. assumption is often violated in real-world applications, and shifts between training and testing sets may exist. In this paper, we consider achieving causal feature selection in FMs so as to enhance the robustness of recommendation when the distributions of training data and testing data are different. What's more, different from other machine learning tasks like image classification, which usually select a global set of causal features for a predictive model, we emphasize the importance of considering personalized causal feature selection in recommendation scenarios since the causal features for different users may be different. To achieve our goal, we propose a personalized feature selection method for FMs and refer to the confounder balancing approach to balance the confounders for every treatment feature. We conduct experiments on three real-world datasets and compare our method with both representative shallow and deep FM-based baselines to show the effectiveness of our method in enhancing the robustness of recommendations and improving the recommendation accuracy.

Similar Papers
  • Research Article
  • Citations542

Local Causal and Markov Blanket Induction for Causal Discovery and Feature Selection for Classification Part I: Algorithms and Empirical Evaluation

  • Mar 01, 2010
  • Journal of Machine Learning Research
  • Constantin F Aliferis +4
  • Research Article
  • Citations126

Bearing fault diagnosis using transfer learning and optimized deep belief network

  • Mar 07, 2022
  • Measurement Science and Technology
  • Huimin Zhao +4
  • Research Article
  • Citations12

Causal feature selection using a knowledge graph combining structured knowledge from the biomedical literature and ontologies: A use case studying depression as a risk factor for Alzheimer’s disease

  • Apr 21, 2023
  • Journal of Biomedical Informatics
  • Scott A Malec +8
  • PDF
  • Research Article
  • Citations56

Pushing the limits of solubility prediction via quality-oriented data selection.

  • Dec 17, 2020
  • iScience
  • Murat Cihan Sorkun +2
  • Book Chapter

Iterative Reinforcement Cross-Domain Text Classification

  • Oct 08, 2008
  • Di Zhang +2
  • Research Article

Optimum Feature Selection for Efficient ISAR Image Classification

  • Feb 01, 2018
  • Indian Journal of Science and Technology
  • B Mamatha +1
  • PDF
  • Research Article
  • Citations11

Radar Emitter Identification under Transfer Learning and Online Learning

  • Dec 25, 2019
  • Information
  • Yuntian Feng +5
  • Front Matter
  • Citations45

Signal quality in cardiorespiratory monitoring

  • Aug 17, 2012
  • Physiological Measurement
  • Gari D Clifford +1
  • Research Article
  • Citations8

Joint nonlinear feature selection and continuous values regression network

  • Oct 01, 2021
  • Pattern Recognition Letters
  • Zheng Wang +4
  • Conference Article
  • Citations7

Adversarial Poisoning of Importance Weighting in Domain Adaptation

  • Nov 01, 2018
  • Muhammad Umer +2
  • Research Article
  • Citations5

Peramalan Data Cuaca Ekstrim Indonesia Menggunakan Model ARIMA dan Recurrent Neural Network

  • Feb 27, 2023
  • Jambura Journal of Mathematics
  • Hikmah Hikmah +3
  • Research Article
  • Citations84

Learning Meta Pattern for Face Anti-Spoofing

  • Jan 01, 2022
  • IEEE Transactions on Information Forensics and Security
  • Rizhao Cai +5
  • Conference Article
  • Citations2

IO-aware Factorization Machine for User Response Prediction

  • Jul 01, 2020
  • Zhenhao Hu +3
  • Supplementary Content

Causal Feature Selection in Neuroscience

  • Dec 18, 2020
  • Universitätsbibliothek Tübingen
  • Anastasia Atalanti Mastakouri
  • Research Article
  • Citations8

Analysis of Individual Loan Defaults Using Logit under Supervised Machine Learning Approach

  • May 01, 2019
  • Asian Journal of Probability and Statistics
  • Dominic M Obare +2
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.