基于PyTorch实现EdgeCNN的实战教程

更新时间：2023年02月20日 15:42:10 作者：༺࿈ 海洋༒之心 ࿈༻

本文我们将使用PyTorch来简易实现一个EdgeCNN，不使用PyG库，让新手可以理解如何PyTorch来搭建一个简易的图网络实例demo，感兴趣的朋友跟随小编一起看看吧

前言

大家好，我是阿光。

本专栏整理了《图神经网络代码实战》，内包含了不同图神经网络的相关代码实现（PyG以及自实现），理论与实践相结合，如GCN、GAT、GraphSAGE等经典图网络，每一个代码实例都附带有完整的代码。

正在更新中~ ✨

在这里插入图片描述

🚨 我的项目环境：

平台：Windows10
语言环境：python3.7
编译器：PyCharm
PyTorch版本：1.11.0
PyG版本：2.1.0

💥 项目专栏：【图神经网络代码实战目录】

本文我们将使用PyTorch来简易实现一个EdgeCNN，不使用PyG库，让新手可以理解如何PyTorch来搭建一个简易的图网络实例demo。

一、导入相关库

本项目是采用自己实现的EdgeCNN，并没有使用 PyG 库，原因是为了帮助新手朋友们能够对EdgeConv的原理有个更深刻的理解，如果熟悉之后可以尝试使用PyG库直接调用 EdgeConv 这个图层即可。

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch_geometric.utils import scatter
from torch_geometric.datasets import Planetoid

二、加载Cora数据集

本文使用的数据集是比较经典的Cora数据集，它是一个根据科学论文之间相互引用关系而构建的Graph数据集合，论文分为7类，共2708篇。

Genetic_Algorithms
Neural_Networks
Probabilistic_Methods
Reinforcement_Learning
Rule_Learning
Theory

这个数据集是一个用于图节点分类的任务，数据集中只有一张图，这张图中含有2708个节点，10556条边，每个节点的特征维度为1433。

# 1.加载Cora数据集
dataset = Planetoid(root='./data/Cora', name='Cora')

三、定义EdgeCNN网络

3.1 定义EdgeConv层

这里我们就不重点介绍EdgeCNN网络了，相信大家能够掌握基本原理，本文我们使用的是PyTorch定义网络层。

对于EdgeConv的常用参数：

nn：进行节点特征转换使用的 MLP网络，需要自己定义传入
aggr：聚合邻居节点特征时采用的方式，默认为 max

我们在实现时也是考虑这几个常见参数

对于EdgeConv的传播公式为：

上式子中的 x i x_i xi 代表中心节点特征信息， x j x_j xj 代表邻居节点的特征信息，对于 h θ h_{\theta} hθ 代表每个 EdgeConv 层的可学习参数，也就是对应传入的MLP层中的可学习参数。

所以我们的任务无非就是获取这几个变量，然后进行传播计算即可

3.1.1 特征拼接

该环节实现的公式为： x i ∣ ∣ x j − x i x_i||x_j-x_i xi∣∣xj−xi，对于这个公式来说，我们要获得两个变量，一个是中心节点 x i x_i xi（target）的特征信息，一个是邻居节点 x j x_j xj（source）的特征信息。

对于这两个变量的获取很容易，利用 edge_index 就可以提取出来，edge_index 中保存的是每一条边的一对起始节点与终止节点，对于起始节点可以认为就是 i，对于终止节点就可以认为是 j，然后我们就会获得两个向量，分别为 row 和 col ，这两个向量就是起始顶点和终止顶点的集合。

然后我们在根据索引进行提取特征，利用 x_i = x[row] 和 x_j = x[col] 就可以将中心节点和终止节点对应的特征获取，维度为【E，feature_size】。

然后就可以按照公式实现做差然后与中心节点的特征进行拼接，获得拼接后的特征维度为原来的2倍。

row, col = edge_index # 获取target、source节点索引 [E]
x_i = x[row] # 获取target节点信息，中心节点 [E, feature_size]
x_j = x[col] # 获取source节点信息，邻居节点 [E, feature_size]

x_cat = torch.cat([x_i, x_j - x_i], dim=1) # 拼接特征 [E, 2 * feature_size]

对于这里 x_i 和 x_j 以及起始节点的索引初学可能混淆，所以多多打印中间结果一步一步调试进行理解。

3.1.2 max聚合

对于 EdgeConv 的默认聚合方式为 max，其实还可以使用 mean 、sum 等排列不变函数进行聚合。

对于聚合操作就是公式中求和符号那里，只不过框架给的公式是 sum ，对于聚合我们希望做的是将中心节点的邻居特征按照指定的聚合方式进行聚合。

