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.

Dilated Capsule Network for Brain Tumor Type Classification Via MRI Segmented Tumor Region

Brain Tumor Type Classification via Capsule Networks

Capsule Networks for Brain Tumor Classification based on MRI Images and Course Tumor Boundaries

MLAF-CapsNet: Multi-lane Atrous Feature Fusion Capsule Network with Contrast Limited Adaptive Histogram Equalization for Brain Tumor Classification from MRI Images

MRI-Based Brain Tumor Classification Using a Dilated Parallel Deep Convolutional Neural Network

Brain Tumor Diagnosis Using Machine Learning, Convolutional Neural Networks, Capsule Neural Networks and Vision Transformers, Applied to MRI: A Survey

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

An improved capsule network for glioma segmentation on MRI images: A curriculum learning approach

Artificial Intelligence-Based Brain Tumor Segmentation Using Adaptive Hybrid CNN and Classification by Multi-Scale Dilated MobileNet with Attention Mechanism for MRI Images

Classification of Benign and Malignancy in Lung Cancer Using Capsule Networks with Dynamic Routing Algorithm on Computed Tomography Images

Brain Tumor Classification from MRI Using Image Enhancement and Convolutional Neural Network Techniques

Enhancing brain tumor classification in MRI scans with a multi-layer customized convolutional neural network approach

Robust clinical applicable CNN and U-Net based algorithm for MRI classification and segmentation for brain tumor

Advancing Brain Tumor Detection: A Thorough Investigation of CNNs, Clustering, and SoftMax Classification in the Analysis of MRI Images

Enhancing Brain Tumor Classification in MRI: Leveraging Deep Convolutional Neural Networks for Improved Accuracy

MRI brain tumor detection and classification using parallel deep convolutional neural networks

Enhancement of brain tumor classification from MRI images using multi-path convolutional neural network with SVM classifier

Brain Tumor MRI Classification Using a Novel Deep Residual and Regional CNN

Brain Tumor Classification of MRI Images Using Deep Convolutional Neural Network

UCapsNet: A Two-Stage Deep Learning Model Using U-Net and Capsule Network for Breast Cancer Segmentation and Classification in Ultrasound Imaging

Brain Tumor Classification Using Convolutional Neural Network and Feature Extraction