GraphSage

news2024/11/18 4:31:52

背景

大型图中节点的低维嵌入在各种预测任务中非常有用。GraphSage是一种通用的归纳框架，它利用节点特征信息（例如，文本属性）有效地为以前看不见的数据生成节点嵌入。相比于对每个节点训练单独的嵌入，GraphSage学习了一个函数，通过对节点局部邻域的特征进行采样和聚合来生成嵌入。

如图1所示，不同于对每个节点训练一个对应的嵌入向量，GraphSage训练一组聚合器函数，这些函数学习聚合来自节点邻居的特征信息。

算法实现

GraphSage的核心思想是聚合节点本地邻域的特征信息。具体算法如下：

输入:

包括图 $g(V, \epsilon)$ ，特征 $\{x_v, \forall v \in V\}$ ，遍历次数K，

权重矩阵 $W^k, \forall k \in \{1, 2, 3, ..., K\}$ ，非线性函数 $\sigma$ ，

可微聚合器函数 ${AGGREGATE}_k, \forall k \in \{1, 2, ...K \}$ ，邻居函数 $N: v \rightarrow 2^v$

输出：

每个节点的向量表示 $z_v$

第一步（1）：对于每个节点获取其特征 $x_v$ 作为基准特征。

第二步（2—8）：

遍历1到K，在每次遍历中遍历每个节点v。以第一次遍历为例，对于节点v，首先获取其邻居集合N(v)，使用可微聚合器函数聚合邻居集合里面所有的节点的基准特征得到聚合邻居特征 $h^{k}_{N(v)}$ 。连接基准特征和聚合邻居特征得到联合特征，使用权重矩阵与联合特征相乘得到此次遍历后的节点特征 $h_v^k$ 。最后对特征进行归一化，结束此次遍历。