经典CNN模型之AlexNet（Paddle）

虽然我是一个深度学习方面的小白&＃xff0c;但是对于卷积神经网络我仍然有很强的好奇心&＃xff0c;所以特地学习了一下&＃xff0c;如果出现了错误还望大家及时指正*~*。

一、简介

在LeNet提出后的几十年里&＃xff0c;神经网络一度被其他机器学习的方法超越&＃xff0c;而导致出现这种情况的原因主要有&＃xff1a;&＃xff08;1&＃xff09;数据&＃xff0c;一个包含许多特征的深度模型它需要大量的有标签的数据才能表现得比其他经典方法更好&＃xff1b;&＃xff08;2&＃xff09;硬件&＃xff08;算力&＃xff09;&＃xff0c;深度学习对计算资源要求很高&＃xff0c;但是早期的硬件计算能力非常有限。直到2012年&＃xff0c;这两点都得到了很大程度的改善&＃xff0c;这也就造就了AlexNet模型的横空出世。它首次证明了学习到的特征可以超越手工设计的特征&＃xff0c;一举打破了困扰计算机视觉研究的瓶颈。

二、网络结构

AlexNet与LeNet的设计理念非常相似&＃xff0c;但也有非常明显的区别。第一、AlexNet包含5层卷积、2层全连接隐藏层以及1层全连接输出层&＃xff1b;第二、AlexNet模型将sigmoid激活函数更改为了更为简单的ReLU激活函数&＃xff1b;第三、AlexNet通过丢弃法&＃xff08;Dropout&＃xff09;有效的控制了全连接层的模型复杂度&＃xff0c;防止引入过多的参数&＃xff1b;第四、AlexNet引入了大量的图像增广&＃xff0c;如翻转、裁剪以及颜色变化等&＃xff0c;有效的增加了数据样本的数量&＃xff0c;从而缓解了过拟合现象的发生。
由于ImageNet数据集中的图像要比MINST数据集大许多&＃xff0c;所以需要更大的卷积核来捕捉物体&＃xff0c;因此第一层卷积核的窗口为11×11。而第二层卷积核就减少到5×5&＃xff0c;之后就一直采用3×3。此外&＃xff0c;第一、第二和第五个卷积层之后都使用了窗口形状为3×3、步幅为2的最大池化层&＃xff0c;最后就是一般的全连接层。借用书中的图来表示一下&＃xff08;主要是我自己作图太难看*^*&＃xff09;&＃xff1a;

但是呢&＃xff0c;你只要细心一点就会发现&＃xff0c;第一层卷积层的224×224是有问题的&＃xff0c;应该是227×227才对&＃xff0c;这一点在吴恩达老师的视频中也被提到&＃xff0c;所以正确的图应该是下面这个。

三、代码实现&＃xff08;Paddle&＃xff09;

虽然知道上面的224×224是有点问题的&＃xff0c;但是我还只能用它来完成我的AlexNet实验&＃xff0c;这是因为当我使用227×227时&＃xff0c;出现了下面的错误&＃xff1a;

总的来说就是电脑配置有点差&＃xff0c;没法完成这个实验*^*&＃xff0c;所以就将就一下使用224×224来进行测试了&＃xff0c;框架使用的是百度的paddle飞桨&＃xff0c;代码如下所示。

