arXiv reaDer
Momentum カプセル ネットワーク
Momentum Capsule Networks
カプセル ネットワークは、多くのコンピューター ビジョン タスクで有望な結果を達成したニューラル ネットワークのクラスです。ただし、ベースラインのカプセル ネットワークは、計算とメモリの要件が高いため、より複雑なデータセットで最先端の結果を得ることができませんでした。 Momentum Capsule Network (MoCapsNet) と呼ばれる新しいネットワーク アーキテクチャを提案することで、この問題に取り組みます。 MoCapsNets は Momentum ResNets に着想を得ています。Momentum ResNets は、可逆的な残差ビルディング ブロックを適用するネットワークの一種です。可逆ネットワークでは、バックプロパゲーション アルゴリズムでフォワード パスのアクティベーションを再計算できるため、これらのメモリ要件を大幅に削減できます。このホワイト ペーパーでは、カプセル ネットワークに可逆な残差ビルディング ブロックを適用する方法に関するフレームワークを提供します。 MoCapsNet は、MNIST、SVHN、CIFAR-10、および CIFAR-100 のベースライン カプセル ネットワークの精度よりもはるかに少ないメモリ使用量で優れていることを示します。ソース コードは https://github.com/moejoe95/MoCapsNet で入手できます。
Capsule networks are a class of neural networks that achieved promising results on many computer vision tasks. However, baseline capsule networks have failed to reach state-of-the-art results on more complex datasets due to the high computation and memory requirements. We tackle this problem by proposing a new network architecture, called Momentum Capsule Network (MoCapsNet). MoCapsNets are inspired by Momentum ResNets, a type of network that applies reversible residual building blocks. Reversible networks allow for recalculating activations of the forward pass in the backpropagation algorithm, so those memory requirements can be drastically reduced. In this paper, we provide a framework on how invertible residual building blocks can be applied to capsule networks. We will show that MoCapsNet beats the accuracy of baseline capsule networks on MNIST, SVHN, CIFAR-10 and CIFAR-100 while using considerably less memory. The source code is available on https://github.com/moejoe95/MoCapsNet.
updated: Thu Aug 25 2022 19:06:46 GMT+0000 (UTC)
published: Wed Jan 26 2022 17:53:18 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アソシエイト