提出两种新方法,让带符号图整体表示更准确
Whole-Graph Representation Learning For the Classification of Signed Networks
- 用改进的图遍历法和主节点机制学习图整体表示
- 在三个数据集上最高达81.20的F-measure,优于基线58.57
- 适合需要图级分类的科研人员,开源代码可复现
图广泛用于建模包含结构化数据与关系的复杂系统。近年来,图表示学习旨在自动学习图的低维表示,受到广泛关注。现有方法大多处理无符号图,但带符号图能建模对立关系,在越来越多领域出现。已有研究关注顶点级表示,但仅有一种方法适用于图级表示,且仅限于完全连通图。本文提出两种通用带符号图的图级表示学习方法:第一种是SG2V,为Graph2vec的带符号推广,基于修改的Weisfeiler-Lehman重标记过程;第二种是WSGCN,为SGCN的图级推广,通过引入主节点实现。我们还提出了多种变体。为解决图级方法缺乏数据的问题,我们构建了一个由三个带标签图集合组成的基准测试集。在该基准上评估,结果表明所提方法在图分类任务中表现更优:基线F-measure为58.57,而SG2V和WSGCN分别达到73.01和81.20。源代码与数据集均公开可用。
原文摘要 · Abstract (English)
Graphs are ubiquitous for modeling complex systems involving structured data and relationships. Consequently, graph representation learning, which aims to automatically learn low-dimensional representations of graphs, has drawn a lot of attention in recent years. The overwhelming majority of existing methods handle unsigned graphs. However, signed graphs appear in an increasing number of application domains to model systems involving two types of opposed relationships. Several authors took an interest in signed graphs and proposed methods for providing vertex-level representations, but only one exists for whole-graph representations, and it can handle only fully connected graphs. In this article, we tackle this issue by proposing two approaches to learning whole-graph representations of general signed graphs. The first is a SG2V, a signed generalization of the whole-graph embedding method Graph2vec that relies on a modification of the Weisfeiler--Lehman relabelling procedure. The second one is WSGCN, a whole-graph generalization of the signed vertex embedding method SGCN that relies on the introduction of master nodes into the GCN. We propose several variants of both these approaches. A bottleneck in the development of whole-graph-oriented methods is the lack of data. We constitute a benchmark composed of three collections of signed graphs with corresponding ground truths. We assess our methods on this benchmark, and our results show that the signed whole-graph methods learn better representations for this task. Overall, the baseline obtains an F-measure score of 58.57, when SG2V and WSGCN reach 73.01 and 81.20, respectively. Our source code and benchmark dataset are both publicly available online.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。