arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2104.13014cs.LG
收藏

基于互信息和自监督的双层聚合节点嵌入

Node Embedding using Mutual Information and Self-Supervision based Bi-level Aggregation

Independent University Bangladesh(孟加拉国独立大学)

Kashob Kumar Roy, Amit Roy, A K M Mahbubur Rahman, M Ashraful Amin, Amin Ahsan Ali

摘要:

图神经网络(GNN)通过聚集来自图中的邻域的信息来学习节点的低维表示。然而,传统的GNN由于其本地($l$ -跳邻域)聚合方案而遭受两个基本缺点。第一,并非邻域中的所有节点都携带目标节点的相关信息。由于GNN不排除其邻域中的噪声节点,因此不相关的信息被聚合,这降低了表示的质量。其次,传统GNN也无法捕获节点之间的远程非本地依赖关系。为了解决这些限制,我们利用互信息(MI)来定义两种类型的邻居,1)\textit{Local Neighborhood},其中节点在社区内密集连接,每个节点将与其邻居共享更高的MI,以及2)\textit{Non-Local Neighborhood},其中引入基于MI的节点聚类来组装信息丰富但图形上遥远的节点在同一个集群。为了生成节点表示,我们结合了由两级聚合生成的嵌入-局部聚合以聚合来自局部邻域的特征以避免噪声信息和非局部聚合以聚合来自非局部邻域的特征。此外,我们利用自我监督学习来估计MI与少量的标记数据。最后,我们表明,我们的模型显着优于国家的最先进的方法,在广泛的双折射和双折射图。

英文摘要:

Graph Neural Networks (GNNs) learn low dimensional representations of nodes by aggregating information from their neighborhood in graphs. However, traditional GNNs suffer from two fundamental shortcomings due to their local ($l$-hop neighborhood) aggregation scheme. First, not all nodes in the neighborhood carry relevant information for the target node. Since GNNs do not exclude noisy nodes in their neighborhood, irrelevant information gets aggregated, which reduces the quality of the representation. Second, traditional GNNs also fail to capture long-range non-local dependencies between nodes. To address these limitations, we exploit mutual information (MI) to define two types of neighborhood, 1) \textit{Local Neighborhood} where nodes are densely connected within a community and each node would share higher MI with its neighbors, and 2) \textit{Non-Local Neighborhood} where MI-based node clustering is introduced to assemble informative but graphically distant nodes in the same cluster. To generate node presentations, we combine the embeddings generated by bi-level aggregation - local aggregation to aggregate features from local neighborhoods to avoid noisy information and non-local aggregation to aggregate features from non-local neighborhoods. Furthermore, we leverage self-supervision learning to estimate MI with few labeled data. Finally, we show that our model significantly outperforms the state-of-the-art methods in a wide range of assortative and disassortative graphs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