Feature recommendation strategy for graph convolutional network

Jisheng Qin, Xiaoqin Zeng, Shengli Wu, Yang Zou

Research output: Contribution to journalArticlepeer-review

Abstract

Graph Convolutional Network (GCN) is a new method for extracting, learning, and inferencing graph data that builds an embedded representation of the target node by aggregating information from neighbouring nodes. GCN is decisive for node classification and link prediction tasks in recent research. Although the existing GCN performs well, we argue that the current design ignores the potential features of the node. In addition, the presence of features with low correlation to nodes can likewise limit the learning ability of the model. Due to the above two problems, we propose Feature Recommendation Strategy (FRS) for Graph Convolutional Network in this paper. The core of FRS is to employ a principled approach to capture both node-to-node and node-to-feature relationships for encoding, then recommending the maximum possible features of nodes and replacing low-correlation features, and finally using GCN for learning of features. We perform a node clustering task on three citation network datasets and experimentally demonstrate that FRS can improve learning on challenging tasks relative to state-of-the-art (SOTA) baselines.
Original languageEnglish
Pages (from-to)1697-1718
Number of pages22
JournalConnection Science
Volume34
Issue number1
Early online date14 Jun 2022
DOIs
Publication statusE-pub ahead of print - 14 Jun 2022

Bibliographical note

Funding Information:
This work was supported by the Postgraduate Research & Practice Innovation Program of Jiangsu Province (KYCX21_0408).

Publisher Copyright:
© 2022 The Author(s). Published by Informa UK Limited, trading as Taylor & Francis Group.

Keywords

  • Artificial Intelligence
  • Human-Computer Interaction
  • Software
  • graph convolutional network
  • GCN
  • Feature recommendation strategy

Cite this