Abstract:Convolutional Neural Networks (CNNs) are frequently used algorithms because of their propensity to learn relevant and hierarchical features through their feature extraction technique. However, the availability of enormous volumes of data in various variations is crucial for their performance. Capsule networks (CapsNets) perform well on a small amount of data but perform poorly on complex images. To address this, we proposed a new Capsule Network architecture called Tri Texton-Dense CapsNet (TTDCapsNet) for better complex and medical image classification. The TTDCapsNet is made up of three hierarchic blocks of Texton-Dense CapsNet (TDCapsNet) models. A single TDCapsNet is a CapsNet architecture composed of a texton detection layer to extract essential features, which are passed onto an eight-layered block of dense convolution that further extracts features, and then the output feature map is given as input to a Primary Capsule (PC), and then to a Class Capsule (CC) layer for classification. The resulting feature map from the first PC serves as input into the second-level TDCapsNet, and that from the second PC serves as input into the third-level TDCapsNet. The routing algorithm receives feature maps from each PC for the various CCs. Routing the concatenation of the three PCs creates an additional CC layer. All these four feature maps combined, help to achieve better classification. On fashion-MNIST, CIFAR-10, Breast Cancer, and Brain Tumor datasets, the proposed model is evaluated and achieved validation accuracies of 94.90%, 89.09%, 95.01%, and 97.71% respectively. Findings from this work indicate that TTDCapsNet outperforms the baseline and performs comparatively well with the state-of-the-art CapsNet models using different performance metrics. This work clarifies the viability of using Capsule Network on complex tasks in the real world. Thus, the proposed model can be used as an intelligent system, to help oncologists in diagnosing cancerous diseases and administering treatment required.

COMPARATIVE STUDY OF CAPSULE NEURAL NETWORK IN VARIOUS APPLICATIONS

Capsule Network with Its Limitation, Modification, and Applications—A Survey

Examining the Benefits of Capsule Neural Networks

Capsule networks for computer vision applications: a comprehensive review

Capsule networks for image classification: A review

A Survey on Capsule Networks: Evolution, Application, and Future Development

CapsNet comparative performance evaluation for image classification

Overview of Capsule Neural Networks

Learning with Capsules: A Survey

Capsule Network Performance on Complex Data

RS-CapsNet: an Advanced Capsule Network.

A Context-aware Capsule Network for Multi-label Classification

Subspace Capsule Network

Pushing the Limits of Capsule Networks

Object-centric Learning with Capsule Networks: A Survey

DeepCaps: Going Deeper With Capsule Networks

Handwritten Indic Character Recognition using Capsule Networks

TTDCapsNet: Tri Texton-Dense Capsule Network for complex and medical image recognition

Hierarchical Object-Centric Learning with Capsule Networks

Parallel Capsule Networks for Classification of White Blood Cells

Capsule Networks against Medical Imaging Data Challenges