我们可以利用 PyG 工具库中提供的 scatter 函数进行操作，该函数可以指定聚合方式以及聚合维度等参数，使用方法就是需要传入需要聚合的 Tensor ，此外还需要传入一个 index ，指明哪些向量为同一个邻居的节点，举个例子，我们传入的 index=[0,0,0,1,1] ，这就代表第一个、第二个、第三个为同一节点的邻居，所以就会将待聚合的 Tensor 的第一个向量、第二个向量、第三个向量按照指定聚合方式进行聚合。

这里说的有点抽象，自己尝试一个简单示例就明白了。

out = scatter(src=x_cat, index=row, dim=0, reduce='max') # max聚合操作 [num_nodes, feature_size]

3.1.3 特征映射

在公式中有个 h θ h_{\theta} hθ，这个就代表 MLP 做特征映射做的，对于官方给的 EdgeConv 需要我们手动传入 MLP 模型，所以本项目自实现也是按照这种方式，MLP 的操作在 EdgeConv 中并没有实现，而是利用传入的模型进行操作。

这里注意一点就是定义的 MLP 模型的输入维度应该为原始维度的2倍，因为我们在这之前进行了特征拼接操作，所以特征维度进行了加倍。

out = self.mlp(out) # 特征映射 [num_nodes, out_channels]

3.1.4 EdgeConv层

接下来就可以定义EdgeConv层了，该层实现了1个函数，为 forward()

forward()：这个函数定义模型的传播过程，也就是上面公式的 x i ′ = ∑ j ∈ N ( i ) h θ ( x i ∣ ∣ x j − x i ) x_i'=\sum_{j\in N(i)}h_{\theta}(x_i||x_j-x_i) xi′=∑j∈N(i)hθ(xi∣∣xj−xi)

# 2.定义EdgeConv层
class EdgeConv(nn.Module):
    def __init__(self, nn, aggr='max'):
        super(EdgeConv, self).__init__()
        self.mlp = nn # MLP网络
    
    def forward(self, x, edge_index):
        row, col = edge_index # 获取target、source节点索引 [E]
        x_i = x[row] # 获取target节点信息，中心节点 [E, feature_size]
        x_j = x[col] # 获取source节点信息，邻居节点 [E, feature_size]
        
        x_cat = torch.cat([x_i, x_j - x_i], dim=1) # 拼接特征 [E, 2 * feature_size]
        out = scatter(src=x_cat, index=row, dim=0, reduce='max') # max聚合操作 [num_nodes, feature_size]
        
        out = self.mlp(out) # 特征映射 [num_nodes, out_channels]
        
        return out

对于我们实现这个网络的实现效率上来讲比PyG框架内置的 EdgeConv 层稍差一点，因为我们是按照公式来一步一步利用矩阵计算得到，没有对矩阵计算以及算法进行优化，不然初学者可能看不太懂，不利于理解EdgeConv公式的传播过程，有能力的小伙伴可以看下官方源码学习一下，框架内是按照消息传递方式实现的。

3.2 定义EdgeCNN网络

上面我们已经实现好了 EdgeConv 的网络层，之后就可以调用这个层来搭建 EdgeCNN 网络。

# 3.定义EdgeConv网络
class EdgeCNN(nn.Module):
    def __init__(self, num_node_features, num_classes):
        super(EdgeCNN, self).__init__()
        self.conv1 = EdgeConv(nn=nn.Linear(2 * num_node_features, 16), aggr='max')
        self.conv2 = EdgeConv(nn=nn.Linear(2 * 16, num_classes), aggr='max')
        
    def forward(self, data):
        x, edge_index = data.x, data.edge_index
        
        x = self.conv1(x, edge_index)
        x = F.relu(x)
        x = F.dropout(x, training=self.training)
        x = self.conv2(x, edge_index)
        
        return F.log_softmax(x, dim=1)

上面网络我们定义了两个EdgeConv层，第一层的参数的输入维度就是初始每个节点的特征维度 * 2，输出维度是16。

第二个层的输入维度为16 * 2，输出维度为分类个数，因为我们需要对每个节点进行分类，最终加上softmax操作。

这里说明一下为什么要将输入乘以2，原因是在使用MLP进行特征转换之前，会将中心节点的特征与中心节点和邻居节点的差向量做拼接，所以得到的输出维度为节点的特征维度 * 2。

四、定义模型

下面就是定义了一些模型需要的参数，像学习率、迭代次数这些超参数，然后是模型的定义以及优化器及损失函数的定义，和pytorch定义网络是一样的。

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 设备
epochs = 10 # 学习轮数
lr = 0.003 # 学习率
num_node_features = dataset.num_node_features # 每个节点的特征数
num_classes = dataset.num_classes # 每个节点的类别数
data = dataset[0].to(device) # Cora的一张图

