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

通过上下文相关性建模实现精确的二进制神经网络

Towards Accurate Binary Neural Networks via Modeling Contextual Dependencies

Beihang University(北京航空航天大学);SenseTime Group(商汤集团);Nanyang Technological University(南洋理工大学)

Xingrun Xing, Yangguang Li, Wei Li, Wenrui Ding, Yalong Jiang, Yufeng Wang, Jing Shao, Chunlei Liu, Xianglong Liu

摘要:

现有的二值神经网络主要是对二值化函数进行局部卷积运算。然而,这种简单的位操作缺乏建模上下文依赖关系的能力,这对于学习视觉模型中的区分性深度表示至关重要。在这项工作中,我们通过提出新的二进制神经模块设计来解决这个问题,这使得BNN能够学习有效的上下文依赖关系。首先,我们提出了一个二进制多层感知器(MLP)块作为替代二进制卷积块直接建模上下文依赖。短距离和长距离特征依赖性都由二进制MLP建模,其中前者提供局部归纳偏差,后者打破了二进制卷积中有限的感受野。其次,为了提高具有上下文依赖性的二进制模型的鲁棒性,我们计算上下文动态嵌入以确定一般二进制卷积块中的二进制化阈值。有了我们的二进制MLP块和改进的二进制卷积,我们用显式上下文依赖模型构建了BNN,称为BCDNet。在标准的ImageNet-1 K分类基准测试中,BCDNet达到了72.3%的Top-1准确率,并且远远优于领先的二进制方法。特别是,拟议的BCDNet超过了最先进的ReActNet-A的2.9%的前1名的准确性与类似的操作。我们的代码可在https://github.com/Sense-GVT/BCDN上获得

英文摘要:

Existing Binary Neural Networks (BNNs) mainly operate on local convolutions with binarization function. However, such simple bit operations lack the ability of modeling contextual dependencies, which is critical for learning discriminative deep representations in vision models. In this work, we tackle this issue by presenting new designs of binary neural modules, which enables BNNs to learn effective contextual dependencies. First, we propose a binary multi-layer perceptron (MLP) block as an alternative to binary convolution blocks to directly model contextual dependencies. Both short-range and long-range feature dependencies are modeled by binary MLPs, where the former provides local inductive bias and the latter breaks limited receptive field in binary convolutions. Second, to improve the robustness of binary models with contextual dependencies, we compute the contextual dynamic embeddings to determine the binarization thresholds in general binary convolutional blocks. Armed with our binary MLP blocks and improved binary convolution, we build the BNNs with explicit Contextual Dependency modeling, termed as BCDNet. On the standard ImageNet-1K classification benchmark, the BCDNet achieves 72.3% Top-1 accuracy and outperforms leading binary methods by a large margin. In particular, the proposed BCDNet exceeds the state-of-the-art ReActNet-A by 2.9% Top-1 accuracy with similar operations. Our code is available at https://github.com/Sense-GVT/BCDN

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

↑