PATCHY-SAN:卷积神经网络处理图结构数据

论文标题:Learning Convolutional Neural Networks for Graphs
论文链接:https://arxiv.org/abs/1605.05273
论文来源:ICML 2016

一、概述

在这篇论文中,我们的目的是将卷积神经网络应用于大量的基于图的学习问题。考虑以下两个问题:
①使用图的集合作为训练数据,来实现未见图上的分类和回归问题,例如,集合内的每个图可以模拟一个化合物,输出可以是一个函数,将未见化合物映射到它们对抗癌细胞的活性水平;
②对于一个大的图,可以学习图的表示来用来推断未见图的属性,比如节点的类型和缺失的边。

本文提出的PATCHY-SAN适用于处理这些问题,所面对的图数据可以是有向图或者无向图,图的节点和边可以有离散或者连续的属性并且可以有多种类型的边。

一张图片可以认为是一种特殊的图,即grid图,它的节点代表像素。在处理图像数据时,CNN处理的过程可以理解为:
①在一个节点序列上进行移动(下图(a)中的1-4节点);
②生成这些节点的固定大小的邻域图(下图(b)中3\times 3的grid)的过程。

CNN处理grid数据

这里的邻域图相当于感受野。上图(b)表示的是将这个邻域图从图的表示转化为向量表示的过程,这个过程也就是按照从左到右、从上到下的顺序将邻域图展平即可,然后将这个向量加权累加得到卷积后的结果。

由于图像数据存在天然的空间顺序,因而节点序列创建邻域图的顺序(也就是卷积核移动的顺序)总是从左到右、从上到下,然而在图数据中,图没有固定的空间顺序,而且不同的图可能有不同的拓扑结构。

类比上述CNN处理图像的过程,使用CNN来处理图数据可以分为以下两个类似的步骤:
①决定用于生成邻域图的节点序列;
②计算一个标准化的邻域图,也就是从图的表示到向量空间表示。

本文提出的PATCHY-SAN架构就用来实现上述的过程,其大体的流程如下图所示:

PATCHY-SAN

PATCHY-SAN有以下几个优点:
①它是高效的,天然能够并行化,并且适用于大型图;
②对于从计算生物学到社会网络分析的许多应用来说,将网络的可视化是很重要的,PATCHY-SAN支持特性可视化,能够观察图的结构属性;
③PATCHY-SAN学习 application dependent的特征,而不需要进行特征工程(feature engineering)。

PATCHY-SAN利用图标注(graph labeling)程序来为图的节点施加一个顺序。一个图标注程序l是一个函数l:V\rightarrow S,一个从V映射到S的函数,S是一个有序集合(ordered set),元素可以是实数或者整数。另外,一个ranking(或者叫coloring)是一个函数r:V\rightarrow \left \{1,\cdots ,|V|\right \}。我们利用图标注程序对图的节点进行标注,然后根据标注结果进行排序得到节点的ranking,规则是r(u)< r(v)当且仅当l(u)> l(v)。如果l是单射的,那么图G的ranking就是唯一的,按照这样的ranking组织的邻接矩阵A^{l}(G)也是唯一的。另外,图标注程序也表明了图的一个划分(partition)\left \{V_{1},\cdots ,V_{n}\right \},也就是u,v\in V_i当且仅当l(u)=l(v)

图标记程序的例子有在网络分析中常用的节点度和其他中心性(centrality)度量。比如节点v的betweeness centrality度量是计算通过v的最短路径的分数。总之图标记的方法有很多种,本文中并未详述采用的图标记方法。

二、方法

给定图数据的集合,PATCHY-SAN按照以下步骤进行处理每个图:
①从图中选择一个固定长度的节点序列;
②在选定的序列中为每个节点分配(assemble)一个固定大小的邻域;
③将提取的邻域图标准化;
④使用卷积神经网络学习邻域表示。

  1. 节点序列选择

Algorithm 1表示如何选择节点序列,其中w表示要选择的节点个数,这里的stride s代表步长,如果创建的感受野数量少于w(原因是因为s>1),那么要为选择到但未被创建感受野的节点创建全0感受野:

Algorithm 1

创建感受野的Algorithm 3如下:

Algorithm 3
  1. 邻域图的分配

Algorithm 1中需要为节点创建感受野,需要Algorithm 2应用广度优先搜索来获取节点的邻域节点,搜索距离v越来越远的节点,并将这些节点添加到集合N中作为候选(节点数量可能大于感受野大小k):

Algorithm 2
  1. 图的标准化

对于每个节点分配的邻域图需要进行进一步的标准化,标准化的过程为邻域图的节点施加了一个顺序,以便于能够将邻域图从图的表示转换到向量空间表示,下图展示了这个过程:

图的标准化

算法Algorithm 4如下:

Algorithm 4

