当前位置: 首页 > news >正文

深入理解残差网络(ResNet):原理与PyTorch实现

引言:深度神经网络的瓶颈

随着深度学习的发展,研究者发现简单地增加网络层数反而会导致模型性能下降——这种现象称为退化问题(Degradation Problem)。传统深层网络面临的主要挑战:

  1. 梯度消失/爆炸:反向传播时梯度指数级衰减或增大

  2. 训练困难:深层网络难以收敛到理想状态

  3. 性能饱和:深度增加时准确率反而下降

一、残差学习的核心思想

2015年,何恺明团队提出的残差网络(ResNet)通过引入跳跃连接(Skip Connection) 解决了这一难题,核心公式:

y=F(x,{Wi})+xy=F(x,{Wi​})+x

其中:

  • $\mathbf{x}$:输入特征

  • $\mathcal{F}$:残差函数

  • $\mathbf{y}$:输出特征

这种设计允许网络学习输入与输出之间的残差(差值),而非直接映射。当原始映射接近恒等映射时,学习残差$\mathcal{F} = \mathbf{y} - \mathbf{x}$比学习完整映射更容易。

二、残差块结构解析

import torch
import torch.nn as nnclass ResidualBlock(nn.Module):def __init__(self, in_channels, out_channels, stride=1):super().__init__()# 第一卷积层self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=3, stride=stride, padding=1, bias=False)self.bn1 = nn.BatchNorm2d(out_channels)# 第二卷积层self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=3, stride=1, padding=1, bias=False)self.bn2 = nn.BatchNorm2d(out_channels)# 跳跃连接处理维度变化self.shortcut = nn.Sequential()if stride != 1 or in_channels != out_channels:self.shortcut = nn.Sequential(nn.Conv2d(in_channels, out_channels,kernel_size=1, stride=stride, bias=False),nn.BatchNorm2d(out_channels))def forward(self, x):identity = self.shortcut(x)  # 保留原始输入out = self.conv1(x)out = self.bn1(out)out = nn.ReLU()(out)out = self.conv2(out)out = self.bn2(out)out += identity  # 关键步骤:添加跳跃连接out = nn.ReLU()(out)return out

三、ResNet网络架构

ResNet由多个残差块堆叠而成,不同深度的网络配置:

网络层ResNet-18ResNet-34ResNet-50
conv17×7, 64, stride 2
pool13×3 max pool, stride 2
conv2_x3×3, 64
3×3, 64 ×2
×31×1, 64
3×3, 64
1×1, 256 ×3
conv3_x3×3, 128
3×3, 128 ×2
×41×1, 128
3×3, 128
1×1, 512 ×4
conv4_x3×3, 256
3×3, 256 ×2
×61×1, 256
3×3, 256
1×1, 1024 ×6
conv5_x3×3, 512
3×3, 512 ×2
×31×1, 512
3×3, 512
1×1, 2048 ×3
全连接层1000-d fc

四、PyTorch实现ResNet-18

class ResNet(nn.Module):def __init__(self, block, layers, num_classes=1000):super().__init__()self.in_channels = 64# 初始卷积层self.conv1 = nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3, bias=False)self.bn1 = nn.BatchNorm2d(64)self.relu = nn.ReLU(inplace=True)self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)# 残差层self.layer1 = self._make_layer(block, 64, layers[0], stride=1)self.layer2 = self._make_layer(block, 128, layers[1], stride=2)self.layer3 = self._make_layer(block, 256, layers[2], stride=2)self.layer4 = self._make_layer(block, 512, layers[3], stride=2)# 分类头self.avgpool = nn.AdaptiveAvgPool2d((1, 1))self.fc = nn.Linear(512, num_classes)def _make_layer(self, block, out_channels, blocks, stride=1):layers = []# 第一个块可能需要下采样layers.append(block(self.in_channels, out_channels, stride))self.in_channels = out_channels# 剩余块for _ in range(1, blocks):layers.append(block(out_channels, out_channels))return nn.Sequential(*layers)def forward(self, x):x = self.conv1(x)x = self.bn1(x)x = self.relu(x)x = self.maxpool(x)x = self.layer1(x)x = self.layer2(x)x = self.layer3(x)x = self.layer4(x)x = self.avgpool(x)x = torch.flatten(x, 1)x = self.fc(x)return x# 实例化ResNet-18
def resnet18(num_classes=1000):return ResNet(ResidualBlock, [2, 2, 2, 2], num_classes)

五、关键优势分析

  1. 梯度传播优化:跳跃连接提供梯度高速公路

  2. 恒等映射保障:当残差接近0时自动退化为恒等函数

  3. 参数效率:相比传统网络,参数量更少但性能更好

六、训练技巧与注意事项

  1. 权重初始化:使用He初始化

    for m in self.modules():if isinstance(m, nn.Conv2d):nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu')

  2. 学习率调度:余弦退火策略

  3. 数据增强:随机裁剪、水平翻转

  4. 优化器选择:SGD with momentum (0.9)

http://www.lqws.cn/news/523909.html

相关文章:

  • Mysql数据库操作大全万字详解
  • 【Redis】Redis的下载安装和配置
  • 检查StringBuilder是否包含字符串
  • ARM内核之CMSIS
  • 【机器学习】非参数贝叶斯回归方法 GPR
  • ipfs在windows下载和安装
  • JSON框架转化isSuccess()为sucess字段
  • C++(智能指针)
  • Liunx操作系统笔记2
  • linux-修改文件命令(补充)
  • IT运维效率提升: 当IT监控遇上3D可视化
  • 三步实现B站缓存视频转MP4格式
  • 记一次AWS 中RDS优化费用使用的案例
  • Postman鉴权动态传参?对比脚本变量vs环境变量!
  • 理论加案例,一文读懂数据分析中的分类建模
  • 通过pyqt5学习MVC
  • 代理型 AI 重塑营销格局:国产 R²AIN SUITE 如何破解数据与技术瓶颈,实现 AI 赋能全链路提效
  • VScode常用快捷键【个人总结】
  • 2024年AEI SCI1区TOP,强化学习人工兔优化算法RLTARO+山地森林地形无人机编队路径规划,深度解析+性能实测
  • Dify、n8n、Coze、FastGPT 和 Ragflow 对比分析:如何选择最适合你的智能体平台?
  • Wpf的Binding
  • 数据库1.0
  • Python 爬虫入门:从数据爬取到转存 MySQL 数据库
  • 【Ansible】Ansible入门
  • Git常用操作详解
  • Python核心可视化库:Matplotlib与Seaborn深度解析
  • React 第六十四节Router中HashRouter的使用详细介绍及案例分析
  • 重置 MySQL root 密码
  • 基于STM32的智能节能风扇的设计
  • KNN算法(K近邻算法)