Highlight

CSPNet: A New Backbone that can Enhance Learning Capability of CNN

Proposes Cross Stage Partial Network (CSPNet), a CNN backbone that cuts inference computation by reducing duplicate gradient information.

Based on

CSPNet: A New Backbone that can Enhance Learning Capability of CNN

By Chien-Yao Wang, H. Liao, I-Hau Yeh et al.2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)
Read original article →

CSPNet (Cross Stage Partial Network) is a backbone design aimed at reducing the costly inference computation of convolutional neural networks. The authors trace this cost to duplicate gradient information that arises during network optimization, and they address it by integrating feature maps from the beginning and the end of each network stage, so the network respects the variability of the gradients.

In experiments, CSPNet reduces computation by about 20% while achieving equivalent or even superior accuracy on ImageNet, and it significantly outperforms prior state-of-the-art approaches on AP50 for MS COCO object detection. Because the module is easy to implement and general enough to work with ResNet, ResNeXt, and DenseNet architectures, it offers a way to bring advanced vision models to devices with limited computation resources.

Abstract

CSPNet addresses the heavy inference computation of CNNs, which the authors attribute to duplicate gradient information during optimization. By integrating feature maps from the beginning and end of each network stage, the design respects gradient variability. Experiments show a 20% reduction in computation with equal or superior ImageNet accuracy, and stronger AP50 on MS COCO object detection. The module is easy to implement and works with ResNet, ResNeXt, and DenseNet.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

CNN backbonecomputational efficiencyobject detectiongradient flowImageNet
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.