从整个算法中可以看出来图标记程序l起到了非常关键的作用,可以认为一个图标记程序的效果好坏直接关系算法的性能。图标记程序的原则是当且仅当两个不同图的节点在图中的结构角色(structural role)相似时,将它们分配到各自邻接矩阵中相似的相对位置。那么怎么定义一个比较好的图标记程序呢?本文定义了最优图标准化问题(optimal graph normalization problem)来求解一个最优的图标记程序:

\mathcal{G}代表有k个节点的无标注图集合,l表示一个单射的图标注程序,d_{G}代表有k个节点的图的距离度量,d_{A}代表k\times k矩阵的距离度量,最优的图标记程序\hat{l}就是:
\hat{l}=\underset{l}{argmin}\; E_{\mathcal{G}}\left [\left |d_{A}\left (A^{l} (G),A^{l}(G^{'})\right )-d_{G}(G,G^{'})\right |\right ]

求解的过程也就是寻找一个图标记程序,满足任意两个从\mathcal{G}中采样的图,向量空间中图的距离(关于基于l的邻接矩阵)与图空间中图的距离之间的期望差异是最小的。

这样的求解方法符合前面提到的图标记程序的原则,d_{G}代表图的距离度量,可以采用图的编辑距离,d_{A}来衡量邻接矩阵的相似度,可以采用邻接矩阵的汉明距离。

最优图标准化问题是经典图规范化问题(graph canonicalization problem)的一种推广。然而,规范标注算法只适用于同构图,对于相似但非同构的图可能表现不佳。

虽然上面给出了如何求解一个最优的图标记程序,但是事实上这是一个NP-hard的问题。PATCHY-SAN并不能解决这个NP-hard的问题,也就是说最优的\hat{l}是求不出来的,不过PATCHY-SAN提供一种度量的标准来对比不同的图标记程序,然后挑选最优的:

\mathcal{G}代表有k个节点的无标注图集合,(G_{1},G_{1}^{'}),\cdots ,(G_{N},G_{N}^{'})是一个序列,这里面的图的pair都是从\mathcal{G}中随机采样出来的。另外有
\hat{\theta }_{l}:=\sum_{i=1}^{N}d_{A}\left (A^{l}(G),A^{l}(G^{'})\right )/N
以及
\theta _{l}:=E_{\mathcal{G}} \left [\left |d_{A}\left (A^{l} (G),A^{l}(G^{'}) \right )-d_{G}(G,G^{'})\right | \right ]
如果d_A\geq d_G,那么E_{\mathcal{G}} [\hat{\theta }_{l_{1}}] < E_{\mathcal{G}}[\hat{\theta }_{l_{2}}]当且仅当\theta _{l_{1}}< \theta _{l_{2}}

上述定理可以使我们比较不同的图标记程序时使用无监督的方式进行,在假设d_A\geq d_G下,估计量\hat{\theta }_{l}越小,\theta _{l}就会越小,因此我们会选择\hat{\theta }_{l}最小的图标记程序。假设d_A\geq d_G是可以成立,在d_{G}采用图的编辑距离,d_{A}采用邻接矩阵的汉明距离时即可成立。说明一点,上述方法可以拓展到有向图。

图的归一化问题以及在邻域图结构的标准化中应用适当的图标记程序是本文方法的核心。在Algorithm 4中对节点v进行标准化时还要受限于节点与v的距离,对于任意两个顶点u,w,如果uw更接近v,那么u的ranking总是比w高,在Algorithm 4用d来表示这样的排序方式。这样的限制确保v总在ranking为1的位置,并且离v越近的节点,在向量空间中的ranking就会越高。

  1. 卷积架构

PATCHY-SAN能够既能够处理节点特征也可以处理边的特征,使用a_v代表节点属性的数量(也就是特征的维度),a_e代表边属性的数量,那么对于每个输入图G,按照前述算法应用标准化感受野以后会得到两个tensor,形状分别为(w,k,a_v)(w,k,k,a_e),这两个tensor可以reshape成(wk,a_v)(wk^2,a_e)形状的tensor,这里a_va_e就相当于input channel的数量,然后对两个tensor分别应用卷积核感受野大小、步长分别为kk^2(感受野大小与步长一致)的1D卷积操作。

三、实验

简单展示一下一部分实验的效果,具体实验设置参看原文。

  1. 效率

下图表示了利用前述算法在不同数据上生成感受野的效率:

效率
  1. 性能
性能
性能
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 221,695评论 6 515
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,569评论 3 399
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 168,130评论 0 360
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,648评论 1 297
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,655评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,268评论 1 309
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,835评论 3 421
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,740评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,286评论 1 318
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,375评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,505评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,185评论 5 350
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,873评论 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,357评论 0 24
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,466评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,921评论 3 376
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,515评论 2 359

推荐阅读更多精彩内容