【深度学习与NLP】——深度卷积神经网络AlexNet

本文主要是介绍【深度学习与NLP】——深度卷积神经网络AlexNet，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一、卷积神经网络的发展历程

二、简要介绍

三、代码实现

四、缺点和过时的地方

一、卷积神经网络的发展历程

早期理论基础阶段（20 世纪 60 年代 - 80 年代）：
- 1968 年，Hubel 和 Wiesel 通过对猫视觉神经的研究，发现了视觉神经元对图像边缘的响应特性，提出了感受野的概念，为卷积神经网络的发展奠定了生物学基础 3。
- 1980 年，日本科学家福岛邦彦提出了 Neocognitron，它模拟了脑神经科学的结构，具备现代 CNN 的一些基本元素，如逐步的滤波器、使用 ReLU 提供非线性、平均池化下采样等，保证了网络的平移不变性，实现了稀疏交互，但无法进行有监督学习。
初步发展阶段（20 世纪 90 年代）：
- 1990 年，Yann LeCun 将反向传播算法应用到类似 Neocognitron 的网络上，实现了一个用于手写数字识别的神经网络，简化了卷积操作以便于反向传播的应用，这是 CNN 用于有监督学习的早期重要实践。
- 1998 年，Yann LeCun 提出 LeNet - 5，这是具有里程碑意义的 CNN 架构。LeNet - 5 定义了 CNN 的基本框架，包括卷积层、池化层和全连接层，在手写数字识别任务上取得了良好效果。不过，当时受限于计算机算力和数据量，其应用范围相对有限。
沉寂阶段（2000 年 - 2011 年）：这一时期，由于计算资源有限、数据集规模较小以及其他机器学习方法（如支持向量机）的竞争等原因，卷积神经网络的发展相对缓慢，处于沉寂状态。
复兴与突破阶段（2012 年 - 至今）：
- 2012 年，AlexNet 诞生。它在当年的 ImageNet 大规模视觉识别挑战赛中以显著优势夺冠，标志着神经网络的复苏和深度学习的崛起。AlexNet 采用了更深的网络结构，使用 ReLU 激活函数、数据增强、mini - batch SGD 优化、在 GPU 上训练以及 Dropout 技术来避免过拟合等创新方法，极大地推动了 CNN 的发展，也让更多研究者关注到深度学习的潜力。
- 2014 年，VGGNet 被提出，它通过增加网络深度（如 VGG - 16、VGG - 19），证明了增加网络深度可以提升模型性能，为后续研究提供了思路。
- 2015 年，ResNet 出现，它通过引入残差结构，有效解决了随着网络深度增加而导致的梯度消失问题，使得训练更深的网络成为可能，并且第一个在 ImageNet 图片分类上表现超过人类水准，将 CNN 的性能推向新高度。此后，各种基于 ResNet 的改进和衍生模型不断涌现。
- 2017 年，SENet 提出，通过引入注意力机制，让网络能够自适应地关注重要特征，进一步提升了模型的性能和表达能力。
- 近年来，CNN 不断与其他技术融合，如与生成对抗网络（GAN）结合用于图像生成、与强化学习结合用于智能决策等，同时在自动驾驶、医疗影像分析、智能安防等众多领域得到广泛应用，且随着硬件计算能力的持续提升和大规模数据集的不断丰富，其性能和应用场景还在不断拓展和深化。

二、简要介绍

AlexNet 是一种深度卷积神经网络，在 2012 年的 ImageNet 大规模视觉识别挑战赛（ILSVRC）中取得了突破性的成果。

1. 网络结构

AlexNet 包含 8 层，其中前 5 层为卷积层，后 3 层为全连接层。具体结构如下：

