近日，一个关于图注意力网络可视化的项目吸引了大批研究人员的兴趣，上线仅仅一天，收获 200+ 星。该项目是关于用 PyTorch 实现的图注意力网络(GAT)，包括易于理解的可视化。

项目地址：https://g微软ithub.com/gordicaleksa/pytorch-GAT

在正式介绍项目之前，先提一下图神经网络（GNN）。GNN 是一类基于深度学习的处理图域信息的方法。由于其较好的性能和可解释性，GNN 最近已成为工程一种广泛应用的图分析方法。现已广泛应用于计算生物学、计算药理学、推荐系统等。

GNN 把深度学习应用到图结构 (Graph) 中，其中的图卷积网络 GCN 可以在 Graph 上进行卷积操作，但是 GCN 存在一些缺陷。因此，Ben师用gio 团队在三年前提出了图注意力网络(GAT，Graph Attention Network)，旨在解决 GCN 存在的问题。

GAT 是空间（卷积）GNN 的代表。由于 CNNs 在计算机视觉领域取得了巨大的成功，研究人Py员决定将其推广到图形上，因此 GAT 应运而生。

现在，有人用 PyTorch 实现了 GAT 可视化。我们来看看该项目是如何实现的。

可视化

Cora 可视化

说到 GNN，就不得不介绍一下 Cora 数据集。Cora 数据集由许多机To器学习论文组成，是近年来图深度学习很喜欢使用的数据集。Cora 中的节点代表研究论文，链接是这些论文之间的引用。项目作者添加了一个用于可视化 Cora 和进行基本网络分析的实用程序。Cora 如下图所示：

节点大rc小对应于其等级（即进出边的数量）。边的粗细大致对应于边的「popular」或「连接」程度。以下是显示 Cora 上等级（进出边的数量）分布的图：

进和出的等级图是一样的，因为处理的是无向图。在底部的图（等级分布h实）上，我们可以看到一个有趣的峰值发生在 [2，4] 范围内。这意味着多数节点有少量的边，但是有 1 个节点有 169 条边（绿色大节点）。

注意力可视化

有了一个训练好的 GAT 模型以后，我们就可以将某些节点所学的注意力可现图视化。节点利用注意力来决定如何聚合周围的节点，如下图所示：

这是 Cora 节点中边数最多的节点之一(引用)。颜色表示同一类的节点。

熵直方图

另一种理解 GAT 没有在 Cora 上学习注意力模式 (即它在学习常量注意力) 的方注意法是，将节点邻域的注意力权重视为概率分布，计算熵，并在每个节点邻域积累信息。

我们希望 GAT 的注意力分布有偏差。你可以看到橙色的直方图是理想均匀分布的样子，而浅蓝色的是学习后的分布，它们是完全一样的。

分析 C力网ora 嵌入空间 (t-SNE)

GAT 的输出张量为 shape=（2708，7），其中 2708 是 Cora 中的节点数，7 是类数。用 t-SNE 把这些 7 维向量投影成 2D，得到：

使用方法

方法 1：Jupyter Notebook

只络，需从 Anaconda 控制台运行 Jupyter Notebook，它将在你的默认浏览器中打开 session。打开 The Annotated GAT.ipynb 即可开始。

注意，如果你得到了 DLL load failed whil可视e importing win32api: The specified module could not be found，只需要 pip uninstall pywin32，或者 pip install pywin32、onda instal化效l pywin32。

方法 2：使用你选择的 IDE

如果使用自己选择的 IDE，只需要将 Python 环境和设置部分连接起来。

• ·标签: 九号公司 新国标 电动自行车
• ·爆改漫画师？OpenAI 发布 ChatGPT Images 2.0，单次可生成8张连贯图像
• ·医学教育革命来袭！复旦中山医院推出AI教学大模型，培养有温度的未来医生
• ·人脸识别时一定要穿衣服引热议，专家称不必过分担心
• ·标签: 千里科技 汽车 AI
• ·AI技术被滥用成“退款神器”，商家无奈：假图太逼真，有苦说不出
• ·OpenAI 呼吁美国加大电力投资 助力 AI 领域领跑中国
• ·AI跳舞哪家强？谷歌3D舞者闻歌起舞，挑战DanceNet
• ·云计算核心技术Docker教程：清理未使用的Docker网络
• ·标签: 索尼 PlayStation 5 羊蹄山之魂 Bravia
• ·标签: 追觅科技 俞浩 围脖侠 清朗行动
• ·​OpenAI 推出家长控制功能，强化 ChatGPT 安全性
• ·标签: 本田汽车 电动汽车 电气化战略 日本车企
• ·标签: 腾讯控股 华纳兄弟探索 派拉蒙 天舞影业
• ·Twitter将允许用户直接把推文分享到Snapchat和Instagram
• ·标签: 滴滴租车