arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.00272cs.CV

An Efficient Agreement Mechanism in CapsNets By Pairwise Product

TalkingData;Inception Institute of Artificial Intelligence(Inception人工智能研究院)

Lei Zhao, Xiaohui Wang, Lei Huang

英文摘要:

Capsule networks (CapsNets) are capable of modeling visual hierarchical relationships, which is achieved by the "routing-by-agreement" mechanism. This paper proposes a pairwise agreement mechanism to build capsules, inspired by the feature interactions of factorization machines (FMs). The proposed method has a much lower computation complexity. We further proposed a new CapsNet architecture that combines the strengths of residual networks in representing low-level visual features and CapsNets in modeling the relationships of parts to wholes. We conduct comprehensive experiments to compare the routing algorithms, including dynamic routing, EM routing, and our proposed FM agreement, based on both architectures of original CapsNet and our proposed one, and the results show that our method achieves both excellent performance and efficiency under a variety of situations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