A novel capsule network based on residual module and attention mechanism
Capsule network(CapsNet) is an improvement of the classic convolutional neural network(CNN), which has achieved good results in the field of image classification. However, the high computational complexity of the routing process hinders them from building larger models. Fortunately, reducing the complexity of CapsNet routing through spatial attention mechanisms has achieved significantly good results. To this end, we propose a capsule network based on residual pose routing and attention mechanism. While reducing the routing calculation complexity, the accuracy can be improved. Experimental results on MNIST, CIFAR-10 and SmallNORB show that the proposed method can achieve better classification results compared with traditional methods.
Authors
- Lanfeng Zhou (ORCID: https://orcid.org/0000-0003-4819-867X)
- C. Qiu
Institutions
- Shanghai Institute of Technology (CN)
Publication Details
- Published
- 2024-11-21
- DOI
- https://doi.org/10.1109/iciibms62405.2024.10792813
- Primary Topic
- Text and Document Classification Technologies
- Type
- article
- Field-Weighted Citation Impact
- 0.00