面向类胶囊判别力的路由算法
机器学习
2021-03-09 v1 计算机视觉与模式识别
摘要
胶囊网络近来被提出作为现代神经网络架构的一种替代方案。神经元被胶囊单元取代，后者以归一化向量或矩阵表示特定特征或实体。较低层胶囊的激活通过路由链接影响后续胶囊的行为，这些链接在训练期间通过特定路由算法构建。我们讨论了动态路由算法中的协议路由方案，该方案在某些情况下会使网络偏离最优。为获得更好且更快的收敛，我们提出一种融入正则化二次规划问题的路由算法，该问题可被高效求解。特别地，所提路由算法直接针对类胶囊的判别力，从而对输入实例做出正确决策。我们在MNIST、MNIST-Fashion和CIFAR-10上开展实验，并展现出与现有胶囊网络相比具有竞争力的分类结果。
引用
@article{arxiv.2103.04278,
title = {Routing Towards Discriminative Power of Class Capsules},
author = {Haoyu Yang and Shuhe Li and Bei Yu},
journal= {arXiv preprint arXiv:2103.04278},
year = {2021}
}
备注
6 pages,4 figures