Paper Information
Title:Simple Unsupervised Graph Representation Learning
Authors: Yujie Mo、Liang Peng、Jie Xu, Xiaoshuang Shi、Xiaofeng Zhu
Sources:2022 AAAI
Paper:download
Code:download
Abstract
作者提出了一种简单的无监督图表示学习方法来进行有效和高效的对比学习。具体而言,通过构造多重损失探索结构信息与邻域信息之间的互补信息来扩大类间变化,并通过增加一个上限损失来实现正嵌入与锚嵌入之间的有限距离来减小类内变化。因此,无论是扩大类间变化还是减少类内变化,都能使泛化误差很小,从而得到一个有效的模型。此外,作者的方法消除了以往图对比学习方法中广泛使用的数据增强和鉴别器,同时可以输出低维嵌入,从而得到一个高效的模型。在各种真实数据集上的实验结果表明,与最先进的方法相比,该方法是有效和高效的。
1 Introduction
As one of the representative methods of the UGRL,contrastive learning was proposed to maximize Mutual Information (MI) between the input content and its related content.
The key difference among graph contrastive learning methods is the definitions of the input contents and their related contents.
-
- Deep Graph Infomax (DGI) maximizes the MI between the node representations and the summary of the graph.
- Graphical Mutual Information (GMI) maximizes the MI between the input graph and the output graph.
- GCA (Zhu et al. 2021) maximize the MI between two views for each node through a variety of data augmentations, e.g., attribute masking or edge perturbation.
当前对比学习现状:通常依赖于数据增强来生成输入内容( input content)及其相关内容(related content),以实现 MI 最大化,从而导致训练过程的计算成本昂贵。
对比学习常存在的问题:数据增强、高维嵌入表示、对比损失设计,有关方法对比如 Figure 1 所示。
比如:
-
- 数据增强:Grace 和 GCA 通过 remove edges 和 mask node 特性,以生成多个视图。因此,数据增强(包括数据生成和数据编码)的计算成本大约需要 20%-40% 的训练时间。
- 高维度嵌入:现有的工作增加了嵌入的维数来提高表示质量,从而增加了训练时间。原因是这些方法的有效性对维度很敏感。如 DGI 和 GMI 在 512 维空间上达到了它们最好的精度。
- 目标函数设计为一个鉴别器:DGI 和 MVGRL 使用一个 discriminator 来测量节点嵌入和图嵌入的一致性,花费约 10%-30% 的训练时间。
基于上述三个问题,本文提出 Simple Unsupervised Graph Representation Learning (SUGRL),框架如 Figure 2 所示:
方法步骤概述:
-
- 首先使用一个多层感知器(MLP)在带语义信息(semantic information)的输入上生成 anchor embedding ;
- 接着基于 Graph structure 和 Semantic feature 使用 GCN 生成 Positive embedding,此外在 Anchor embedding 上通过 Neighbour sampling 生成另外一种 Positive embedding;
- 然后在 Anchor embedding 上通过 Row shuffling 生成一种 Negative embedding;
- 计算损失。
本文设计了一种新的多重损失,思想是锚点嵌入接近于正嵌入,而远离负嵌入。【通过减少类内距离,加大类间距离】
本文贡献:
-
- 首先,为了保证其有效性,我们提出联合考虑结构信息和邻居信息来探索它们的互补信息,以扩大类间的变异,并设计一个上界损失来实现较小的类内变异;
- 其次,为了实现效率,我们在对比学习中去掉了数据增强和鉴别器。这使得我们的方法在大规模数据集上实现可伸缩性;
- 最后,对 8 个公共基准数据集进行了综合的实证研究,验证了该方法与 11 种比较方法在节点分类方面的有效性和有效性;
2 Method
Notations
-
- Letting $\mathcal{G}=(\mathcal{V}, \mathcal{E})$ denote a graph, where $\mathcal{V}=\left\{v_{1}, v_{2}, \cdots, v_{N}\right\}$ and $\mathcal{E} \subseteq \mathcal{V} \times \mathcal{V}$ represent the node set and the edge set, respectively. We denote the feature matrix and the adjacency matrix as $\mathbf{X}=\left\{\mathbf{x}_{i}\right\}_{i=1}^{N}$ and $\mathbf{A} \in\{0,1\}^{N \times N} $, respectively, where $\mathbf{x}_{i}$ is the feature of the node $v_{i}$ , and $a_{i j}=1$ if $\left(v_{i}, v_{j}\right) \in \mathcal{E}$ otherwise $a_{i j}=0$ .
- In addition, we assume the existence of a set of latent classes $\mathcal{C}$ over the representation space $\mathcal{H}$ , since neither the embeddings nor the samples are labeled in unsupervised learning.
2.1 Anchor and negative embedding generation
现有的工作大多将 node representation 和 graph summary 作为一个 anchor 。
比如
-
- DGI and MVGRL treat the graph summary as anchors, which is first convolved by GCN and then summarized by a readout function.
- GRACE and GCA regard the node embedding generated in one view as anchors.
上述产生的问题:大多需要经过 GCN 的传播,比较耗费时间。
本文生成 Amchor embedding 的方法是使用 MLP 作用在输入 $X$ 上,从而生成带语义信息的 anchor embedding 。
$\begin{array}{l}\mathbf{X}^{(l+1)}=\text { Dropout }\left(\sigma\left(\mathbf{X}^{(l)} \mathbf{W}^{(l)}\right)\right) \quad\quad \quad\quad (1)\\ \mathbf{H}=\mathbf{X}^{(l+1)} \mathbf{W}^{(l+1)}\quad\quad\quad \quad\quad\quad\quad\quad\quad\quad\quad(2)\end{array}$
关于生成 Negative embedding,比较流行的方法如 DGI、GIC、MVGRL 采用的策略是:从原始图中得到一个被破坏的图,然后用 GCN 进行处理。本文生成 Negative embedding 的方法是直接通过 row-shuffle 去打乱 Amchor embedding ,从而生成 Negative embedding 。
$\mathbf{H}^{-}=\operatorname{Shuffle}\left(\left[\mathbf{h}_{1}, \mathbf{h}_{2}, \ldots, \mathbf{h}_{N}\right]\right)\quad\quad\quad(2)$
2.2 Positive embedding generation
现有的工作通常将结构信息 [ 数据增强后的视图为新的图结构 ] 视为正嵌入(Positive embedding),比如相关的工作有 DGI、MVGRL、GRACE、GCA、GIC 。
比如:
-
- 在 GCA 和 GRACE 中采用随机图数据增强;
- 在 MVGRL 中的图扩散;
本文生成了两种 Positive embedding :
-
- structural embeddings
- neighbor embeddings
2.2.1 Structural information
为了获取图的结构信息,本文采用了广泛使用的 GCN(以邻接矩阵 $A$ 和特征矩阵 $X$ 为输入) 作为基编码器:
$\mathbf{H}^{+^{(l+1)}}=\sigma\left(\widehat{\mathbf{A}} \mathbf{H}^{+^{(l)}} \mathbf{W}^{(l)}\right)\quad\quad\quad\quad(4)$
其中:
-
- $\mathbf{H}^{+(0)}=\mathbf{X}$ 代表输入特征;
- $\mathbf{H}^{+(l)}$ 代表了第 $l^{\text {th }}$ 层的特征;
- $ \widehat{\mathbf{A}}=\hat{\mathbf{D}}^{-1 / 2} \tilde{\mathbf{A}} \hat{\mathbf{D}}^{-1 / 2} \in \mathbb{R}^{N \times N}$ 代表了对称标准化的邻接矩阵;
- $\hat{\mathbf{D}} \in \mathbb{R}^{N \times N}$ 是度矩阵;
- $\tilde{\mathbf{A}}=\mathbf{A}+\mathbf{I}_{N}$ 是带自环的邻接矩阵;
需要注意的是本文生成 Anchor embedding 的 MLP 是和 这里的 GCN 共享权重矩阵 $W$ 的。【减少运行时间】
2.2.2 Neighbor information
为了得到具有邻居信息的 Positive embedding,作者首先存储所有节点的邻居嵌入索引,然后对其进行抽样,然后计算样本的平均值。这样可以有效地获取节点的邻居信息:
$\widetilde{\mathbf{h}}_{i}^{+}=\frac{1}{m} \sum\limits _{j=1}^{m}\left\{\mathbf{h}_{j} \mid v_{j} \in \mathcal{N}_{i}\right\}\quad\quad\quad (5)$
其中
-
- $m$ 为采样邻居的个数;
- $\mathcal{N}_{i}$ 表示节点 $v_i$ 的一阶邻居(1-hop)集合;
总的来说,
-
- 结构嵌入(structural embedding)关注的是所有邻居信息,因为采用 GCN 获得结构信息,而 GCN 信息传递公式简化为 $AXW$ ;
- 邻居嵌入(neighbor embedding)分别的邻居的某一部分,通过采样部分邻居信息;
因此,他们从不同的角度解释样本,从而将它们放在一起考虑,可能获得它们的互补信息。
2.3 Multiplet loss
对比学习目的:Make positive pairs (i.e., anchor and positive embeddings) close while keeping negative pairs (i.e., anchor and negative embeddings) far apart.
DGI、GMI、MVGRL、GIC 使用双线性层作为判别器来区分正样本对和负样本对,非常耗时间,具体对比可以参考 Figure 6 。
本文三目的:
-
- 使得 positive pairs 更近,保持 negative pairs 更远;
- 减少泛化误差,减小泛化误差对 UGRL 来说也很重要,因为在训练过程中如果泛化误差小,可能会提高对比学习的泛化能力
- 拉近类内距离,加大类间距离;
在 SUGRL中,本文以 Triplet loss 损失为基础,设计一个上限损失来避免使用除鉴别器。拉近类内距离,加大类间距离,具体来说,每个样本的 Triplet loss 可以表示为:
$\alpha+d\left(\mathbf{h}, \mathbf{h}^{+}\right)
-
- $d(.)$ 是相似度测量方法,如 $\ell_{2}$ -norm distance;
- $\alpha$ 是非负值,代表着 positive 和 negative embeddings 的 "安全距离" ;
补充:triplets loss 知识点
公式为:
$L=\max (d(a, p)-d(a, n)+\operatorname{margin}, 0)$
优化目标:拉近 $a$,$p$ 的距离,拉远 $a$,$n$ 的距离
- easy triplets : $L=0$ 即 $d(a, p)+ margin
- hard triplets: $d(a, n)
- semi-hard triplets : $d(a, p)
通过对所有负样本求和,Eq.6 扩展为:
$\mathcal{L}_{\text {triplet }}=\frac{1}{k} \sum_{i=1}^{k}\left\{d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2}-d\left(\mathbf{h}, \mathbf{h}_{i}^{-}\right)^{2}+\alpha\right\}_{+}\quad\quad\quad(7)$
其中
-
- $\{\cdot\}_{+}=\max \{\cdot, 0\} $
- $k$ 代表负样本数
为了增加类间差异,作者拉大正负样本之间的距离,作者在两种 Positive embedding (考虑 Structural information 和 Neighbor information)上执行以下操作:
$\mathcal{L}_{S}=\frac{1}{k} \sum_{i=1}^{k}\left\{d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2}-d\left(\mathbf{h}, \mathbf{h}_{i}^{-}\right)^{2}+\alpha\right\}_{+}\quad\quad\quad(8)$
$\mathcal{L}_{N}=\frac{1}{k} \sum_{j=1}^{k}\left\{d\left(\mathbf{h}, \widetilde{\mathbf{h}}^{+}\right)^{2}-d\left(\mathbf{h}, \mathbf{h}_{j}^{-}\right)^{2}+\alpha\right\}_{+}\quad\quad\quad(9)$
显然上式有两种情况:
-
- Case 1:$d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2} \geq d\left(\mathbf{h}, \widetilde{\mathbf{h}}^{+}\right)^{2}$
- Case 2:$d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2}
对于 Case 1:$d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2} \geq d\left(\mathbf{h}, \widetilde{\mathbf{h}}^{+}\right)^{2} $ ,那么即使 $\text{Eq.9}$ 为 $0$ ,$\text{Eq.8}$ 也不可能小于 $0$ (非负)。此时,我们可以认为 $\mathcal{L}_{S}$ 任然有效,$ \mathcal{L}_{N}$ 是无效的。因此,负嵌入可以通过 $Eq.8$ 推离锚点嵌入,这样类间差异增大。与 Case 1 类似,Case 2 也可以扩大类间差异。
基于以上分析,Case 1 或 Case 2 都可以增大类间差异。特别是,如果其中一种无效,另一种仍将有效地进一步扩大类间差异。因此,$\text{Eq.8}$ 和 $\text{Eq.9}$ 可以从结构嵌入和邻域嵌入中获得互补的信息,从而能够扩大类间差异。
分析 $\text{Eq.7}$:
$\text{Eq.7}$ 需要保证 $d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2}$ 和 $d\left(\mathbf{h}, \mathbf{h}_{i}^{-}\right)^{2}$ 需要比 $\alpha$ 大,但是它忽略了 Anchor embedding 和 Positive embedding 之间的距离。如果 Anchor embedding 和 Positive embedding 直接的距离特别大,那么 $\text{Eq.7}$ 任然非负。在这种情况下,类内的变化可能会很大,但这并不有利于泛化误差的减少。
为了解决这个问题,作者通过以下目标函数研究了负对和正对的上界(即 $\alpha + \beta$):
$\alpha+d\left(\mathbf{h}, \mathbf{h}^{+}\right)
其中:
-
- $\beta$ is a non-negative tuning parameter;
由式 $\alpha+d\left(\mathbf{h}, \mathbf{h}^{+}\right)
由式 $d\left(\mathbf{h}, \mathbf{h}^{-}\right)
对所有负嵌入的损失加和后,提出的减少类内变化的上界损失定义如下:
$\mathcal{L}_{U}=-\frac{1}{k} \sum\limits _{i=1}^{k}\left\{d\left(\mathbf{h}, \mathbf{h}^{+}\right)^{2}-d\left(\mathbf{h}, \mathbf{h}_{i}^{-}\right)^{2}+\alpha+\beta\right\}_{-}\quad\quad\quad(11)$
其中
-
- $\{\cdot\}_{-}=\min \{\cdot, 0\} $
注意到这里并没有考虑 neighbor information ,这是由于
-
- 每种信息都得到了相似的结果;
- 在实验中同时使用它们并不能显著提高模型的性能;
最后,将上述损失整合,可以表示为:
$\mathcal{L}=\omega_{1} \mathcal{L}_{S}+\omega_{2} \mathcal{L}_{N}+\mathcal{L}_{U}\quad\quad\quad(12)$
3 Experiments
3.1 datasets
在实验中,作者使用了8个常用的基准数据集,包括3个引文网络数据集(即 Cora, Citeseer 和 Pubmed ), 2个亚马逊销售数据集(即照片和计算机),3个大型数据集(即 Ogbn-arxiv、Ogbn-mag 和 Ogbn-products )。
3.2 Baseline
对比方法包括 1 种传统算法(即 DeepWalk ), 2种半监督学习算法(即 GCN 和 GAT),以及8种非监督学习算法(即Graph Auto-Encoders (GAE) ,变分图自动编码器(VGAE)、DGI 、GRACE 、GMI 、MVGRL 和 GCA 、GIC 。
3.3 Results
Table 1 和 Table 2 总结了所有方法在 8 个实际图结构数据集上的分类精度和执行时间。
首先,SUGRL在分类精度方面优于所有自监督方法(如DGI、GMI、GRACE、MVGRL、GIC和GCA);例如,作者的方法与最差的方法DGI和最好的比较方法MVGRL相比,平均分别提高了4.0%和1.9%。与学习过程中采用标签信息的半监督方法(GCN和GAT)相比,SUGRL也取得了更好的性能。其次,作者的SUGRL的效率是最好的。在8个数据集上,与其他自监督方法相比,SUGRL分别比最慢的比较方法GMI和最快的比较方法GIC平均快了122.4和4.4。
总之,作者的方法在几乎所有数据集上,在模型性能和执行时间方面,在节点分类方面都优于其他比较方法。原因可以总结如下。首先,SUGRL综合考虑结构信息和邻域信息,生成两种正嵌入及其损失函数,这可以将负嵌入推离锚嵌入更远(即实现较大的类间变异)。其次,SUGRL采用一个上界来保证正埋点和锚埋点之间的距离是有限的(即实现较小的类内变化)。第三,SUGRL去掉了数据增大和鉴别器的步骤,大大减少了训练时间。最后,SUGRL可用于输出低维高质量的嵌入,在保持模型有效性的同时减少训练时间。
3.4 Ablation study
UGRL考虑三种信息,即语义信息、结构信息和邻居信息,生成两种具有对应对比损失的正对(LS和LN)。为了验证框架中各成分的有效性,作者分别研究了结构信息、邻居信息和上界的有效性,以及对比损失中各成分的有效性。
类内和类间差异之比的有效性。考虑到类内和类间变化的大小不同,将比率归一化为 $[0,1]$,在图3的数据集Photo上报告类内与类间变化的比率。首先,与本文方法相比,没有结构或邻居信息的方法通常输出更大的比率(即更小的类间变异)。二是无上界的方法,与作者的方法相比,还会输出更大的比率(即更大的类内变量)。这样就可以验证结构信息、邻居信息或上界信息的有效性。
4 Conclusion
在本文中,作者设计了一个简单的框架,即简单无监督图表示学习(SUGRL),以实现有效和高效的对比学习。为了获得有效性,作者设计了两个三重组损失函数来探索结构信息和邻居信息之间的互补信息,以扩大类间的变化,以及一个上限损失来减少类内的变化。为了提高算法的效率,作者设计了一种去除GCN的生成锚点和负嵌入的方法,并从之前的图对比学习中去除数据增强和鉴别器。作者对各种真实世界的数据集进行了全面的实验,实验结果表明,作者的方法在准确性和可扩展性方面都优于目前最先进的方法。
论文:
GMI——Graph Representation Learning via Graphical Mutual Information Maximization
DGI——Deep Graph Infomax
GCA——Graph Contrastive Learning with Adaptive Augmentation
MVGRL
CPC——Contrastive Predictive Coding
CMC——Contrastive Multiview Coding
GRACE
GCA
GIC