arXiv reaDer
自己分配型バイナリニューラルネットワーク
Self-Distribution Binary Neural Networks
この作業では、重みとアクティブ化の両方がバイナリ(つまり、1ビット表現)であるバイナリニューラルネットワーク(BNN)を研究します。特徴表現はディープニューラルネットワークにとって重要ですが、BNNでは、特徴は符号のみが異なります。以前の作業では、スケーリング係数をバイナリの重みとアクティブ化に導入して、量子化エラーを減らし、BNNの分類精度を効果的に向上させています。ただし、スケーリング係数はネットワークの計算の複雑さを増すだけでなく、バイナリ機能の兆候にも意味がありません。この目的のために、自己分配バイナリニューラルネットワーク(SD-BNN)が提案されています。まず、アクティベーション自己分布(ASD)を利用して、アクティベーションの符号分布を適応的に調整し、それによって畳み込みの出力の符号差を改善します。次に、Weight Self Distribution(WSD)を使用して重みの符号分布を調整し、畳み込みの出力の符号分布を微調整します。さまざまなネットワーク構造を持つCIFAR-10およびImageNetデータセットでの広範な実験により、提案されたSD-BNNは常に最先端の(SOTA)BNNを上回っています(たとえば、CIFAR-10で92.5%、ImageNetで66.5%を達成しています。 ResNet-18)より少ない計算コストで。コードはhttps://github.com/pingxue-hfut/SD-BNNで入手できます。
In this work, we study the binary neural networks (BNNs) of which both the weights and activations are binary (i.e., 1-bit representation). Feature representation is critical for deep neural networks, while in BNNs, the features only differ in signs. Prior work introduces scaling factors into binary weights and activations to reduce the quantization error and effectively improves the classification accuracy of BNNs. However, the scaling factors not only increase the computational complexity of networks, but also make no sense to the signs of binary features. To this end, Self-Distribution Binary Neural Network (SD-BNN) is proposed. Firstly, we utilize Activation Self Distribution (ASD) to adaptively adjust the sign distribution of activations, thereby improve the sign differences of the outputs of the convolution. Secondly, we adjust the sign distribution of weights through Weight Self Distribution (WSD) and then fine-tune the sign distribution of the outputs of the convolution. Extensive experiments on CIFAR-10 and ImageNet datasets with various network structures show that the proposed SD-BNN consistently outperforms the state-of-the-art (SOTA) BNNs (e.g., achieves 92.5% on CIFAR-10 and 66.5% on ImageNet with ResNet-18) with less computation cost. Code is available at https://github.com/ pingxue-hfut/SD-BNN.
updated: Wed Mar 03 2021 13:39:52 GMT+0000 (UTC)
published: Wed Mar 03 2021 13:39:52 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト