论文笔记系列:主干网络(二)-- DenseNet

简介: 前言: DenseNet(Densely connected convolutional networks) 模型,它的基本思路与ResNet一致,但是它建立的是前面所有层与后面层的密集连接(dense connection),它的名称也是由此而来。DenseNet的另一大特色是通过特征在channel上的连接来实现特征重用(feature reuse)。这些特点让DenseNet在参数和计算成本更少的情形下实现比ResNet更优的性能,DenseNet也因此斩获CVPR 2017的最佳论文奖。本篇文章首先介绍DenseNet的原理以及网路架构,然后讲解DenseNet在Pytorch上的实现

稠密连接卷积神经网络Densely Connected Concolutional Networks



前言: DenseNet(Densely connected convolutional networks) 模型,它的基本思路与ResNet一致,但是它建立的是前面所有层与后面层的密集连接(dense connection),它的名称也是由此而来。DenseNet的另一大特色是通过特征在channel上的连接来实现特征重用(feature reuse)。这些特点让DenseNet在参数和计算成本更少的情形下实现比ResNet更优的性能,DenseNet也因此斩获CVPR 2017的最佳论文奖。本篇文章首先介绍DenseNet的原理以及网路架构,然后讲解DenseNet在Pytorch上的实现。


对比Resnet设计理念:

相比ResNet,DenseNet提出了一个更激进的密集连接机制:即互相连接所有的层,具体来说就是每个层都会接受其前面所有层作为其额外的输入。图1为ResNet网络的连接机制,作为对比,图2为DenseNet的密集连接机制。可以看到,ResNet是每个层与前面的某层(一般是2~3层)短路连接在一起,连接方式是通过元素级相加。而在DenseNet中,每个层都会与前面所有层在channel维度上连接(concat)在一起(这里各个层的特征图大小是相同的,后面会有说明),并作为下一层的输入。对于一个 L LL 层的网络,DenseNet共包含 L ( L + 1 ) 2 \frac{L(L+1)}{2}2L(L+1)个连接,相比ResNet,这是一种密集连接。而且DenseNet是直接concat来自不同层的特征图,这可以实现特征重用,提升效率,这一特点是DenseNet与ResNet最主要的区别。


image.png

ResNet网络的短路连接机制(其中+代表的是元素级相加操作)

image.png


DenseNet网络的密集连接机制(其中c代表的是channel级连接操作)

image.png



论文结构


摘要: 捷径连接有效;本文提出激进的捷径连接FenseNet;在四个数据集获得SOTA

1. Introduction: CNN的发展,网络深度越来越深;网络深存在信息流通不畅问题;Short path广泛应用;DenseNet简介。

2. Related Work: 级联结构、捷径连接结构的网络模型简介;DenseNet简介。

3. DenseNets: Dense connectivity,pooling layer,Growth rate,bottleneck layer,compression等,网络细节

4. Experiments: 数据集介绍,训练超参介绍,4个数据的分类结果

5. Discussion: 实验结果分析探讨

6. Conclusion: 全文总结,总结本文工作,本文优点,将来可研究方向。


一、摘要核心

背景介绍:近年卷积神经网络中加入捷径连接之后,可训练更深、精度更高、更高效的网络

研究内容:本文提出DenseNet,其中的每一层会作为其后一层的输入层。对于L层的网络,传统方法有L层连接,而DenseNet有L×(L+1)/2个连接,所以叫DenseNet

本文优点:减轻梯度消失,增强特征传播,加强特征复用,减少权重参数

实验结果:各项指标超越ResNet

开源代码


二、DenseNet 结构

① Dense connectivity

稠密连接:在一个Block中,每一层输入来自于它前面所有层的特征图,每一层输出均会直接连接到它后面所有层的输入

L层block中,有L×(L+1)/2个连接

优点:

用较少参数获得更多特征,减少了参数;

低级特征得以复用,特征更丰富;

更强的梯度流动,跳层连接更多,梯度可更容易向前传播。


② Composite function

将BN层、ReLU层和3×3卷积层组成一组操作对特征进行提取 。


之前接触的操作:x->conv->BN->ReLU


DenseNet:[x0,x1,x2]->BN->ReLU->conv


模型接收的不同网络层的特征图的尺度可能会有很大的差异,再去做卷积不方便。所以在网络层之前做BN和ReLU。


③ Pooling layers

池化层用于降低特征图分辨率,DenseNet将各个block之间的特征图分辨率下降操作成为transition layer,由BN、1×1卷积、2×2池化构成。


transition layer用于dense block之间,Dense block内部要求特征图分辨率保持一致。


④ Growth rate

