|
作者 | Shu Wu2 ; Zekun Li ; Yunyue Su2; Zeyu Cui ; Xiaoyu Zhang1; Liang Wang2
|
刊名 | Machine Intelligence Research
 |
出版日期 | 2024
|
页码 | 1 |
文献子类 | 期刊论文(录用)
|
英文摘要 |
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 Factoriza-
tion 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 Neu-
ral 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 ratio-
nality and effectiveness of our proposed approach. The code and
data are available at https://github.com/CRIPAC-DIG/GraphCTR. |
源URL | [http://ir.ia.ac.cn/handle/173211/57490]  |
专题 | 自动化研究所_智能感知与计算研究中心
|
作者单位 | 1.中国科学院信息工程研究所 2.中国科学院自动化研究所
|
推荐引用方式 GB/T 7714 |
Shu Wu,Zekun Li,Yunyue Su,et al. GraphFM: Graph Factorization Machines for Feature Interaction Modeling[J]. Machine Intelligence Research,2024:1.
|
APA |
Shu Wu,Zekun Li,Yunyue Su,Zeyu Cui,Xiaoyu Zhang,&Liang Wang.(2024).GraphFM: Graph Factorization Machines for Feature Interaction Modeling.Machine Intelligence Research,1.
|
MLA |
Shu Wu,et al."GraphFM: Graph Factorization Machines for Feature Interaction Modeling".Machine Intelligence Research (2024):1.
|