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

用于紧凑高效深度神经网络的空间自适应滤波器单元

Spatially-Adaptive Filter Units for Compact and Efficient Deep Neural Networks

University of Ljubljana(卢布尔雅那大学);University of Birmingham(伯明翰大学)

Domen Tabernik, Matej Kristan, Aleš Leonardis

摘要:

卷积神经网络在许多计算机视觉任务中表现出色。它们最关键的架构元素之一是有效的感受野大小,必须手动设置以适应特定的任务。标准解决方案涉及大内核,下/上采样和扩张卷积。这些需要测试各种膨胀和下/上采样因子,并导致非紧凑表示和过多的参数。我们解决这个问题,提出了一个新的卷积滤波器组成的位移聚合单元(DAU)。DAU学习空间位移,并根据给定的问题调整各个卷积滤波器的感受野大小,从而消除了手工修改的需要。DAU在现有的最先进的架构中提供了卷积滤波器的无缝替代,我们在AlexNet,ResNet 50,ResNet 101,DeepLab和SRN-DeburNet上进行了演示。这种设计的好处在各种计算机视觉任务和数据集上得到了证明,例如图像分类(ILSVRC 2012),语义分割(PASCAL VOC 2011,Cityscape)和盲图像去模糊(GOPRO)。结果表明,DAU有效地分配参数,从而在类似或更好的性能下,使网络更紧凑,达到四倍。

英文摘要:

Convolutional neural networks excel in a number of computer vision tasks. One of their most crucial architectural elements is the effective receptive field size, that has to be manually set to accommodate a specific task. Standard solutions involve large kernels, down/up-sampling and dilated convolutions. These require testing a variety of dilation and down/up-sampling factors and result in non-compact representations and excessive number of parameters. We address this issue by proposing a new convolution filter composed of displaced aggregation units (DAU). DAUs learn spatial displacements and adapt the receptive field sizes of individual convolution filters to a given problem, thus eliminating the need for hand-crafted modifications. DAUs provide a seamless substitution of convolutional filters in existing state-of-the-art architectures, which we demonstrate on AlexNet, ResNet50, ResNet101, DeepLab and SRN-DeblurNet. The benefits of this design are demonstrated on a variety of computer vision tasks and datasets, such as image classification (ILSVRC 2012), semantic segmentation (PASCAL VOC 2011, Cityscape) and blind image de-blurring (GOPRO). Results show that DAUs efficiently allocate parameters resulting in up to four times more compact networks at similar or better performance.

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

↑