基于成对乘积的胶囊网络高效一致性机制
计算机视觉与模式识别
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