オブジェクトは、幾何学的に構成されたパーツのセットで構成されます。監視されていないカプセルオートエンコーダ(SCAE)を導入します。これは、部品間の幾何学的関係を明示的に使用してオブジェクトについて推論します。これらの関係は視点に依存しないため、このモデルは視点の変化に対してロバストです。 SCAEは2つの段階で構成されています。最初の段階では、モデルはパーツテンプレートの存在とポーズを画像から直接予測し、テンプレートを適切に配置することで画像の再構築を試みます。第2段階では、SCAEがいくつかのオブジェクトカプセルのパラメーターを予測します。これらのパラメーターは、パーツポーズの再構築に使用されます。このモデルの推論は、以前のカプセルネットワークとは異なり、既製のニューラルエンコーダによって償却および実行されます。オブジェクトカプセルの存在は、オブジェクトクラスの情報が豊富であり、SVHN(55%)およびMNIST(98.7%)の教師なし分類の最新の結果につながることがわかります。コードはhttps://github.com/google-research/google-research/tree/master/stacked_capsule_autoencodersで入手できます。
Objects are composed of a set of geometrically organized parts. We introduce an unsupervised capsule autoencoder (SCAE), which explicitly uses geometric relationships between parts to reason about objects. Since these relationships do not depend on the viewpoint, our model is robust to viewpoint changes. SCAE consists of two stages. In the first stage, the model predicts presences and poses of part templates directly from the image and tries to reconstruct the image by appropriately arranging the templates. In the second stage, SCAE predicts parameters of a few object capsules, which are then used to reconstruct part poses. Inference in this model is amortized and performed by off-the-shelf neural encoders, unlike in previous capsule networks. We find that object capsule presences are highly informative of the object class, which leads to state-of-the-art results for unsupervised classification on SVHN (55%) and MNIST (98.7%). The code is available at https://github.com/google-research/google-research/tree/master/stacked_capsule_autoencoders