CapsNet comparative performance evaluation for image classification

Abstract

Image classification has become one of the main tasks in the field of computer vision technologies. In this context, a recent algorithm called CapsNet that implements an approach based on activity vectors and dynamic routing between capsules may overcome some of the limitations of the current state of the art artificial neural networks (ANN) classifiers, such as convolutional neural networks (CNN). In this paper, we evaluated the performance of the CapsNet algorithm in comparison with three well-known classifiers (Fisher-faces, LeNet, and ResNet). We tested the classification accuracy on four datasets with a different number of instances and classes, including images of faces, traffic signs, and everyday objects. The evaluation results show that even for simple architectures, training the CapsNet algorithm requires significant computational resources and its classification performance falls below the average accuracy values of the other three classifiers. However, we argue that CapsNet seems to be a promising new technique for image classification, and further experiments using more robust computation resources and re-fined CapsNet architectures may produce better outcomes.

References in corpus (3)

• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• Capsule Network Performance on Complex Data
• Non-Parametric Transformation Networks

Cited by in corpus (9)

• An overview of artificial intelligence techniques for diagnosis of Schizophrenia based on magnetic resonance imaging modalities: Methods, challenges, and future works
• Few Shot Speaker Recognition using Deep Neural Networks
• Capsule Networks Need an Improved Routing Algorithm
• CapsAttacks: Robust and Imperceptible Adversarial Attacks on Capsule Networks
• DESCNet: Developing Efficient Scratchpad Memories for Capsule Network Hardware
• On Learning and Learned Data Representation by Capsule Networks
• Overhead-MNIST: Machine Learning Baselines for Image Classification
• Towards the Characterization of Representations Learned via Capsule-based Network Architectures
• Enabling Highly Efficient Capsule Networks Processing Through A PIM-Based Architecture Design