中文版 | English
CSpace
Joint hyperbolic and Euclidean geometry contrastive graph neural networks
Xu, Xiaoyu1,2; Pang, Guansong3,4; Wu, Di1,2; Shang, Mingsheng1,2
2022-09-01
摘要Graph Neural Networks (GNNs) have demonstrated state-of-the-art performance in a wide variety of analytical tasks. Current GNN approaches focus on learning representations in a Euclidean space, which are effective in capturing non-tree-like structural relations, but they fail to model complex relations in many real-world graphs, such as tree-like hierarchical graph structure. This paper instead proposes to learn representations in both Euclidean and hyperbolic spaces to model these two types of graph geometries. To this end, we introduce a novel approach - Joint hyperbolic and Euclidean geometry contrastive graph neural networks (JointGMC). JointGMC is enforced to learn multiple layer-wise optimal combinations of Euclidean and hyperbolic geometries to effectively encode diverse complex graph structures. Further, the performance of most GNNs relies heavily on the availability of large-scale manually labeled data. To mitigate this issue, JointGMC exploits proximitybased self-supervised information in different geometric spaces (i.e., Euclidean, hyperbolic, and Euclidean-hyperbolic interaction spaces) to regularize the (semi-) supervised graph learning. Extensive experimental results on eight real-world graph datasets show that JointGMC outperforms eight state-of-the-art GNN models in diverse graph mining tasks, including node classification, link prediction, and node clustering tasks, demonstrating JointGMC's superior graph representation ability. Code is available at https://github.com/chachatang/jointgmc. (c) 2022 Elsevier Inc. All rights reserved.
关键词Graph neural networks Hyperbolic embedding Contrastive learning Graph representation learning
DOI10.1016/j.ins.2022.07.060
发表期刊INFORMATION SCIENCES
ISSN0020-0255
卷号609页码:799-815
通讯作者Pang, Guansong(gspang@smu.edu.sg) ; Shang, Mingsheng(msshang@cigit.ac.cn)
收录类别SCI
WOS记录号WOS:000848168800012
语种英语