如何加速胶囊网络中的胶囊卷积
人工智能
2021-04-07 v1
摘要
如何提升胶囊网络(CapsNets)中路由过程的效率已被广泛研究。然而，胶囊卷积的效率在很大程度上被忽视了。胶囊卷积以胶囊而非神经元作为基本计算单元，使其与当前深度学习框架的优化方案不兼容。因此，在这些框架中胶囊卷积通常非常缓慢。我们观察到胶囊卷积可视为“多个小矩阵相乘”的操作加上基于张量的组合。基于该观察，我们利用 CUDA API 开发了两种加速方案，并在一个定制的 CapsNet 上进行了测试。结果表明我们的方案实现了 4 倍加速。
引用
@article{arxiv.2104.02621,
title = {How to Accelerate Capsule Convolutions in Capsule Networks},
author = {Zhenhua Chen and Xiwen Li and Qian Lou and David Crandall},
journal= {arXiv preprint arXiv:2104.02621},
year = {2021}
}