arXiv reaDer
ディープニューラルネットワークにおけるブロックウォルシュ-アダマール変換ベースのバイナリ層
Block Walsh-Hadamard Transform Based Binary Layers in Deep Neural Networks
畳み込みは、現代のディープニューラルネットワークのコアオペレーションです。畳み込みをフーリエ変換領域で実装できることはよく知られています。この論文では、フーリエ変換の代わりにバイナリブロックウォルシュ-アダマール変換(WHT)を使用することを提案します。 WHTベースのバイナリレイヤーを使用して、ディープニューラルネットワークの通常の畳み込みレイヤーの一部を置き換えます。このホワイトペーパーでは、1次元(1-D)と2次元(2-D)の両方のバイナリWHTを使用します。 1次元層と2次元層の両方で、入力特徴マップのバイナリWHTを計算し、ソフトしきい値とtanh関数を組み合わせて得られる非線形性を使用してWHTドメイン係数のノイズを除去します。ノイズ除去後、逆WHTを計算します。 1D-WHTを使用して1×1の畳み込み層を置き換え、2D-WHT層を使用して3×3の畳み込み層とスクイーズアンドエキサイト層を置き換えることができます。トレーニング可能なウェイトを持つ2D-WHTレイヤーは、高密度レイヤーを支援するためにグローバル平均プーリング(GAP)レイヤーの前に挿入することもできます。このようにして、トレーニング可能なパラメーターをわずかに減らすことで、トレーニング可能なパラメーターの数を大幅に減らすことができます。このホワイトペーパーでは、WHTレイヤーをMobileNet-V2、MobileNet-V3-Large、およびResNetに実装して、精度の低下を無視してパラメーターの数を大幅に削減します。さらに、速度テストによると、2D-FWHTレイヤーは通常の3×3畳み込みの約24倍の速度で実行され、NVIDIA Jetson Nano実験ではRAM使用量が19.51%少なくなっています。
Convolution has been the core operation of modern deep neural networks. It is well-known that convolutions can be implemented in the Fourier Transform domain. In this paper, we propose to use binary block Walsh-Hadamard transform (WHT) instead of the Fourier transform. We use WHT-based binary layers to replace some of the regular convolution layers in deep neural networks. We utilize both one-dimensional (1-D) and two-dimensional (2-D) binary WHTs in this paper. In both 1-D and 2-D layers, we compute the binary WHT of the input feature map and denoise the WHT domain coefficients using a nonlinearity which is obtained by combining soft-thresholding with the tanh function. After denoising, we compute the inverse WHT. We use 1D-WHT to replace the 1×1 convolutional layers, and 2D-WHT layers can replace the 3×3 convolution layers and Squeeze-and-Excite layers. 2D-WHT layers with trainable weights can be also inserted before the Global Average Pooling (GAP) layers to assist the dense layers. In this way, we can reduce the number of trainable parameters significantly with a slight decrease in trainable parameters. In this paper, we implement the WHT layers into MobileNet-V2, MobileNet-V3-Large, and ResNet to reduce the number of parameters significantly with negligible accuracy loss. Moreover, according to our speed test, the 2D-FWHT layer runs about 24 times as fast as the regular 3×3 convolution with 19.51% less RAM usage in an NVIDIA Jetson Nano experiment.
updated: Fri Jan 07 2022 23:52:41 GMT+0000 (UTC)
published: Fri Jan 07 2022 23:52:41 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アソシエイト