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

基于矩阵分解的动态卷积算法研究

Revisiting Dynamic Convolution via Matrix Decomposition

University of California San Diego(加利福尼亚大学圣迭戈分校);Microsoft(微软公司)

Yunsheng Li, Yinpeng Chen, Xiyang Dai, Mengchen Liu, Dongdong Chen, Ye Yu, Lu Yuan, Zicheng Liu, Mei Chen, Nuno Vasconcelos

摘要:

动态卷积的最新研究表明,由于K个静态卷积核的自适应聚合,高效CNN的性能得到了大幅提升。它有两个局限性:(a)它将卷积权重的数量增加了K倍,以及(b)动态注意力和静态卷积内核的联合优化具有挑战性。在本文中,我们重新审视它从矩阵分解的一个新的角度,并揭示了关键问题是,动态卷积应用动态注意力投影到一个更高的维度潜在空间后的通道组。为了解决这个问题,我们提出了动态通道融合,以取代动态注意通道组。动态通道融合不仅可以显著降低潜在空间的维数,而且可以减轻联合优化的难度。因此,我们的方法更容易训练,并需要显着更少的参数,而不牺牲精度。源代码位于https://github.com/liyunsheng13/dcd。

英文摘要:

Recent research in dynamic convolution shows substantial performance boost for efficient CNNs, due to the adaptive aggregation of K static convolution kernels. It has two limitations: (a) it increases the number of convolutional weights by K-times, and (b) the joint optimization of dynamic attention and static convolution kernels is challenging. In this paper, we revisit it from a new perspective of matrix decomposition and reveal the key issue is that dynamic convolution applies dynamic attention over channel groups after projecting into a higher dimensional latent space. To address this issue, we propose dynamic channel fusion to replace dynamic attention over channel groups. Dynamic channel fusion not only enables significant dimension reduction of the latent space, but also mitigates the joint optimization difficulty. As a result, our method is easier to train and requires significantly fewer parameters without sacrificing accuracy. Source code is at https://github.com/liyunsheng13/dcd.

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

↑