第一层：卷积层，使用 96 个大小为 11×11×3 的卷积核，步长为 4。
第二层：最大池化层，池化窗口大小为 3×3，步长为 2。
第三层：卷积层，使用 256 个大小为 5×5 的卷积核。
第四层：最大池化层，池化窗口大小为 3×3，步长为 2。
第五层：卷积层，使用 384 个大小为 3×3 的卷积核。
第六层：卷积层，使用 384 个大小为 3×3 的卷积核。
第七层：卷积层，使用 256 个大小为 3×3 的卷积核。
第八层：全连接层，包含 4096 个神经元。最后两层全连接层也分别有 4096 个神经元，输出层则根据具体任务确定神经元数量。

2. 特点

非线性激活函数
使用 ReLU（Rectified Linear Unit）作为激活函数，相比传统的 sigmoid 和 tanh 函数，ReLU 能够加速训练过程，并且在一定程度上缓解了梯度消失问题。

数据增强
通过对图像进行随机裁剪、水平翻转等操作，增加了数据的多样性，提高了模型的泛化能力。

Dropout
在训练过程中随机将一些神经元的输出置为 0，有效地减少了过拟合。

多 GPU 训练
由于网络规模较大，训练数据也很多，AlexNet 采用了多 GPU 并行训练的方式，加快了训练速度。

3. 影响

AlexNet 的出现极大地推动了深度学习在计算机视觉领域的发展。它证明了深度神经网络在图像识别等任务上的强大能力，为后续的研究提供了重要的参考和启示。此后，各种深度神经网络架构不断涌现，性能也不断提升。总之，AlexNet 是深度学习发展历程中的一个重要里程碑，它的创新之处和优异性能对计算机视觉领域产生了深远的影响。

三、代码实现

 AlexNet 网络特点
它在多方面使用了创新性的结构 ；
（1）提出了 非饱和神经元 ReLU 减小 梯度下降 的训练时间；
（2）提用了 多GPU并行卷积操作 实现模型训练 加速 ；
（3）提用了 LRN(Local Response Normalization) 实现局部响应 归一化 ；
（4）提出了 Overlapping Pooling 使用 stride=2，kernal_size=3 使池化重叠，优于之前的 stride=2，kernal_size=2 ；
（5）引入了 dropout 正则化方法减少 全连接层中的 过拟合 ；
（6）此外，还采用 数据增强 的方法扩充数据集用以减小 过拟合 线性；
论文中 数据增强 采用的方式 ：1、图像的平移和水平旋转；2、改变图像RGB通道的强度；

实现代码：

import torch
import torch.nn as nn
from torchinfo import summary# 定义 AlexNet 类，继承自 nn.Module
class AlexNet(nn.Module):def __init__(self, num_classes=1000):# 调用父类的初始化方法super(AlexNet, self).__init__()# 定义特征提取部分的网络结构self.features = nn.Sequential(# 第一个卷积层，输入通道数为 3（彩色图像），输出通道数为 48，卷积核大小为 11x11，步长为 4，填充为 2nn.Conv2d(3, 48, kernel_size=11, stride=4, padding=2),# 使用 ReLU 激活函数，inplace=True 表示在原张量上进行操作，节省内存nn.ReLU(inplace=True),# 第一个最大池化层，池化核大小为 3x3，步长为 2nn.MaxPool2d(kernel_size=3, stride=2),# 第二个卷积层，输入通道数为 48，输出通道数为 128，卷积核大小为 5x5，填充为 2nn.Conv2d(48, 128, kernel_size=5, padding=2),nn.ReLU(inplace=True),# 第二个最大池化层，池化核大小为 3x3，步长为 2nn.MaxPool2d(kernel_size=3, stride=2),# 第三个卷积层，输入通道数为 128，输出通道数为 192，卷积核大小为 3x3，填充为 1nn.Conv2d(128, 192, kernel_size=3, padding=1),nn.ReLU(inplace=True),# 第四个卷积层，输入通道数为 192，输出通道数为 192，卷积核大小为 3x3，填充为 1nn.Conv2d(192, 192, kernel_size=3, padding=1),nn.ReLU(inplace=True),# 第五个卷积层，输入通道数为 192，输出通道数为 128，卷积核大小为 3x3，填充为 1nn.Conv2d(192, 128, kernel_size=3, padding=1),nn.ReLU(inplace=True),# 第三个最大池化层，池化核大小为 3x3，步长为 2nn.MaxPool2d(kernel_size=3, stride=2),)# 定义分类器部分的网络结构self.classifier = nn.Sequential(# 第一个全连接层，输入维度为 128*6*6，输出维度为 2048nn.Linear(128 * 6 * 6, 2048),nn.ReLU(inplace=True),# Dropout 层，用于防止过拟合nn.Dropout(),# 第二个全连接层，输入维度为 2048，输出维度为 2048nn.Linear(2048, 2048),nn.ReLU(inplace=True),# 第三个全连接层，输入维度为 2048，输出维度为 num_classes（分类的类别数）nn.Linear(2048, num_classes),)def forward(self, x):# 前向传播过程，将输入 x 通过特征提取部分x = self.features(x)# 将特征图展平为一维向量x = torch.flatten(x, 1)# 将展平后的向量通过分类器部分得到输出out = self.classifier(x)return out# 定义测试函数
def test():# 创建一个 AlexNet 实例net = AlexNet()# 生成一个随机输入张量，形状为 (1, 3, 224, 224)，表示一个批量大小为 1 的彩色图像，尺寸为 224x224y = net(torch.randn(1, 3, 224, 224))# 打印输出张量的大小print(y.size())# 使用 torchinfo 库的 summary 函数打印网络结构和参数信息summary(net, (1, 3, 224, 224))# 如果当前文件作为主程序运行
if __name__ == '__main__':# 调用测试函数test()

