论文信息

论文标题:Graph Contrastive Learning with Adaptive Augmentation
论文作者:Yanqiao Zhu、Yichen Xu3、Feng Yu4、Qiang Liu、Shu Wu、Liang Wang
论文来源:2021, WWW
论文地址:download
论文代码:download

摘要

  近年来,对比学习(CL)已成为一种成功的无监督图表示学习方法。大多数图CL方法首先对输入图进行随机增强,以获得两个图视图,并最大化两个视图中表示的一致性。尽管图CL方法得到了繁荣的发展,但图增强方案的设计——CL中的一个关键组成部分——仍然很少被探索。我们认为,数据增强方案应该保留图的内在结构和属性,这将迫使模型学习对不重要的节点和边缘的扰动不敏感的表示。然而,现有的方法大多采用统一的数据增强方案,如统一降边和统一变换特征,导致性能次优。在本文中,我们提出了一种新的具有自适应增强的图对比表示学习方法,该方法包含了图的拓扑和语义方面的各种先验。具体来说,在拓扑层面上,我们设计了基于节点中心性度量的增强方案来突出重要的连接结构。在节点属性级别上,我们通过向不重要的节点特征添加更多的噪声来破坏节点特征,以强制模型识别底层的语义信息。我们在各种真实世界的数据集上进行了广泛的节点分类实验。实验结果表明,我们提出的方法始终优于现有的先进基线,甚至超过一些监督的方法,这验证了所提出的对比框架自适应增强的有效性。

1-介绍

  出发角度:倾向于保持重要的结构和属性不变,同时干扰可能不重要的边连接和特征。

  自适应数据增强方面:

    • 拓扑结构:基于节点中心性度量,突出重要连接;
    • 语义信息:对不重要的节点属性添加噪声;

2 Method

2.1 Framework

  框架如下:

    

 

  形如 GRACE 架构。

  算法流程:

   

  编码器

    GCi(X,A)=σ(D^12A^D^12XWi)(12)f(X,A)=GC2(GC1(X,A),A)(13)

  损失函数

    J=12Ni=1N[(ui,vi)+(vi,ui)](2)

  其中:

    logeθ(ui,vi)/τeθ(ui,vi)/τpositive pair +kieθ(ui,vk)/τinter-view negative pairs +kieθ(ui,uk)/τintra-view negative pairs (1)

2.2 Adaptive Graph Augmentation

2.2.1 Topology-level augmentation

  利用 Eq.3 中的概率从原始边集合中采样一个边子集合

    P{(u,v)E~}=1puve(3)

  其中:

    • (u,v)E
    • puve 是删除边 (u,v) 的概率;
    • E~ 将作为生成视图的边集合;

  分析知: puve 应该反映边 (u,v) 的重要性,目的是大概率删除不重要的边,同时保留增强视图中重要的边。

  节点中心性量化了节点的重要性,本文为边 (u,v) 定义边中心性 wuve,用于衡量边(u,v) 对两个相连节点的影响。给定节点中心性度量 φc():VR+,将边中心性定义为两个相邻节点中心性得分的均值,即 wuve=(φc(u)+φc(v))/2。在有向图上,只使用尾部节点的中心性,即 wuve=φc(v),因为边的重要性通常是它们指向的节点。

  接下来,根据每条边的中心性值来计算它的概率。由于采用度作为节点中心性这种度量方法在不同数量级上变化差别过大,所以本文首先设置 suve=logwuve 以缓解具有高度密集连接的节点的影响。然后通过将边中心性的值转换为概率:

    puve=min(smaxesuvesmaxeμsepe,pτ)(4)

  其中,pe 是一个控制去除边的总体概率的超参数,smaxeμsesuve 的最大值和平均值。而 pτ<1 是一个临界概率(cut-off probability),对于边中心性低(高概率删除)的边,采用 pτ 删除,对于边中心性低的边(低概率删除),采用 smaxesuvesmaxeμsepe  删除。

  这里提供三种 节点中心性度量 方法:

  1、点度中心性(Degree centrality):节点度本身可以是一个中心性度量。在有向网络上,使用内度,因为有向图中的一个节点的影响主要是由指向它的节点赋予的。  

  2、特征向量中心性(Eigenvector centrality):基本思想是一个节点的中心性是相邻节点中心性的函数。也就是说,与你连接的人越重要,你也就越重要。

  3、PageRank中心性(PageRank centrality):基于有向图

  对于 PageRank 中心性分数计算公式如下:

    σ=αAD1σ+1(5)

  其中,σRN 是每个节点的 PageRank中心性得分的向量,α 是一个阻尼因子,它可以防止图中的 sinks 从连接到它们的节点中吸收所有 ranks。这里设置α=0.85。对于无向图,我们对转换后的有向图执行PageRank,其中每条无向边都被转换为两条有向边。

  例子:

  

  从图中可以看出,三种方案存在细微差别,但都强调了连接两个教练(橙色节点)的边,而较少关注边缘节点。

