1. 首页
  2. 人工智能
  3. 论文/代码
  4. 用于深度CNN的基于预定义稀疏内核的卷积

用于深度CNN的基于预定义稀疏内核的卷积

上传者: 2021-01-23 05:30:58上传 .PDF文件 1.84 MB 热度 22次

对计算和存储资源的高需求严重阻碍了深度卷积神经网络(CNN)在有限资源设备中的部署。最近的CNN架构已经提出了降低复杂性的版本(例如SuffleNet和MobileNet),但是以适度降低不准确性为代价。..

A Pre-defined Sparse Kernel Based Convolution for Deep CNNs

The high demand for computational and storage resources severely impede the deployment of deep convolutional neural networks (CNNs) in limited-resource devices. Recent CNN architectures have proposed reduced complexity versions (e.g. SuffleNet and MobileNet) but at the cost of modest decreases inaccuracy.This paper proposes pSConv, a pre-defined sparse 2D kernel-based convolution, which promises significant improvements in the trade-off between complexity and accuracy for both CNN training and inference. To explore the potential of this approach, we have experimented with two widely accepted datasets, CIFAR-10 and Tiny ImageNet, in sparse variants of both the ResNet18 and VGG16 architectures. Our approach shows a parameter count reduction of up to 4.24x with modest degradation in classification accuracy relative to that of standard CNNs. Our approach outperforms a popular variant of ShuffleNet using a variant of ResNet18 with pSConv having 3x3 kernels with only four of nine elements not fixed at zero. In particular, the parameter count is reduced by 1.7x for CIFAR-10 and 2.29x for Tiny ImageNet with an increased accuracy of ~4%.

用户评论