稠密连接的方式,使得特征图通道数随着网络层的加深不断增多,因此要控制特征图数量。所以引入Growth rate超参数k来衡量dense block中特征图的多少


Growth rate在代码中就是卷积层的卷积核个数,就像ResNeXt中的cardinality就是分组卷积的分组数而已


⑤ Bottleneck layers——DenseNet-B

瓶颈层来降低特征图通道数,减少后续卷积的计算量,当采用率以下形式进行特征提取,则网络会成为DenseNet-B


采用了1×1卷积来降低特征图分辨率,降到4k。k是growth rate增长率

image.png


【一个DenseNet中有多个(3个或4个)Dense Block】

【一个Dense Block中有多个Bottleneck layers】

【代码中,Bottleneck layers也称为Dense layers】


⑥ Compression

为进一步使DenseNet紧凑,在transition layer处采用较少的卷积核进行特征提取,从而压缩特征图通道数,压缩率用θ表示,典型值为0.5


当采用了Compression时,模型称为DenseNet-C


当同时采用Bottleneck layer和Compression时,称为DenseNet-BC


通常用的DenseNet指的DenseNet-BC


⑦ DenseNet 网络结构

整体分为三大块:头部卷积、Dense Block堆叠、全连接输出分类概率


头部卷积池化降低分辨率,

中间多个dense block+transition layer堆叠,

池化+全连接输出。

image.png


7×7的感受野比较大

进入一系列block堆叠的时候,都是56×56


三、实验结果及分析

① 在3个数据集(Cifar-10/100,SVHN数据)上进行对比实验

C10+表示基于数据增强

image.png


结论:


DenseNet-BC-190获得最优成绩

DenseNet性能随k和L增大而提升

DenseNet-BC-250省参数,仅15.3M

粉色看出过拟合,可用数据增强

绿色看出省参数,十分之一不到

② 在Cifar-10数据集上分析DenseNet各种方案的结果

image.png


最左边的图:

DenseNet-C优于DenseNet

DenseNet-B优于DenseNet-C

DenseNet-BC优于DenseNet-B

image.png

右边两张图:对于ResNet

同样精度时,仅需ResNet的三分之一参数,

表明DenseNet更省参数,更高效


③ ImageNet-1K分类数据集


最左边的图:

结论:


10-crop均比single-crop精度高

网络越深、精度越高

右边两张图:

与ResNet对比精度与参数

3. 同参数,精度全面超越ResNet

4. 同精度,参数全面少于ResNet


④ 分析特征重要性

image.png

分析的是DenseNet-40,其中有三个Denseblock,

第一行表示DenseBlock前一层会传给后面所有层,

最后一列表示Transition layer与前面所有层的连接


结论:


网络层的权重,不仅关注上一层的特征图,还关注前面所有层的特征图。

transition layer的权重均匀分布在各个layer的特征图。

block2和3中,最上面那一行,对transition输出的特征的权重很低,DenseNet-BC正是压缩了这一部分(因为不太重要)

block3中看到分类输出层对靠后的特征更关注,可能因为后面得到的是高级特征,更有分辨力。


四、论文总结

① 关键点、创新点

将捷径连接与特征复用思想结合,同时借鉴级联结构的思想,设计出稠密连接结构

针对稠密连接,提出DenseNet-BC形式,分别占block内部和transition部分采用1×1卷积进行参数量的控制,使模型紧凑

② 备用参考文献知识点

信息流通不顺畅时,普遍采用捷径连接来处理

as information about the input or gradient passes through many layers, it can vanish and “wash out” by the time it reaches the end (or beginning) of the network.

they create short paths from early layers to later layers.

(论文1的第二段)


采用捷径连接,便于信息流通,以及梯度传递,课可使模型便于训练

one big advantage of DenseNets is their improved flow of information and gradients throughout the network, which makes them easy to train.(论文1的第五段)


稠密连接在小数据集上起到正则作用,减少过拟合。

we also observe that dense connections have a regularizing effect, which reduces over-fitting on tasks with smaller training set sizes.(论文1的第五段)


ResNet的求和形式有缺点,会阻碍信息的流通。对要用拼接而不是求和时,可以引用这篇论文的这一点。