# 3.定义模型
model = EdgeCNN(num_node_features, num_classes).to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=lr) # 优化器
loss_function = nn.NLLLoss() # 损失函数

五、模型训练

模型训练部分也是和pytorch定义网络一样，因为都是需要经过前向传播、反向传播这些过程，对于损失、精度这些指标可以自己添加。

# 训练模式
model.train()

for epoch in range(epochs):
    optimizer.zero_grad()
    pred = model(data)
    
    loss = loss_function(pred[data.train_mask], data.y[data.train_mask]) # 损失
    correct_count_train = pred.argmax(axis=1)[data.train_mask].eq(data.y[data.train_mask]).sum().item() # epoch正确分类数目
    acc_train = correct_count_train / data.train_mask.sum().item() # epoch训练精度
    
    loss.backward()
    optimizer.step()
    
    if epoch % 20 == 0:
        print("【EPOCH: 】%s" % str(epoch + 1))
        print('训练损失为：{:.4f}'.format(loss.item()), '训练精度为：{:.4f}'.format(acc_train))

print('【Finished Training！】')

六、模型验证

下面就是模型验证阶段，在训练时我们是只使用了训练集，测试的时候我们使用的是测试集，注意这和传统网络测试不太一样，在图像分类一些经典任务中，我们是把数据集分成了两份，分别是训练集、测试集，但是在Cora这个数据集中并没有这样，它区分训练集还是测试集使用的是掩码机制，就是定义了一个和节点长度相同纬度的数组，该数组的每个位置为True或者False，标记着是否使用该节点的数据进行训练。

# 模型验证
model.eval()
pred = model(data)

# 训练集（使用了掩码）
correct_count_train = pred.argmax(axis=1)[data.train_mask].eq(data.y[data.train_mask]).sum().item()
acc_train = correct_count_train / data.train_mask.sum().item()
loss_train = loss_function(pred[data.train_mask], data.y[data.train_mask]).item()

# 测试集
correct_count_test = pred.argmax(axis=1)[data.test_mask].eq(data.y[data.test_mask]).sum().item()
acc_test = correct_count_test / data.test_mask.sum().item()
loss_test = loss_function(pred[data.test_mask], data.y[data.test_mask]).item()

print('Train Accuracy: {:.4f}'.format(acc_train), 'Train Loss: {:.4f}'.format(loss_train))
print('Test  Accuracy: {:.4f}'.format(acc_test), 'Test  Loss: {:.4f}'.format(loss_test))

七、结果

【EPOCH: 】1
训练损失为：1.9629 训练精度为：0.1214
【EPOCH: 】21
训练损失为：1.6709 训练精度为：0.5714
【EPOCH: 】41
训练损失为：1.3965 训练精度为：0.7571
【EPOCH: 】61
训练损失为：1.1095 训练精度为：0.8643
【EPOCH: 】81
训练损失为：0.9088 训练精度为：0.9286
【EPOCH: 】101
训练损失为：0.7454 训练精度为：0.9643
【EPOCH: 】121
训练损失为：0.5841 训练精度为：0.9643
【EPOCH: 】141
训练损失为：0.4985 训练精度为：0.9714
【EPOCH: 】161
训练损失为：0.3954 训练精度为：0.9714
【EPOCH: 】181
训练损失为：0.3339 训练精度为：0.9857
【Finished Training！】

>>>Train Accuracy: 1.0000 Train Loss: 0.3133
>>>Test Accuracy: 0.4230 Test Loss: 1.6562

训练集	测试集
Accuracy	1.0000	0.4230
Loss	0.3133	1.6562

完整代码：

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch_geometric.utils import scatter
from torch_geometric.datasets import Planetoid

# 1.加载Cora数据集
dataset = Planetoid(root='./data/Cora', name='Cora')

# 2.定义EdgeConv层
class EdgeConv(nn.Module):
    def __init__(self, nn, aggr='max'):
        super(EdgeConv, self).__init__()
        self.mlp = nn # MLP网络
    
    def forward(self, x, edge_index):
        row, col = edge_index # 获取target、source节点索引 [E]
        x_i = x[row] # 获取target节点信息，中心节点 [E, feature_size]
        x_j = x[col] # 获取source节点信息，邻居节点 [E, feature_size]
        
        x_cat = torch.cat([x_i, x_j - x_i], dim=1) # 拼接特征 [E, 2 * feature_size]
        out = scatter(src=x_cat, index=row, dim=0, reduce='max') # max聚合操作 [num_nodes, feature_size]
        
        out = self.mlp(out) # 特征映射 [num_nodes, out_channels]
        
        return out