输出：

四、缺点和过时的地方

缺点：
计算资源需求大：AlexNet 包含大量的参数，例如中间两个全连接层有很大的 4096 个神经元，这导致计算量庞大，对硬件要求非常高，需要大量的训练数据和计算资源，在训练和部署时成本较高。
过拟合风险：尽管使用了一些方法如 Dropout 来减少过拟合，但在某些情况下仍可能存在过拟合问题。例如在数据量不够丰富或模型复杂度相对数据规模过高时，容易出现对训练数据过度拟合，而对新数据的泛化能力不足。
缺乏对多尺度特征的有效融合：主要依赖固定大小的卷积核和池化操作来提取特征，对于不同尺度的物体，可能不能很好地自适应地提取到最有效的特征，在处理多尺度目标方面的能力相对有限。
过时的地方：
网络结构设计：随着技术发展，后续出现了许多更高效、更精巧的网络结构。比如 VGGNet 通过重复使用简单的卷积层堆叠，构建了更深层且性能更优的网络；GoogleNet/Inception 引入了 Inception 模块，通过不同尺寸的卷积和池化层并行处理，提高了计算效率和准确率；ResNet 引入了残差学习框架，解决了深层网络训练困难的问题，能构建极深的网络并取得更好的性能等。相比之下，AlexNet 的结构设计显得相对简单和基础 1。
激活函数：虽然 AlexNet 使用 ReLU 激活函数在当时是一个重要创新，解决了传统激活函数（如 Sigmoid 和 Tanh）在训练时的梯度消失问题，加快了训练速度。但后续又出现了如 Leaky ReLU、PReLU、ELU 等改进的激活函数，它们在某些方面能更好地处理负值或解决神经元 “死亡” 问题，进一步提升了网络的性能和稳定性。
训练方法和优化技术：在训练过程中，AlexNet 使用的随机梯度下降（SGD）及其一些基本的优化策略，在当下看来也较为简单。现在有许多更先进的优化算法，如 Adagrad、Adadelta、RMSProp、Adam 等，它们能够自适应地调整学习率，更好地处理复杂的损失函数曲面，加快收敛速度并提高训练效果。
缺乏对硬件的进一步优化：如今的硬件平台（如 GPU、TPU 等）不断发展，新的神经网络架构设计会更充分地考虑如何与硬件特性相结合，以实现更高效的计算和推理。而 AlexNet 在设计时主要基于当时的硬件条件，没有充分利用后续硬件发展带来的新特性和优势进行针对性优化