2.2.2 Node-attribute-level augmentation

  节点特征隐藏:

    X~=[x1m~;x2m~;;xNm~]

  其中:m~iBern(1pif),即用 1pif 的概率取 1,用 pif 的概率取 0

  这里 pif 应该反映出节点特征的第 i 个维数的重要性。我们假设经常出现在有影响的节点中的特征维度应该是重要的,并定义特征维度的权重如下。

  对于稀疏的 one-hot 节点特征,即 xui{0,1},对于任何节点 u 和特征维 i,我们计算维度 i 的权重为

    wif=uVxuiφc(u)(7)

  其中,

    • φc() 是一个用于量化节点重要性的节点中心性度量;
    • 第一项 xui{0,1} 表示节点 u 中维度 i 的出现;
    • 第二项 φi(u) 表示每次出现的节点重要性;

  对于稠密、连续的节点特征 xu,本文用绝对值  |xui|  来测量节点  u  的  i  维的特征的值的大小:

    wif=uV|xui|φc(u)(8)

  与 Topology-level augmentation 类似,我们对权值进行归一化,以获得表示特征重要性的概率。形式上:

    pif=min(smaxfsifsmaxμsfpf,pτ)(9)

  其中,sif=logwifsmaxf  和 μsf 分别为 sif 的最大值和平均值, pf 是控制特征增强的总体幅度的超参数。 

3 Experiment

数据集

  

  【 Wiki-CSAmazon-ComputersAmazon-PhotoCoauthor-CSCoauthor-Physics  】

节点分类

  基线实验:

     

消融实验

   

灵敏度分析:

    

4 Conclusion

  开发了一种自适应数据增强对比学习框架。 

修改历史

2021-04-12 创建文章
2022-06-14 精读

编码器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from typing import Optional
import torch
from torch import nn
import torch.nn.functional as F
from torch_geometric.nn import GCNConv
 
class Encoder(nn.Module):
    def __init__(self, in_channels: int, out_channels: int, activation, base_model=GCNConv, k: int = 2, skip=False):
        super(Encoder, self).__init__()
        self.base_model = base_model
        assert k >= 2
        self.k = k
        self.skip = skip
        if not self.skip:
            self.conv = [base_model(in_channels, 2 * out_channels).jittable()]
            for _ in range(1, k - 1):
                self.conv.append(base_model(2 * out_channels, 2 * out_channels))
            self.conv.append(base_model(2 * out_channels, out_channels))
            self.conv = nn.ModuleList(self.conv)
            self.activation = activation
        else:
            self.fc_skip = nn.Linear(in_channels, out_channels)
            self.conv = [base_model(in_channels, out_channels)]
            for _ in range(1, k):
                self.conv.append(base_model(out_channels, out_channels))
            self.conv = nn.ModuleList(self.conv)
 
            self.activation = activation
 
    def forward(self, x: torch.Tensor, edge_index: torch.Tensor):
        if not self.skip:
            for i in range(self.k):
                x = self.activation(self.conv[i](x, edge_index))
            return x
        else:
            h = self.activation(self.conv[0](x, edge_index))
            hs = [self.fc_skip(x), h]
            for i in range(1, self.k):
                u = sum(hs)
                hs.append(self.activation(self.conv[i](u, edge_index)))
            return hs[-1]

节点度重要性

1
2
3
4
5
6
7
8
from torch_geometric.utils import degree, to_undirected
def degree_drop_weights(edge_index):
    edge_index_ = to_undirected(edge_index)
    deg = degree(edge_index_[1])    # 转换成无向图之后,计算节点的度
    deg_col = deg[edge_index[1]].to(torch.float32)   # 计算节点的入度,入度越大,则节点越重要
    s_col = torch.log(deg_col)
    weights = (s_col.max() - s_col) / (s_col.max() - s_col.mean())
    return weights

