GAT-GRAPH ATTENTION NETWORKS(论文笔记)

CCF等级:A

发布时间:2018年

代码位置

25年4月21日交

目录

一、简介

二、原理

1.注意力系数

2.归一化

3.特征组合与非线性变换

4.多头注意力

4.1特征拼接操作

4.2平均池化操作

三、实验性能

四、结论和未来工作


一、简介

图注意力网络(GATs),可以在图结构数据上操作的神经网络架构。在GAT模型中,节点能够在它们的邻域特征上进行"关注",从而隐式地为邻域中的不同节点指定不同的权重。简单理解为节点能够知道关联的每个节点对自己的影响权重是多少。


二、原理

首先将节点转化为向量,然后通过自我注意机制计算每个节点对其邻居的重要性权重(即注意力系数),并通过softmax函数进行归一化。接下来,利用这些归一化的注意力系数对邻近节点的特征加权求和,得到每个节点的新特征表示,并通常应用非线性激活函数进一步处理。为了增强模型的表现力和稳定性,GAT可以采用多头注意力机制独立执行上述过程多次,并将结果拼接或平均,最终生成用于分类或其他任务的节点特征表示。

1.注意力系数

先将节点转化为向量,节点的向量为。(这一部分不重要,就不展开讲解)

  • 分别是节点和节点的特征向量。
  • 是一个权重矩阵,用于线性变换这些特征向量。
  • 是一个注意力机制函数,用于计算两个变换后的特征向量之间的相似度或相关性。

使用权重矩阵对节点进行线性变换,再通过注意力机制函数计算这两个变换后特征向量之间的注意力分数

节点对其邻居节点的注意力系数

注意:只衡量了节点关注节点的程度,而不是同时考虑关注程度。换句话说是独立计算的,它们分别代表了从节点到节点和从节点到节点的注意力权重,这允许模型对每个方向的重要性进行不同的赋值。

2.归一化

  • 是注意力分数的指数函数值。
  • 是节点所有邻居节点的注意力分数的指数函数值之和。
  • 表示节点的邻居节点集合。

通过函数,每个节点对其邻居节点的注意力权重被归一化到区间内,并且所有邻居节点的注意力权重之和为 1。归一化不仅提高了模型性能,还增强了模型的鲁棒性和解释性。

计算注意力系数和归一化的公式可以合并为

3.特征组合与非线性变换

利用标准化后的注意力系数,对邻近节点的特征进行加权求和,从而得到每个节点的新特征表示。

  • 是节点更新后的特征向量。
  • 表示经过归一化后,邻居节点对节点的注意力权重。
  • 是一个可学习的权重矩阵,用于对输入特征进行线性变换。
  • 表示节点的邻居集合。
  • 是一个非线性激活函数,例如,用于引入非线性。

通过上述步骤,图注意力网络(GATs)能够动态地调整每个节点对其邻居节点的关注度,并据此更新自身的特征表示,以捕捉更丰富的局部结构信息。

4.多头注意力

通过为每个注意力头使用不同的权重矩阵,模型可以捕捉到不同类型的特征关系。

类似于卷积神经网络中使用不同的卷积核计算。

多头注意力机制 类似于多尺度特征提取,不同尺度的特征通过不同的卷积核提取。

多头注意力机制来计算节点的新特征表示

4.1特征拼接操作

  • 表示将个注意力头的结果进行拼接,形成最终的特征表示。
  • 是第个注意力头对应的权重矩阵,用于对邻居节点的特征向量进行线性变换。
  • 表示对所有邻居节点的特征向量进行加权求和。这里的是节点的邻居节点集合。

4.2平均池化操作

  • 表示对个注意力头的结果进行平均池化,即将个注意力头的结果相加后除以

总结:

使用单一的注意力权重和权重矩阵,适用于简单的注意力机制。

使用多头注意力机制,特征拼接操作,并通过特征拼接操作组合多个注意力头的结果,从而增强模型的表达能力。

使用多头注意力机制,平均池化操作,并通过平均池化操作组合多个注意力头的结果,从而增强模型的表达能力。


三、实验性能

Cora、Citeseer和Pubmed分类准确率的结果总结。GCN-64 *对应于计算64个隐藏特征的最佳GCN结果。

对于PPI数据集,以微平均F1分数表示的结果摘要。GraphSAGE*对应于我们仅通过修改其架构即可获得的最佳GraphSAGE结果。


四、结论和未来工作

1.**解决稀疏矩阵运算限制的问题:**使用的张量操作框架仅支持秩2张量的稀疏矩阵乘法,这限制了批次处理的能力。

2.扩展模型深度:采用跳连接等技术可以适当扩展模型的深度,从而提高其表达能力。

相关推荐
WSSWWWSSW3 小时前
Seaborn数据可视化实战:Seaborn数据可视化基础-从内置数据集到外部数据集的应用
python·信息可视化·数据分析·matplotlib·seaborn
Small___ming3 小时前
Matplotlib 可视化大师系列(七):专属篇 - 绘制误差线、等高线与更多特殊图表
python·信息可视化·matplotlib
荼蘼5 小时前
CUDA安装,pytorch库安装
人工智能·pytorch·python
杨荧6 小时前
基于Python的农作物病虫害防治网站 Python+Django+Vue.js
大数据·前端·vue.js·爬虫·python
骑驴看星星a7 小时前
数学建模--Topsis(Python)
开发语言·python·学习·数学建模
学习3人组7 小时前
JupyterLab在线调试实验室
python
ygy.白茶8 小时前
线性回归入门级
人工智能·python·机器学习
刘同学Python学习日记9 小时前
Python 办公自动化实战:Excel 批量处理 + 自动发邮件
python
mit6.8249 小时前
[Vid-LLM] 功能分类体系 | 视频如何被“观看“ | LLM的主要作用
人工智能·python
青铜发条9 小时前
【python】python进阶——logging日志模块
python