arXiv reaDer
変分ベイズによるカプセルルーティング
Capsule Routing via Variational Bayes
 カプセルネットワークは、最近提案されたタイプのニューラルネットワークであり、形状認識タスクに挑戦する際に代替手段よりも優れていることが示されています。カプセルネットワークでは、スカラーニューロンは、エントリがオブジェクトの異なるプロパティを表すカプセルベクトルまたはカプセルに置き換えられます。オブジェクトとその部分の間の関係は、訓練可能な視点不変の変換行列を介して学習され、特定のオブジェクトの存在は、その部分からの票の一致のレベルによって決定されます。この相互作用はカプセルレイヤー間で発生し、合意によるルーティングと呼ばれるプロセスです。本論文では、変分ガウス分布の混合に適合する変分ベイズから導出された新しいカプセルルーティングアルゴリズムを提案し、カプセルネットワークをCapsule-VAEに変換できることを示します。ベイズのアプローチは、カプセルポーズパラメーターの不確実性をモデル化することにより、分散崩壊などのMLEベースモデルの固有の弱点に対処します。以前に報告されたよりも50%少ないカプセルを使用して、smallNORBで最先端のパフォーマンスを達成し、CIFAR-10、Fashion-MNIST、SVHNで競争力のあるパフォーマンスを達成し、MNISTでaffNISTの一般化を大幅に改善しました。
Capsule networks are a recently proposed type of neural network shown to outperform alternatives in challenging shape recognition tasks. In capsule networks, scalar neurons are replaced with capsule vectors or matrices, whose entries represent different properties of objects. The relationships between objects and their parts are learned via trainable viewpoint-invariant transformation matrices, and the presence of a given object is decided by the level of agreement among votes from its parts. This interaction occurs between capsule layers and is a process called routing-by-agreement. In this paper, we propose a new capsule routing algorithm derived from Variational Bayes for fitting a mixture of transforming gaussians, and show it is possible transform our capsule network into a Capsule-VAE. Our Bayesian approach addresses some of the inherent weaknesses of MLE based models such as the variance-collapse by modelling uncertainty over capsule pose parameters. We outperform the state-of-the-art on smallNORB using 50% fewer capsules than previously reported, achieve competitive performances on CIFAR-10, Fashion-MNIST, SVHN, and demonstrate significant improvement in MNIST to affNIST generalisation over previous works.
updated: Tue Dec 03 2019 16:49:01 GMT+0000 (UTC)
published: Mon May 27 2019 19:08:42 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アソシエイト