Explicit Feature Interaction-aware Graph Neural Networks

04/07/2022
by   Minkyu Kim, et al.
0

Graph neural networks are powerful methods to handle graph-structured data. However, existing graph neural networks only learn higher-order feature interactions implicitly. Thus, they cannot capture information that occurred in low-order feature interactions. To overcome this problem, we propose Explicit Feature Interaction-aware Graph Neural Network (EFI-GNN), which explicitly learns arbitrary-order feature interactions. EFI-GNN can jointly learn with any other graph neural network. We demonstrate that the joint learning method always enhances performance on the various node classification tasks. Furthermore, since EFI-GNN is inherently a linear model, we can interpret the prediction result of EFI-GNN. With the computation rule, we can obtain an any-order feature's effect on the decision. By that, we visualize the effects of the first-order and second-order features as a form of a heatmap.

READ FULL TEXT

page 7

page 9

research
08/19/2019

Graph Neural Networks with High-order Feature Interactions

Network representation learning, a fundamental research problem which ai...
research
06/08/2021

Principled Hyperedge Prediction with Structural Spectral Features and Neural Networks

Hypergraph offers a framework to depict the multilateral relationships i...
research
06/28/2019

GNN-FiLM: Graph Neural Networks with Feature-wise Linear Modulation

This paper presents a new Graph Neural Network (GNN) type using feature-...
research
05/25/2021

GraphFM: Graph Factorization Machines for Feature Interaction Modeling

Factorization machine (FM) is a prevalent approach to modeling pairwise ...
research
03/04/2019

Graph Neural Networks for Modelling Traffic Participant Interaction

By interpreting a traffic scene as a graph of interacting vehicles, we g...
research
08/20/2021

TabGNN: Multiplex Graph Neural Network for Tabular Data Prediction

Tabular data prediction (TDP) is one of the most popular industrial appl...
research
07/27/2022

Gaia: Graph Neural Network with Temporal Shift aware Attention for Gross Merchandise Value Forecast in E-commerce

E-commerce has gone a long way in empowering merchants through the inter...

Please sign up or login with your details

Forgot password? Click here to reset