arXiv reaDer
スパース カーネル選択によるバイナリ ニューラル ネットワークの圧縮
Compacting Binary Neural Networks by Sparse Kernel Selection
バイナリ ニューラル ネットワーク (BNN) は、畳み込みの重みを 1 ビット値で表し、ストレージと計算の効率を高めます。この論文は、成功した BNN のバイナリ カーネルがほぼベキ分布しているという以前に明らかにされた現象に動機付けられています。この現象により、典型的な BNN を圧縮し、バイナリ カーネル サブスペース内で非反復カーネルを学習することで、さらに近いパフォーマンスを得ることができます。具体的には、2 値化プロセスをバイナリ コードブックの観点からカーネル グループ化と見なし、完全なコードブックからコードワードのより小さなサブセットを選択することを学習します。次に、Gumbel-Sinkhorn 手法を活用してコードワード選択プロセスを近似し、選択プロセスをエンドツーエンドで最適化するだけでなく、非反復占有を維持できる順列ストレートスルー推定器 (PSTE) を開発します。選択されたコードワード。実験により、私たちの方法がモデルサイズとビット単位の計算コストの両方を削減し、同等の予算の下で最先端の BNN と比較して精度の向上を達成することが確認されています。
Binary Neural Network (BNN) represents convolution weights with 1-bit values, which enhances the efficiency of storage and computation. This paper is motivated by a previously revealed phenomenon that the binary kernels in successful BNNs are nearly power-law distributed: their values are mostly clustered into a small number of codewords. This phenomenon encourages us to compact typical BNNs and obtain further close performance through learning non-repetitive kernels within a binary kernel subspace. Specifically, we regard the binarization process as kernel grouping in terms of a binary codebook, and our task lies in learning to select a smaller subset of codewords from the full codebook. We then leverage the Gumbel-Sinkhorn technique to approximate the codeword selection process, and develop the Permutation Straight-Through Estimator (PSTE) that is able to not only optimize the selection process end-to-end but also maintain the non-repetitive occupancy of selected codewords. Experiments verify that our method reduces both the model size and bit-wise computational costs, and achieves accuracy improvements compared with state-of-the-art BNNs under comparable budgets.
updated: Sat Mar 25 2023 13:53:02 GMT+0000 (UTC)
published: Sat Mar 25 2023 13:53:02 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アソシエイト