arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2007.05223cs.CV
收藏

二进制卷积神经网络的蒸馏引导残差学习

Distillation Guided Residual Learning for Binary Convolutional Neural Networks

Microsoft(微软)

Jianming Ye, Shiliang Zhang, Jingdong Wang

摘要:

弥合二进制CNN(BCNN)和浮点CNN(FCNN)之间的性能差距具有挑战性。我们观察到,这种性能差距导致BCNN和FCNN的中间特征图之间存在大量残差。为了最小化性能差距,我们强制BCNN产生与FCNN类似的中间特征图。这种训练方法,即利用从FCNN导出的逐块蒸馏损失来优化每个二进制卷积块,导致对BCNN的更有效的优化。这也促使我们更新二进制卷积块架构,以促进逐块蒸馏损失的优化。具体地,将轻量级快捷分支插入到每个二进制卷积块中以补充每个块处的残差。得益于其挤压和相互作用(SI)结构,这个快捷分支引入了一小部分参数,例如,10\%的间接费用,但有效地补充了残差。在ImageNet上的大量实验证明了我们的方法在分类效率和准确性方面的优越性能,例如,使用我们的方法训练的BCNN在ImageNet上达到了60.45%的准确率。

英文摘要:

It is challenging to bridge the performance gap between Binary CNN (BCNN) and Floating point CNN (FCNN). We observe that, this performance gap leads to substantial residuals between intermediate feature maps of BCNN and FCNN. To minimize the performance gap, we enforce BCNN to produce similar intermediate feature maps with the ones of FCNN. This training strategy, i.e., optimizing each binary convolutional block with block-wise distillation loss derived from FCNN, leads to a more effective optimization to BCNN. It also motivates us to update the binary convolutional block architecture to facilitate the optimization of block-wise distillation loss. Specifically, a lightweight shortcut branch is inserted into each binary convolutional block to complement residuals at each block. Benefited from its Squeeze-and-Interaction (SI) structure, this shortcut branch introduces a fraction of parameters, e.g., 10\% overheads, but effectively complements the residuals. Extensive experiments on ImageNet demonstrate the superior performance of our method in both classification efficiency and accuracy, e.g., BCNN trained with our methods achieves the accuracy of 60.45\% on ImageNet.

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

↑