Author(s): Ning, X (Ning, Xin); Tian, WJ (Tian, Weijuan); Li, WJ (Li, Weijun); Lu, YY (Lu, Yueyue); Nie, S (Nie, Shuai); Sun, LJ (Sun, Linjun); Chen, ZH (Chen, Ziheng)

Source: IEEE ACCESS Volume: 8 Pages: 59059-59068 DOI: 10.1109/ACCESS.2020.2982782 Published: 2020

Abstract: In order to improve the accuracy of capsule network in disentangled representation, and further expand its application in computer vision, a novel BDARS_CapsNet (bi-directional attention routing sausage capsule network) architecture is proposed in this paper. Firstly, the bi-directional routing, namely bottom-up and top-down attention is used to achieve information feed-forward and feedback mechanism, which contributes to describing the attributes of object entity more accurately and completely. Secondly, inspired by the concept of covering learning, the sausage measure model is introduced into the network. The sausage model measures both the similarities and differences of the capsules and projects them into a more complex curved surface, which makes it possible to approximate any nonlinear function with arbitrary precision and preserving the local responsiveness of capsule entity to the maximum. Finally, the BDARS_CapsNet combines the CNN (Convolutional Neural Network), bi-directional attention routing, and sausage measure into capsule network modeling, and makes full use of high-level category information and low-level vision information; as a result, the reconstruction and classification accuracy is accordingly improved. Experiments demonstrate the effectiveness of proposed information routing, sausage measure, and new framework. Furthermore, the proposed BDARS_CapsNet provides a foundation for future research on disentangled representation learning.

Accession Number: WOS:000549807900004

Author Identifiers:

Author        Web of Science ResearcherID        ORCID Number

Ning, Xin                  0000-0001-7897-1673

shuai, nie                  0000-0003-3141-1972

ISSN: 2169-3536

Full Text: https://ieeexplore.ieee.org/document/9044823