论文阅读“Bilinear Graph Neural Network with Neighbor Interactions”

Zhu H, Feng F, He X, et al. Bilinear graph neural network with neighbor interactions[C]//Proceedings of the Twenty-Ninth International Conference on International Joint Conferences on Artificial Intelligence. 2021: 1452-1458.

摘要导读

现有的GNN的图卷积操作大多使用加权求和或者元素对位相乘以聚合近邻节点的信息。虽然已经得到了广泛的应用,但是在这类的操作中忽略了这些节点之间的关联关系。而存在共线关系的两个近邻节点对于目标节点表示的学习是有指示意义的,而现有的方法无法捕获这种指示关系。
为了捕获近邻节点之间的关联性,本文设计了一种双线性图卷积操作,通过邻居节点表示的成对关联交互来增强加权和。

作者首先介绍了现有的两种图卷积操作:

并指出两个向量之间的乘法是建模交互作用的一种有效方法——强调共同属性和稀释差异信息。

模型浅析

数据声明:G=(A \in \{0,1\}^{N \times N}, X \in \mathbb{R}^{N \times F})A为邻接矩阵,X为特征矩阵。除此之外,还给定了给定节点v的近邻点集合\mathcal{N}(v)=\{i | A_{vi}=1\},扩展的近邻集合\tilde{\mathcal{N}}(v)=\{v\} \cup \mathcal{N}(v)。且定义节点v的度为d_v=|\mathcal{N}(v)|,对应的\tilde{d}_v=|\tilde{\mathcal{N}}(v)|=d_v+1
模型设计的目标是学习每个节点v的表示向量h_v \in \mathbb{R}^D。对于下游的分类任务则可以在不需要图结构和原始节点特征的情况下得出标签输出y_v=f(h_v)

  • Bilinear Aggregator
    受因子分解机的启发,作者提出了一种适用于局部结构中邻居相互作用建模的双线性聚合器:

    其中,\odot代表对位相乘,v是需要学习的目标节点的表示,ij分别是\tilde{\mathcal{N}}(v)集合的下标,i < j的条件限制是为了避免由自关联引入的噪声。b_v=\frac{1}{2}\tilde{d}_v(\tilde{d}_v-1)代表给定节点v对应的关联数目,用于对得到的表示进行正则化。值得注意的是,在双线性聚合操作中也考虑了目标节点本身,并聚合了来自扩展邻居的信息,这些信息虽然看起来与GNN相同,但原因不同。在GNN中,考虑目标节点是在分层聚合过程中保留其信息,类似于残差学习。而在BGNN中,考虑的是目标节点与其邻居之间的相互作用也可能携带有用的信号。例如,对于只有一个邻居的稀疏节点,邻居之间的交互不存在,而目标节点和邻居节点之间的交互可能特别有用。

  • BGNN Model
    由于双线性聚合器(BA)强调节点交互,加权和聚合器(AGG)可以对不同的信号进行编码,在提出的模型中BGNN中将它们结合起来,构建一个更具表现性的图卷积网络。

    这里采用的是简单线性组合机制。H^{(k)}存储的是第k层的节点表示(编码的是k-hop近邻),\alpha是一个超参数来平衡传统的GNN聚合器和提出的双线性聚合器的强度。
    由于AGGBA的置换不变性,组合之后的BGNN也具有相同的性质。

  • 多层BGNN
    传统的GNN模型通过堆叠多个聚合器AGG,以递归的方式编码来自多跳邻居节点的信息。一个两层的GNN模型形式化为如下:

    因此作者也表示,最直接的扩展为多层的方式为:
    ,但是这种方式可能存在并不期望的高阶关联的属性。例如,在两层的表示学习中可能会在2阶的近邻节点中包含部分4阶的关联属性。出于这个原因,重新定义了一个2层的BGNN模型:
    其中,A^{(2)}存储的是图的2-hop连通结构。binarize是一个将矩阵中非零项转成1的操作。\beta是一个调节的超参数。由此,可以进一步定义k层BGNN:
    同理,A^{(k)}存储的是图的k-hop连通结构。因此,k层BGNN的计算的复杂性取决于A^{(k)}中的非零项。在实际的操作中,可以使用采样的方式使用部分高阶节点代替全部节点进行计算。

  • 模型训练
    可以看出,BGNN是一个可以由微分损失进行端到端优化的模型。
    在当前的半监督节点分类任务中,使用的是标记数据的交叉熵损失。
    并且作者指出,因为实验数据的规模都比较小,因此,layer-wise的图卷积操作都是以整个数据矩阵的形式进行运算的,关于batch的实现和近邻节点的采样将会在后续大规模图的工作中。


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

推荐阅读更多精彩内容