跳到主要导航 跳到搜索 跳到主要内容

GraphRARE: Reinforcement Learning Enhanced Graph Neural Network with Relative Entropy

  • Tianhao Peng
  • , Wenjun Wu*
  • , Haitao Yuan*
  • , Zhifeng Bao
  • , Zhao Pengru
  • , Xin Yu
  • , Xuetao Lin
  • , Yu Liang
  • , Yanjun Pu
  • *此作品的通讯作者
  • Beihang University
  • Nanyang Technological University
  • Royal Melbourne Institute of Technology University
  • Beijing University of Technology

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Graph neural networks (GNNs) have shown ad-vantages in graph-based analysis tasks. However, most existing methods have the homogeneity assumption and show poor performance on heterophilic graphs, where the linked nodes have dissimilar features and different class labels, and the semantically related nodes might be multi-hop away. To address this limitation, this paper presents GraphRARE, a general framework built upon node relative entropy and deep reinforcement learning, to strengthen the expressive capability of GNNs. An innovative node relative entropy, which considers node features and structural similarity, is used to measure mutual information between node pairs. In addition, to avoid the sub-optimal solutions caused by mixing useful information and noises of remote nodes, a deep reinforcement learning-based algorithm is developed to optimize the graph topology. This algorithm selects informative nodes and discards noisy nodes based on the defined node relative en-tropy. Extensive experiments are conducted on seven real-world datasets. The experimental results demonstrate the superiority of GraphRARE in node classification and its capability to optimize the original graph topology.

源语言英语
主期刊名Proceedings - 2024 IEEE 40th International Conference on Data Engineering, ICDE 2024
出版商IEEE Computer Society
2489-2502
页数14
ISBN(电子版)9798350317152
DOI
出版状态已出版 - 2024
活动40th IEEE International Conference on Data Engineering, ICDE 2024 - Utrecht, 荷兰
期限: 13 5月 202417 5月 2024

出版系列

姓名Proceedings - International Conference on Data Engineering
ISSN(印刷版)1084-4627
ISSN(电子版)2375-0286

会议

会议40th IEEE International Conference on Data Engineering, ICDE 2024
国家/地区荷兰
Utrecht
时期13/05/2417/05/24

指纹

探究 'GraphRARE: Reinforcement Learning Enhanced Graph Neural Network with Relative Entropy' 的科研主题。它们共同构成独一无二的指纹。

引用此