import paddle
import paddle.nn.functional as F#组网
class AlexNet(paddle.nn.Layer):def __init__(self) -> None:super(AlexNet,self).__init__()self.conv1 &＃61; paddle.nn.Conv2D(in_channels &＃61; 1, out_channels &＃61; 96, kernel_size &＃61; 11, stride &＃61; 4)self.conv2 &＃61; paddle.nn.Conv2D(in_channels &＃61; 96, out_channels &＃61; 256, kernel_size &＃61; 5, stride &＃61; 1, padding &＃61; 2)self.conv3 &＃61; paddle.nn.Conv2D(in_channels &＃61; 256, out_channels &＃61; 384, kernel_size &＃61; 3, stride &＃61; 1, padding &＃61; 1)self.conv4 &＃61; paddle.nn.Conv2D(in_channels &＃61; 384, out_channels &＃61; 384, kernel_size &＃61; 3, stride &＃61; 1, padding &＃61; 1)self.conv5 &＃61; paddle.nn.Conv2D(in_channels &＃61; 384, out_channels &＃61; 256, kernel_size &＃61; 3, stride &＃61; 1, padding &＃61; 1)self.pool1 &＃61; paddle.nn.MaxPool2D(3,2)self.pool2 &＃61; paddle.nn.MaxPool2D(3,2)self.pool3 &＃61; paddle.nn.MaxPool2D(3,2)self.linear1 &＃61; paddle.nn.Linear(in_features &＃61; 256*5*5, out_features &＃61; 4096)self.linear2 &＃61; paddle.nn.Linear(in_features &＃61; 4096, out_features &＃61; 4096)self.linear3 &＃61; paddle.nn.Linear(in_features &＃61; 4096, out_features &＃61; 10)def forward(self,x):x &＃61; self.conv1(x)x &＃61; F.relu(x)x &＃61; self.pool1(x)x &＃61; self.conv2(x)x &＃61; F.relu(x)x &＃61; self.pool2(x)x &＃61; self.conv3(x)x &＃61; F.relu(x)x &＃61; self.conv4(x)x &＃61; F.relu(x)x &＃61; self.conv5(x)x &＃61; F.relu(x)x &＃61; self.pool3(x)x &＃61; paddle.flatten(x,start_axis&＃61;1,stop_axis&＃61;-1)x &＃61; self.linear1(x)x &＃61; F.relu(x)x &＃61; F.dropout(x,0.5)x &＃61; self.linear2(x)x &＃61; F.relu(x)x &＃61; F.dropout(x,0.5)x &＃61; self.linear3(x)return x#读取数据
def load_data_fashion_mnist(batch_size, resize&＃61;None):"""Download the fashion mnist dataset and then load into memory."""trans &＃61; []if resize:trans.append(paddle.vision.transforms.Resize(size&＃61;resize)) #将输入数据调整为指定大小trans.append(paddle.vision.transforms.ToTensor())transform &＃61; paddle.vision.transforms.Compose(trans)mnist_train &＃61; paddle.vision.datasets.FashionMNIST(mode&＃61;&＃39;train&＃39;, transform&＃61;transform) #由于ImageNet太大所以使用了FashionMNIST数据集mnist_test &＃61; paddle.vision.datasets.FashionMNIST(mode&＃61;&＃39;test&＃39;, transform&＃61;transform)train_iter &＃61; paddle.io.DataLoader(mnist_train,batch_size&＃61;batch_size, shuffle&＃61;True,places&＃61;paddle.CPUPlace())test_iter &＃61; paddle.io.DataLoader(mnist_test,batch_size&＃61;batch_size, shuffle&＃61;False, places&＃61;paddle.CPUPlace()) #places&＃61;paddle.CPUPlace()缺少这个关键字会莫名其妙的停止程序&＃xff0c;飞桨官网的解释是当前还不支持在子进程中进行GPU Tensor的操作&＃xff0c;请不要在子进程流程中使用GPU Tensorreturn train_iter , test_iterbatch_size &＃61; 128
# 如出现“out of memory”的报错信息&＃xff0c;可减⼩batch_size或resize
train_iter, test_iter &＃61; load_data_fashion_mnist(batch_size,resize&＃61;224)# 加载训练集 batch_size 设为 128
def train(model):model.train()epochs &＃61; 5optim &＃61; paddle.optimizer.Adam(learning_rate&＃61;0.001, parameters&＃61;model.parameters())# 用Adam作为优化函数for epoch in range(epochs):for batch_id, data in enumerate(train_iter()):x_data &＃61; data[0]y_data &＃61; data[1]predicts &＃61; model(x_data)loss &＃61; F.cross_entropy(predicts, y_data) #交叉熵损失函数# 计算损失acc &＃61; paddle.metric.accuracy(predicts, y_data) #计算数据的准确度loss.backward()if batch_id % 100 &＃61;&＃61; 0:print("epoch: {}, batch_id: {}, loss is: {}, acc is: {}".format(epoch, batch_id, loss.numpy(), acc.numpy()))optim.step()optim.clear_grad()
model &＃61; AlexNet()
train(model)


运行的效果&＃xff1a;

四、小结

当然&＃xff0c;每个深度学习的框架里可能都有现成的AlexNet模型&＃xff0c;我这里之所以选择自己构建一下&＃xff0c;主要是出于学习的目的去更好的了解这个模型的一些细节&＃xff0c;自我感觉这样更有趣一些。

参考资料&＃xff1a;《动手学深度学习》《吴恩达老师视频》