arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.05066cs.CVcs.LGeess.IV
收藏

参数有效卷积神经网络的深度共享滤波器基

Deeply Shared Filter Bases for Parameter-Efficient Convolutional Neural Networks

Incheon National University(仁川大学);Dain Technology, Inc.

Woochul Kang, Daeyeon Kim

摘要:

现代卷积神经网络(CNN)具有大量相同的卷积块,因此,已经提出了在这些块之间递归共享参数以减少参数的量。然而,天真的参数共享提出了许多挑战,如有限的代表性权力和递归共享参数的消失/爆炸梯度问题。在本文中,我们提出了一种递归卷积块的设计和训练方法,其中一个递归共享的部分,或滤波器的基础,分离和学习,同时有效地避免了消失/爆炸梯度问题在训练过程中。我们表明,可以通过强制执行滤波器基正交规范的元素来控制笨拙的消失/爆炸梯度问题,并根据经验证明,所提出的正交正则化改进了训练过程中的梯度流。图像分类和目标检测的实验结果表明,我们的方法,不像以前的参数共享方法,不交易的性能,以节省参数,并始终优于overparameterized对应的网络。这种优异的性能表明,所提出的递归卷积块设计和正交性正则化不仅防止性能下降,而且在递归地共享大量参数的同时始终提高表示能力。

英文摘要:

Modern convolutional neural networks (CNNs) have massive identical convolution blocks, and, hence, recursive sharing of parameters across these blocks has been proposed to reduce the amount of parameters. However, naive sharing of parameters poses many challenges such as limited representational power and the vanishing/exploding gradients problem of recursively shared parameters. In this paper, we present a recursive convolution block design and training method, in which a recursively shareable part, or a filter basis, is separated and learned while effectively avoiding the vanishing/exploding gradients problem during training. We show that the unwieldy vanishing/exploding gradients problem can be controlled by enforcing the elements of the filter basis orthonormal, and empirically demonstrate that the proposed orthogonality regularization improves the flow of gradients during training. Experimental results on image classification and object detection show that our approach, unlike previous parameter-sharing approaches, does not trade performance to save parameters and consistently outperforms overparameterized counterpart networks. This superior performance demonstrates that the proposed recursive convolution block design and the orthogonality regularization not only prevent performance degradation, but also consistently improve the representation capability while a significant amount of parameters are recursively shared.

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

↑