Explicit Feature Interaction-Aware Graph Neural Network

被引:3
作者
Kim, Minkyu [1 ]
Choi, Hyun-Soo [2 ]
Kim, Jinho [3 ]
机构
[1] Ziovis Co Ltd, Dept Res & Dev, Chunchon 24341, South Korea
[2] Seoul Natl Univ Sci & Technol, Dept Comp Sci & Engn, Seoul 01811, South Korea
[3] Kangwon Natl Univ, Dept Comp Sci & Engn, Chunchon 24341, South Korea
基金
新加坡国家研究基金会;
关键词
Graph neural networks; feature interactions; interpretable AI; CONVOLUTIONAL NETWORK;
D O I
10.1109/ACCESS.2024.3357887
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Graph neural networks (GNNs) are powerful tools for handling graph-structured data. However, their design often limits them to learning only higher-order feature interactions, leaving low-order feature interactions overlooked. To address this problem, we introduce a novel GNN method called explicit feature interaction-aware graph neural network (EFI-GNN). Unlike conventional GNNs, EFI-GNN is a multilayer linear network designed to model arbitrary-order feature interactions explicitly within graphs. To validate the efficacy of EFI-GNN, we conduct experiments using various datasets. The experimental results demonstrate that EFI-GNN has competitive performance with existing GNNs, and when a GNN is jointly trained with EFI-GNN, predictive performance sees an improvement. Furthermore, the predictions made by EFI-GNN are interpretable, owing to its linear construction. The source code of EFI-GNN is available at https://github.com/gim4855744/EFI-GNN.
引用
收藏
页码:15438 / 15446
页数:9
相关论文
共 42 条
  • [1] Brody S., 2022, P INT C LEARN REPR I, P1
  • [2] Chen J., 2018, P INT C LEARN REPR I
  • [3] Chen Ming, 2020, P MACHINE LEARNING R, V119
  • [4] Learnable graph convolutional network and feature fusion for multi-view learning
    Chen, Zhaoliang
    Fu, Lele
    Yao, Jie
    Guo, Wenzhong
    Plant, Claudia
    Wang, Shiping
    [J]. INFORMATION FUSION, 2023, 95 : 109 - 119
  • [5] Cheng H.-T., 2016, P ACM C REC SYST, P7, DOI DOI 10.1145/2988450.2988454
  • [6] Daelemans W., 2014, P C EMPIRICAL METHOD, P1724
  • [7] Ding KZ, 2020, Arxiv, DOI arXiv:1908.07110
  • [8] Graph Neural Networks for Social Recommendation
    Fan, Wenqi
    Ma, Yao
    Li, Qing
    He, Yuan
    Zhao, Eric
    Tang, Jiliang
    Yin, Dawei
    [J]. WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, : 417 - 426
  • [9] Fey Matthias, 2019, ICLR WORKSH REPR LEA
  • [10] Gravina A., 2023, P INT C LEARN REPR I