mlsanityData to 14 Oct 2026

GraphFM: Graph Factorization Machines for Feature Interaction Modeling

arXiv 2105.11866 ↗First posted 25 May 2021cs.LG cs.AI cs.IRcited by 13

Not on the map. Why?

Abstract (arXiv)

Factorization machine (FM) is a prevalent approach to modeling pairwise (second-order) feature interactions when dealing with high-dimensional sparse data. However, on the one hand, FM fails to capture higher-order feature interactions suffering from combinatorial expansion. On the other hand, taking into account interactions between every pair of features may introduce noise and degrade prediction accuracy. To solve the problems, we propose a novel approach, Graph Factorization Machine (GraphFM), by naturally representing features in the graph structure. In particular, we design a mechanism to select the beneficial feature interactions and formulate them as edges between features. Then the proposed model, which integrates the interaction function of FM into the feature aggregation strategy of Graph Neural Network (GNN), can model arbitrary-order feature interactions on the graph-structured features by stacking layers. Experimental results on several real-world datasets have demonstrated the rationality and effectiveness of our proposed approach. The code and data are available at https://github.com/CRIPAC-DIG/GraphCTR}{https://github.com/CRIPAC-DIG/GraphCTR

Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Builds on 52 of its 75 references that are in the map. First posted on arXiv 25 May 2021.

Builds on

52 of 75 references.

References · 52

  1. 2017Attention Is All You Needcited by 60,361
  2. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  4. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  5. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  6. 2017Graph Attention Networkscited by 8,923
  7. 2017Inductive Representation Learning on Large Graphscited by 6,320
  8. 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
  9. 2018How Powerful are Graph Neural Networks?cited by 4,393
  10. 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
  11. 2016Understanding deep learning requires rethinking generalizationcited by 1,674
  12. 2017Neural Collaborative Filteringcited by 1,111
38 more
  1. 2013Spectral Networks and Locally Connected Networks on Graphscited by 1,102
  2. 2015Gated Graph Sequence Neural Networkscited by 796
  3. 2016Wide & Deep Learning for Recommender Systemscited by 747
  4. 2017DeepFM: A Factorization-Machine based Neural Network for CTR Predictioncited by 676
  5. 2019Neural Graph Collaborative Filteringcited by 661
  6. 2017Deep & Cross Network for Ad Click Predictionscited by 393
  7. 2018Graph Convolutional Networks for Text Classificationcited by 345
  8. 2020DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systemscited by 327
  9. 2018AutoInt: Automatic Feature Interaction Learning via Self-Attentive Neural Networkscited by 320
  10. 2018Session-based Recommendation with Graph Neural Networkscited by 312
  11. 2018xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systemscited by 278
  12. 2019Learning Discrete Structures for Graph Neural Networkscited by 231
  13. 2020Data Augmentation for Graph Neural Networkscited by 203
  14. 2020GNNGuard: Defending Graph Neural Networks against Adversarial Attackscited by 195
  15. 2017Neural Factorization Machines for Sparse Predictive Analyticscited by 183
  16. 2020Graph Information Bottleneckcited by 168
  17. 2018Adaptive Graph Convolutional Neural Networkscited by 155
  18. 2016Product-based Neural Networks for User Response Predictioncited by 145
  19. 2017Encoding Sentences with Graph Convolutional Networks for Semantic Role Labelingcited by 137
  20. 2017Attentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networkscited by 126
  21. 2020AM-GCN: Adaptive Multi-channel Graph Convolutional Networkscited by 121
  22. 2020Revisiting Graph based Collaborative Filtering: A Linear Residual Graph Convolutional Network Approachcited by 120
  23. 2020Differentiable Graph Module (DGM) for Graph Convolutional Networkscited by 86
  24. 2016Deep Learning over Multi-field Categorical Data: A Case Study on User Response Predictioncited by 67
  25. 2016The More You Know: Using Knowledge Graphs for Image Classificationcited by 63
  26. 2018Graph-to-Sequence Learning using Gated Graph Neural Networkscited by 61
  27. 2019Graph-Revised Convolutional Networkcited by 59
  28. 2019Fi-GNN: Modeling Feature Interactions via Graph Neural Networks for CTR Predictioncited by 54
  29. 2020AutoFIS: Automatic Feature Interaction Selection in Factorization Models for Click-Through Rate Predictioncited by 52
  30. 2019Adaptive Factorization Network: Learning Adaptive-Order Feature Interactionscited by 48
  31. 2016Higher-Order Factorization Machinescited by 45
  32. 2017Situation Recognition with Graph Neural Networkscited by 34
  33. 2018Constructing Narrative Event Evolutionary Graph for Script Event Predictioncited by 33
  34. 2021FM2: Field-matrixed Factorization Machines for Recommender Systemscited by 31
  35. 2019Dressing as a Whole: Outfit Compatibility Learning Based on Node-wise Graph Neural Networkscited by 26
  36. 2020Detecting Beneficial Feature Interactions for Recommender Systemscited by 22
  37. 2019A Quest for Structure: Jointly Learning the Graph Structure and Semi-Supervised Classificationcited by 13
  38. 2021Disentangled Self-Attentive Neural Networks for Click-Through Rate Predictioncited by 5

50 most cited of 52 All 52 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2022: 4; 2023: 5; 2024: 1; 2025: 2; 2026: 1.

13 papers.

Sort:

By area: