arXiv reaDer
同変ブリッジSO(2)-グラフ畳み込みネットワークを使用した不変表現学習
Equivariance-bridged SO(2)-Invariant Representation Learning using Graph Convolutional Network
回転に対してロバストになるように畳み込みニューラルネットワーク(CNN)をトレーニングすることは、ほとんどの場合、データ拡張を使用して行われました。この論文では、ネットワークの構造的な回転不変性を実現することにより、データ拡張への依存を減らすことを奨励するために、研究の方向性に関する別の進歩的なビジョンが強調されています。深い同変ブリッジSO(2)不変ネットワークは、そのようなビジョンをエコーするために提案されています。まず、自重最近傍グラフ畳み込みネットワーク(SWN-GCN)を提案し、画像のグラフ表現にグラフ畳み込みネットワーク(GCN)を実装して、回転同変表現を取得します。GCNは、スペクトルグラフよりも深いネットワークを構築するのに適しているためです。畳み込みベースのアプローチ。次に、SWN-GCNから取得した頂点の同変セットに対して、高次元表現を集約するのに適した順列不変演算であるグローバル平均プーリング(GAP)を使用して、最終的に不変表現を取得します。私たちの方法は、回転したMNISTおよびCIFAR-10画像で最先端の画像分類パフォーマンスを実現します。この場合、モデルは拡張されていないデータセットのみでトレーニングされます。表現の不変性に対する定量的検証は、回転に対するSWN-GCNの深い表現の強い不変性も示しています。
Training a Convolutional Neural Network (CNN) to be robust against rotation has mostly been done with data augmentation. In this paper, another progressive vision of research direction is highlighted to encourage less dependence on data augmentation by achieving structural rotational invariance of a network. The deep equivariance-bridged SO(2) invariant network is proposed to echo such vision. First, Self-Weighted Nearest Neighbors Graph Convolutional Network (SWN-GCN) is proposed to implement Graph Convolutional Network (GCN) on the graph representation of an image to acquire rotationally equivariant representation, as GCN is more suitable for constructing deeper network than spectral graph convolution-based approaches. Then, invariant representation is eventually obtained with Global Average Pooling (GAP), a permutation-invariant operation suitable for aggregating high-dimensional representations, over the equivariant set of vertices retrieved from SWN-GCN. Our method achieves the state-of-the-art image classification performance on rotated MNIST and CIFAR-10 images, where the models are trained with a non-augmented dataset only. Quantitative validations over invariance of the representations also demonstrate strong invariance of deep representations of SWN-GCN over rotations.
updated: Fri Jun 18 2021 08:37:45 GMT+0000 (UTC)
published: Fri Jun 18 2021 08:37:45 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アソシエイト