arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.01410cs.LGcs.AIcs.ITcs.NEmath.IT
收藏

图神经网络的最大熵加权独立集池

Maximum Entropy Weighted Independent Set Pooling for Graph Neural Networks

Amirkabir University of Technology(阿米尔卡比尔理工大学)

Amirhossein Nouranizadeh, Mohammadjavad Matinkia, Mohammad Rahmati, Reza Safabakhsh

摘要:

在本文中,我们提出了一种新的池层图神经网络的基础上最大化之间的互信息池图和输入图。由于最大互信息是很难计算的,我们采用的香农容量的一个图作为归纳偏见,我们的池化方法。更确切地说,我们表明,池化层的输入图可以被视为一个嘈杂的通信信道的表示。对于这样的信道,发送属于图的独立集的符号产生可靠且无差错的信息传输。我们表明,达到最大的互信息是相当于找到一个最大的重量独立集的图的权重传达熵内容。通过这种通信理论的观点,我们提供了一个独特的角度来提出的问题,图池作为最大限度地提高信息传输速率在一个嘈杂的通信信道,实现了一个图神经网络。我们评估我们的方法,称为最大熵加权独立集池(MEWISPool),对图分类任务和最大独立集的组合优化问题。实验结果表明,我们的方法取得了国家的最先进的和竞争力的结果图分类任务和最大独立集问题在几个基准数据集。

英文摘要:

In this paper, we propose a novel pooling layer for graph neural networks based on maximizing the mutual information between the pooled graph and the input graph. Since the maximum mutual information is difficult to compute, we employ the Shannon capacity of a graph as an inductive bias to our pooling method. More precisely, we show that the input graph to the pooling layer can be viewed as a representation of a noisy communication channel. For such a channel, sending the symbols belonging to an independent set of the graph yields a reliable and error-free transmission of information. We show that reaching the maximum mutual information is equivalent to finding a maximum weight independent set of the graph where the weights convey entropy contents. Through this communication theoretic standpoint, we provide a distinct perspective for posing the problem of graph pooling as maximizing the information transmission rate across a noisy communication channel, implemented by a graph neural network. We evaluate our method, referred to as Maximum Entropy Weighted Independent Set Pooling (MEWISPool), on graph classification tasks and the combinatorial optimization problem of the maximum independent set. Empirical results demonstrate that our method achieves the state-of-the-art and competitive results on graph classification tasks and the maximum independent set problem in several benchmark datasets.

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

↑