PageRank  重要性

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from torch_scatter import scatter
def compute_pr(edge_index, damp: float = 0.85, k: int = 10):
    num_nodes = edge_index.max().item() + 1
    deg_out = degree(edge_index[0])
    x = torch.ones((num_nodes, )).to(edge_index.device).to(torch.float32)
    for i in range(k):
        edge_msg = x[edge_index[0]] / deg_out[edge_index[0]]
        agg_msg = scatter(edge_msg, edge_index[1], reduce='sum')
        x = (1 - damp) * x + damp * agg_msg
    return x
 
def pr_drop_weights(edge_index, aggr: str = 'sink', k: int = 10):
    pv = compute_pr(edge_index, k=k)
    pv_row = pv[edge_index[0]].to(torch.float32)
    pv_col = pv[edge_index[1]].to(torch.float32)
    s_row = torch.log(pv_row)
    s_col = torch.log(pv_col)
    if aggr == 'sink':
        s = s_col
    elif aggr == 'source':
        s = s_row
    elif aggr == 'mean':
        s = (s_col + s_row) * 0.5
    else:
        s = s_col
    weights = (s.max() - s) / (s.max() - s.mean())
    return weights
 
pr_drop_weights(data.edge_index, aggr='sink', k=200).to(device)

特征向量 重要性

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from torch_geometric.utils import to_networkx
import networkx as nx
def eigenvector_centrality(data):
    graph = to_networkx(data)
    x = nx.eigenvector_centrality_numpy(graph)
    x = [x[i] for i in range(data.num_nodes)]
    return torch.tensor(x, dtype=torch.float32).to(data.edge_index.device)
def evc_drop_weights(data):
    evc = eigenvector_centrality(data)
    evc = evc.where(evc > 0, torch.zeros_like(evc))
    evc = evc + 1e-8
    s = evc.log()
    edge_index = data.edge_index
    s_row, s_col = s[edge_index[0]], s[edge_index[1]]
    s = s_col
    return (s.max() - s) / (s.max() - s.mean())
 
drop_weights = evc_drop_weights(data).to(device)

边丢弃自适应数据增强

1
2
3
4
5
def drop_edge_weighted(edge_index, edge_weights, p: float, threshold: float = 1.):
    edge_weights = edge_weights / edge_weights.mean() * p
    edge_weights = edge_weights.where(edge_weights < threshold, torch.ones_like(edge_weights) * threshold)
    sel_mask = torch.bernoulli(1. - edge_weights).to(torch.bool)
    return edge_index[:, sel_mask]节点

节点丢弃自适应数据增强

1
2
3
4
5
6
7
8
def drop_feature_weighted(x, w, p: float, threshold: float = 0.7):
    w = w / w.mean() * p
    w = w.where(w < threshold, torch.ones_like(w) * threshold)
    drop_prob = w.repeat(x.size(0)).view(x.size(0), -1)
    drop_mask = torch.bernoulli(drop_prob).to(torch.bool)
    x = x.clone()
    x[drop_mask] = 0.
    return x

特征丢弃自适应数据增强

1
2
3
4
5
6
7
8
9
10
11
12
13
def feature_drop_weights(x, node_c):
    x = x.to(torch.bool).to(torch.float32)
    w = x.t() @ node_c
    w = w.log()
    s = (w.max() - w) / (w.max() - w.mean())
    return s
 
def feature_drop_weights_dense(x, node_c):
    x = x.abs()
    w = x.t() @ node_c
    w = w.log()
    s = (w.max() - w) / (w.max() - w.mean())
    return s
1
2
3
4
5
6
7
8
def drop_feature_weighted_2(x, w, p: float, threshold: float = 0.7):
    w = w / w.mean() * p
    w = w.where(w < threshold, torch.ones_like(w) * threshold)
    drop_prob = w
    drop_mask = torch.bernoulli(drop_prob).to(torch.bool)
    x = x.clone()
    x[:, drop_mask] = 0.
    return x

  

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

  

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 


论文解读目录

posted @   别关注我了,私信我吧  阅读(960)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
· spring官宣接入deepseek,真的太香了~
Live2D
点击右上角即可分享
微信分享提示