中文

基于成对乘积的胶囊网络高效一致性机制

计算机视觉与模式识别 2020-04-02 v1

摘要

胶囊网络(CapsNets)能够通过“路由-by-一致性(routing-by-agreement)”机制对视觉层次关系进行建模。本文受因子分解机(FMs)特征交互的启发,提出一种成对一致性机制来构建胶囊。所提方法具有低得多的计算复杂度。我们进一步提出了一种新的 CapsNet 架构,它结合了残差网络在表示低级视觉特征方面的优势以及 CapsNets 在建模部分到整体关系方面的优势。我们基于原始 CapsNet 和我们所提的架构,开展了全面的实验来比较路由算法,包括动态路由、EM 路由以及我们提出的 FM 一致性,结果表明我们的方法在各种情况下均实现了优异的性能与效率。

关键词

引用

@article{arxiv.2004.00272,
  title  = {An Efficient Agreement Mechanism in CapsNets By Pairwise Product},
  author = {Lei Zhao and Xiaohui Wang and Lei Huang},
  journal= {arXiv preprint arXiv:2004.00272},
  year   = {2020}
}

备注

Accepted to ECAI 2020