# 3.定义EdgeConv网络
class EdgeCNN(nn.Module):
    def __init__(self, num_node_features, num_classes):
        super(EdgeCNN, self).__init__()
        self.conv1 = EdgeConv(nn=nn.Linear(2 * num_node_features, 16), aggr='max')
        self.conv2 = EdgeConv(nn=nn.Linear(2 * 16, num_classes), aggr='max')
        
    def forward(self, data):
        x, edge_index = data.x, data.edge_index
        
        x = self.conv1(x, edge_index)
        x = F.relu(x)
        x = F.dropout(x, training=self.training)
        x = self.conv2(x, edge_index)
        
        return F.log_softmax(x, dim=1)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 设备
epochs = 200 # 学习轮数
lr = 0.0003 # 学习率
num_node_features = dataset.num_node_features # 每个节点的特征数
num_classes = dataset.num_classes # 每个节点的类别数
data = dataset[0].to(device) # Cora的一张图

# 4.定义模型
model = EdgeCNN(num_node_features, num_classes).to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=lr) # 优化器
loss_function = nn.NLLLoss() # 损失函数

# 训练模式
model.train()

for epoch in range(epochs):
    optimizer.zero_grad()
    pred = model(data)
    
    loss = loss_function(pred[data.train_mask], data.y[data.train_mask]) # 损失
    correct_count_train = pred.argmax(axis=1)[data.train_mask].eq(data.y[data.train_mask]).sum().item() # epoch正确分类数目
    acc_train = correct_count_train / data.train_mask.sum().item() # epoch训练精度
    
    loss.backward()
    optimizer.step()
    
    if epoch % 20 == 0:
        print("【EPOCH: 】%s" % str(epoch + 1))
        print('训练损失为：{:.4f}'.format(loss.item()), '训练精度为：{:.4f}'.format(acc_train))

print('【Finished Training！】')

# 模型验证
model.eval()
pred = model(data)

# 训练集（使用了掩码）
correct_count_train = pred.argmax(axis=1)[data.train_mask].eq(data.y[data.train_mask]).sum().item()
acc_train = correct_count_train / data.train_mask.sum().item()
loss_train = loss_function(pred[data.train_mask], data.y[data.train_mask]).item()

# 测试集
correct_count_test = pred.argmax(axis=1)[data.test_mask].eq(data.y[data.test_mask]).sum().item()
acc_test = correct_count_test / data.test_mask.sum().item()
loss_test = loss_function(pred[data.test_mask], data.y[data.test_mask]).item()

print('Train Accuracy: {:.4f}'.format(acc_train), 'Train Loss: {:.4f}'.format(loss_train))
print('Test  Accuracy: {:.4f}'.format(acc_test), 'Test  Loss: {:.4f}'.format(loss_test))

到此这篇关于基于PyTorch实现EdgeCNN的文章就介绍到这了,更多相关PyTorch实现EdgeCNN内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Python sorted函数详解(高级篇)
本文我们用到了sorted 如何进行按照键或者值进行排序，解决了字典的排序问题。文中将进一步详细介绍sorted的强大。希望对大家有所帮助
2018-09-09
Python设计模式之装饰模式实例详解
这篇文章主要介绍了Python设计模式之装饰模式,结合实例形式详细分析了装饰模式的概念、原理并结合Python实例形式分析了装饰模式的相关使用技巧,需要的朋友可以参考下
2019-01-01
一文搞懂python异常处理、模块与包
异常就是系统中的错误，程序是无法改变和处理的，文中有给大家提到异常处理机制，模块与包的相关知识，通过实例代码给大家介绍的非常详细，需要的朋友参考下吧
2021-06-06
Python 中Django安装和使用教程详解
这篇文章主要介绍了python中Django安装和使用教程，本文图文并茂给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-07-07
Python实现一个简单的QQ截图
大家好，本篇文章主要讲的是Python实现一个简单的QQ截图，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下的相关资料
2022-02-02
Python3实现网页内容转换成PDF文档和图片
pdfkit是把 HTML+CSS 格式的文件转换成 PDF 的一种工具，它是 wkhtmltopdf 这个工具包的 python 封装。本文将利用pdfkit实现网页内容转换成PDF文档和图片效果，感兴趣的可以学习一下
2022-06-06
python中datetime模块中strftime/strptime函数的使用
这篇文章主要介绍了python中datetime模块中strftime/strptime函数的使用，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-07-07
Python多进程模式实现多核CPU并行计算
随着计算机硬件的不断发展，多核CPU已经成为普及的硬件设备，在本文中，我们将重点介绍在Python中如何利用多进程模式提高程序的执行效率，感兴趣的可以了解一下
2023-05-05
Python时间序列缺失值的处理方法（日期缺失填充）
这篇文章主要给大家介绍了关于Python时间序列缺失值（日期缺失填充）的处理方法，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
2019-08-08
Python一键安装全部依赖包的方法
今天小编就为大家分享一篇Python一键安装全部依赖包的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-08-08