arrow
Return

Deep Tensor Capsule Network

delete2020-01-01
delete18
delete
OA
AI
K
Kun Sun
L
Liming Yuan *
H
Haixia Xu
温显斌 cover
温显斌 (Xianbin Wen)
DOI:10.1109/ACCESS.2020.2996282delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Capsule network is a promising model in computer vision. It has achieved excellent results on simple datasets such as MNIST, but the performance deteriorates as data becomes complicated. In order to address this issue, we propose a deep capsule network in this paper. To deepen the capsule network, we present a new tensor capsule based routing algorithm and the corresponding convolution operation. Compared to vector capsules, tensor capsules can capture more instance-level information. Together, the relevant convolution operation is beneficial for reducing the amount of parameters in the routing process. Furthermore, we propose a dropout mechanism for vectors and tensors in order to alleviate the potential overfitting problem. Finally, we also inject the multi-scale capsules of the middle layers into a multi-scale decoder to pursue more details of an image and more clear reconstructed image. Experimental results on CIFAR10, Fashion-MNIST, and SVHN demonstrate that the proposed deep tensor network can achieve very competitive performance compared to other state-of-the-art capsule networks.
Keywords:
Tensile stress
Routing
Convolution
Image reconstruction
Decoding
Heuristic algorithms
Kernel
Capsule network
dynamic routing
dropout
CNNs
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

T
Tianjin University of Technology
Scholars:
8.8K
Papers: 5.9K
Citations: 1.0W