arXiv reaDer
RepBNN:繰り返しによる拡張機能マップを備えた正確なバイナリニューラルネットワークに向けて
RepBNN: towards a precise Binary Neural Network with Enhanced Feature Map via Repeating
バイナリニューラルネットワーク(BNN)は、畳み込みニューラルネットワーク(CNN)の極端な量子化バージョンであり、すべての機能と重みが1ビットにマッピングされています。 BNNは、CNNをエッジデバイスまたはモバイルデバイスに適用できるようにするために多くのメモリと計算の需要を節約しますが、2値化後の表現機能が低下するため、ネットワークパフォーマンスが低下します。この論文では、新しい交換可能で使いやすい畳み込みモジュールRepConvを提案します。これは、パラメータの数と畳み込み計算に余分なコストをかけることなく、チャネル次元に沿って入力または出力をβ倍複製することで機能マップを強化します。また、バイナリ畳み込み、完全に接続されたレイヤー、バッチ正規化など、BNNモジュール全体でRepConvを使用するための一連のRepTranルールを定義します。実験は、RepTran変換後、引用数の多いBNNのセットが、元のBNNバージョンよりも普遍的に優れたパフォーマンスを達成したことを示しています。たとえば、Rep-ReCU-ResNet-20のトップ1の精度、つまりRepBconvで強化されたReCU-ResNet-20は、CIFAR-10で88.97%に達し、元のネットワークより1.47%高くなっています。また、Rep-AdamBNN-ReActNet-Aは、ImageNetで71.342%のトップ1精度を達成します。これは、BNNの最新の結果です。コードとモデルは、https://github.com/imfinethanks/Rep_AdamBNNで入手できます。
Binary neural network (BNN) is an extreme quantization version of convolutional neural networks (CNNs) with all features and weights mapped to just 1-bit. Although BNN saves a lot of memory and computation demand to make CNN applicable on edge or mobile devices, BNN suffers the drop of network performance due to the reduced representation capability after binarization. In this paper, we propose a new replaceable and easy-to-use convolution module RepConv, which enhances feature maps through replicating input or output along channel dimension by β times without extra cost on the number of parameters and convolutional computation. We also define a set of RepTran rules to use RepConv throughout BNN modules like binary convolution, fully connected layer and batch normalization. Experiments demonstrate that after the RepTran transformation, a set of highly cited BNNs have achieved universally better performance than the original BNN versions. For example, the Top-1 accuracy of Rep-ReCU-ResNet-20, i.e., a RepBconv enhanced ReCU-ResNet-20, reaches 88.97% on CIFAR-10, which is 1.47% higher than that of the original network. And Rep-AdamBNN-ReActNet-A achieves 71.342% Top-1 accuracy on ImageNet, a fresh state-of-the-art result of BNNs. Code and models are available at:https://github.com/imfinethanks/Rep_AdamBNN.
updated: Tue Jul 19 2022 03:38:34 GMT+0000 (UTC)
published: Tue Jul 19 2022 03:38:34 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アソシエイト