arXiv reaDer
XnODRおよびXnIDR:畳み込みニューラルネットワーク用の2つの正確で高速な完全接続レイヤー
XnODR and XnIDR: Two Accurate and Fast Fully Connected Layers For Convolutional Neural Networks
Capsule Networkは、視覚認識タスク用のディープニューラルネットワークの機能間の位置関係を定義するのに強力ですが、計算コストが高く、モバイルデバイスでの実行には適していません。ボトルネックは、カプセル間で使用される動的ルーティングメカニズムの計算の複雑さにあります。一方、XNOR-Netは、2値化プロセスでの情報損失のために精度が低くなりますが、高速で計算効率が高くなります。ダイナミックルーティングメカニズムの計算負荷に対処するために、このペーパーでは、CapsFCレイヤー内のダイナミックルーティングの外側または内側にリニアプロジェクターをxnoringすることにより、新しい完全接続(FC)レイヤーを提案します。具体的には、提案されたFCレイヤーには、XnODR(動的ルーティングの外側の線形射影線形射影をXnorize)とXnIDR(動的ルーティングの内側の線形射影線形射影をXnorize)の2つのバージョンがあります。 XnODRとXnIDRの両方の一般化をテストするために、それらをMobileNetV2とResNet-50の2つの異なるネットワークに挿入します。 3つのデータセット、MNIST、CIFAR-10、およびMultiMNISTでの実験により、それらの有効性が検証されます。結果は、XnODRとXnIDRの両方が、ネットワークがより低いFLOPとより少ないパラメーターで高精度になるのに役立つことを示しています(たとえば、CIFAR-10での299Mパラメーターと312.04M FLOPで95.32%の精度)。
Although Capsule Network is powerful at defining the positional relationship between features in deep neural networks for visual recognition tasks, it is computationally expensive and not suitable for running on mobile devices. The bottleneck is in the computational complexity of the Dynamic Routing mechanism used between the capsules. On the other hand, XNOR-Net is fast and computationally efficient, though it suffers from low accuracy due to information loss in the binarization process. To address the computational burdens of the Dynamic Routing mechanism, this paper proposes new Fully Connected (FC) layers by xnorizing the linear projector outside or inside the Dynamic Routing within the CapsFC layer. Specifically, our proposed FC layers have two versions, XnODR (Xnorize the Linear Projection Linear Projector Outside Dynamic Routing) and XnIDR (Xnorize the Linear Projection Linear Projector Inside Dynamic Routing). To test the generalization of both XnODR and XnIDR, we insert them into two different networks, MobileNet V2 and ResNet-50. Our experiments on three datasets, MNIST, CIFAR-10, and MultiMNIST validate their effectiveness. The results demonstrate that both XnODR and XnIDR help networks to have high accuracy with lower FLOPs and fewer parameters (e.g., 95.32% correctness with 2.99M parameters and 312.04M FLOPs on CIFAR-10).
updated: Mon Jun 13 2022 01:35:46 GMT+0000 (UTC)
published: Sun Nov 21 2021 16:42:01 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アソシエイト