arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1809.02601cs.CV
收藏

使用空间瓶颈模块加速深度神经网络

Accelerating Deep Neural Networks with Spatial Bottleneck Modules

Chinese Academy of Sciences(中国科学院);The Johns Hopkins University(约翰斯·霍普金斯大学);Microsoft Research(微软研究院)

Junran Peng, Lingxi Xie, Zhaoxiang Zhang, Tieniu Tan, Jingdong Wang

摘要:

本文提出了一个高效的空间瓶颈模块,用于加速深度神经网络中的卷积层。核心思想是将卷积分解为两个阶段,首先降低特征图的空间分辨率,然后将其恢复到所需的大小。该操作降低了空间域中的采样密度,这与信道域中的网络加速方法是独立的但互补的。使用不同的采样率,我们可以在识别精度和模型复杂度之间进行权衡。作为一个基本的构建块,空间瓶颈可以用来替换任何单个卷积层,或者两个卷积层的组合。我们通过将空间瓶颈应用于深度残差网络,实证验证了空间瓶颈的有效性。空间瓶颈算法对常规残差块和通道校验残差块分别实现了2倍和1.4倍的加速比,在识别低分辨率图像时保持了精度,在识别高分辨率图像时甚至有所提高。

英文摘要:

This paper presents an efficient module named spatial bottleneck for accelerating the convolutional layers in deep neural networks. The core idea is to decompose convolution into two stages, which first reduce the spatial resolution of the feature map, and then restore it to the desired size. This operation decreases the sampling density in the spatial domain, which is independent yet complementary to network acceleration approaches in the channel domain. Using different sampling rates, we can tradeoff between recognition accuracy and model complexity. As a basic building block, spatial bottleneck can be used to replace any single convolutional layer, or the combination of two convolutional layers. We empirically verify the effectiveness of spatial bottleneck by applying it to the deep residual networks. Spatial bottleneck achieves 2x and 1.4x speedup on the regular and channel-bottlenecked residual blocks, respectively, with the accuracies retained in recognizing low-resolution images, and even improved in recognizing high-resolution images.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