However, the identity function and the output of H` are combined by summation, which may impede the information flow in the network.(论文3的第二段)


五、研究思想

(1)short paths思想

深度神经网络中存在信息流通不畅问题——前向传播时输入数据消失、反向传播时梯度消失

应用 :

Highway Net 带门控单元的short paths

ResNet、随机深度的ResNet:在运算的部分,设置了一定的概率让它丢失掉,层数就减少了。

FractalNet

(2)多级特征复用

当前的特征不仅会给下一层使用,还会给下下层……使用。

可以增强模型


应用:

FCN: Fully Convolutional Networks

输出的score map由多级特征图共同生成

多尺度有向无环卷积神经网络:采用多个层的特征复用进行分类,提高分类精度

优点:不需要额外的计算量,可得到底层、中层和高层的特征来进行分类,使模型更鲁棒。

(3)结构类似的级联结构(第40篇参考文献)

前面神经元的输出会连到后面的所有神经元。类似于DenseNet的稠密连接。


目录
相关文章
|
4月前
|
机器学习/深度学习 资源调度 算法框架/工具
AI-ANNE: 将神经网络迁移到微控制器的深度探索——论文阅读
AI-ANNE框架探索将深度学习模型迁移至微控制器的可行路径,基于MicroPython在Raspberry Pi Pico上实现神经网络核心组件,支持本地化推理,推动TinyML在边缘设备中的应用。
280 10
|
4月前
|
机器学习/深度学习 边缘计算 算法
SEENN: 迈向时间脉冲早退神经网络——论文阅读
SEENN提出一种时间脉冲早退神经网络,通过自适应调整每个样本的推理时间步数,有效平衡脉冲神经网络的准确率与计算效率。该方法基于置信度判断或强化学习策略,在保证高精度的同时显著降低能耗与延迟,适用于边缘计算与实时处理场景。
251 13
|
4月前
|
机器学习/深度学习 缓存 算法
2025年华为杯A题|通用神经网络处理器下的核内调度问题研究生数学建模|思路、代码、论文|持续更新中....
2025年华为杯A题|通用神经网络处理器下的核内调度问题研究生数学建模|思路、代码、论文|持续更新中....
472 1
|
SQL 监控 安全
网络安全与信息安全:漏洞、加密与安全意识
随着互联网的迅猛发展,网络安全和信息安全问题日益受到关注。本文深入探讨了网络安全漏洞、加密技术以及提高个人和组织的安全意识的重要性。通过分析常见的网络攻击手段如缓冲区溢出、SQL注入等,揭示了计算机系统中存在的缺陷及其潜在威胁。同时,详细介绍了对称加密和非对称加密算法的原理及应用场景,强调了数字签名和数字证书在验证信息完整性中的关键作用。此外,还讨论了培养良好上网习惯、定期备份数据等提升安全意识的方法,旨在帮助读者更好地理解和应对复杂的网络安全挑战。
|
SQL 安全 网络安全
网络安全与信息安全:知识分享####
【10月更文挑战第21天】 随着数字化时代的快速发展,网络安全和信息安全已成为个人和企业不可忽视的关键问题。本文将探讨网络安全漏洞、加密技术以及安全意识的重要性,并提供一些实用的建议,帮助读者提高自身的网络安全防护能力。 ####
290 17
|
SQL 安全 网络安全
网络安全与信息安全:关于网络安全漏洞、加密技术、安全意识等方面的知识分享
随着互联网的普及,网络安全问题日益突出。本文将从网络安全漏洞、加密技术和安全意识三个方面进行探讨,旨在提高读者对网络安全的认识和防范能力。通过分析常见的网络安全漏洞,介绍加密技术的基本原理和应用,以及强调安全意识的重要性,帮助读者更好地保护自己的网络信息安全。
241 10
|
存储 SQL 安全
网络安全与信息安全:关于网络安全漏洞、加密技术、安全意识等方面的知识分享
随着互联网的普及,网络安全问题日益突出。本文将介绍网络安全的重要性,分析常见的网络安全漏洞及其危害,探讨加密技术在保障网络安全中的作用,并强调提高安全意识的必要性。通过本文的学习,读者将了解网络安全的基本概念和应对策略,提升个人和组织的网络安全防护能力。
|
SQL 安全 网络安全
网络安全与信息安全:关于网络安全漏洞、加密技术、安全意识等方面的知识分享
在数字化时代,网络安全和信息安全已成为我们生活中不可或缺的一部分。本文将介绍网络安全漏洞、加密技术和安全意识等方面的内容,并提供一些实用的代码示例。通过阅读本文,您将了解到如何保护自己的网络安全,以及如何提高自己的信息安全意识。
254 10
|
监控 安全 网络安全
网络安全与信息安全:漏洞、加密与意识的交织
在数字时代的浪潮中,网络安全与信息安全成为维护数据完整性、保密性和可用性的关键。本文深入探讨了网络安全中的漏洞概念、加密技术的应用以及提升安全意识的重要性。通过实际案例分析,揭示了网络攻击的常见模式和防御策略,强调了教育和技术并重的安全理念。旨在为读者提供一套全面的网络安全知识框架,从而在日益复杂的网络环境中保护个人和组织的资产安全。