CSPNet论文笔记

论文地址:https://arxiv.org/pdf/1911.11929.pdf
开源代码:https://github.com/WongKinYiu/CrossStagePartialNetworks
作者提出Cross Stage Partial Network (CSPNet)来降低网络架构所需的算力。神经网络的计算量大很大程度上是由于在网络优化过程中出现的重复梯度信息导致的。

作者提出的网络架构可以在降低约20%计算量的情况下维持和原网络架构相同甚至高于原网络架构的准确度。CSPNet这种架构思想可以很方便的应用到像denseNet、ResNet、ResNeXt等结构的CNN网络上。

设计CSPNet的目的是在减少计算量的前提下实现丰富的梯度组合。它的核心思想是将某一阶段的(stage)输入特征图分成两部分,然后经过不同的处理后再合并起来。

CSPNet可以解决以下三个问题:

  • 增强CNN的学习能力;
  • 移除计算瓶颈,让整个模型的计算量分布更均匀,例如将每个stage的输入特征图分成两部分,只让其中一部分进入到dense layer,就能降低该stage的峰值计算量;
  • 降低内存占用,在特征金字塔生成过程中,为了降低内存占用,作者采用跨通道池化技术(cross-channel pooling)来压缩特征图数量。

这篇文章还总结了目前SOTA的网络架构特点。DenseNet通过大量复用特征图来大大降低参数数量和计算量,它将某层之前的所有层的输出作为后续层的输入,这将最大化ca

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值