中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
GraphFM: Graph Factorization Machines for Feature Interaction Modeling

文献类型:期刊论文

作者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.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。