Abstract:Early diagnosis of patients' disease is crucial since it helps doctors and patients devise a treatment plan. Therefore, recognizing medical images using Artificial intelligence-based deep learning techniques has recently increased. Capsule Network (CapsNet) has promising methods in visual tasks due to its ability to keep a high relationship of spatial information compared to convolutional neural networks (CNNs). However, CapsNet faces a critical problem with a complex image background that limits its performance. The traditional CapsNet adopts a standalone convolution (SC) as a feature extractor, Softmax function for normalization of coupling coefficient, and dynamic routing procedure to allow active capsules to perform predictions leading to activation of high-level capsules. The SC is not an effective feature extractor, and SoftMax impedes capsules from distributing optimal coupling coefficient during routing. This paper proposes a CapsNet architecture called SqueezeCapsNet that integrates SqueezeNet and CapsNet to achieve effective feature extraction and fewer parameters. A new squash function named parametric squash function (PSF) was proposed to reduce non-informative capsules and promote discriminative capsules. To the best of our knowledge in literature, we are the first to integrate SqueezeNet into CapsNet. We evaluate our framework on two medical image datasets; Brain tumor and Lung & Colon cancer datasets. Additionally, datasets with varied backgrounds; MNIST, fashion-MNIST, CIFAR-10 were used to evaluate the robustness and generalizability of the model. The SqueezeCapsNet produces 94.85%, 99.76%, 99.87 % , 93.49 % , and 82.45 % on Brain tumor, Lung & Colon Cancer, MNIST, fashion-MNIST, and CIFAR-10 datasets, respectively. Experimental results show that the proposed architecture's compression techniques significantly provide fewer parameters while enhancing stability and accuracy across all the evaluation metrics. Our results show that our method improves CapsNet and can be adopted as a computer-aided diagnostic method to support the diagnosis of medical image tasks.

Capsules for Biomedical Image Segmentation

Seg-CapNet: A Capsule-Based Neural Network for the Segmentation of Left Ventricle from Cardiac Magnetic Resonance Imaging

CapsuleVOS: Semi-Supervised Video Object Segmentation Using Capsule Routing

Capsule networks for segmentation of small intravascular ultrasound image datasets

A Coarse-to-fine Full Attention Guided Capsule Network for Medical Image Segmentation

SqueezeCapsNet: enhancing capsule networks with squeezenet for holistic medical and complex images

Lung CT Image Segmentation Based on Capsule Network

SS-3DCapsNet: Self-supervised 3D Capsule Networks for Medical Segmentation on Less Labeled Data

Deformable Capsules for Object Detection

Retinal Vessel Segmentation with Deep Graph and Capsule Reasoning

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

Capsule Networks against Medical Imaging Data Challenges

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

HistoSegCap: Capsules for Weakly-Supervised Semantic Segmentation of Histological Tissue Type in Whole Slide Images

Coronary artery segmentation from intravascular optical coherence tomography using deep capsules

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

3D-EffiViTCaps: 3D Efficient Vision Transformer with Capsule for Medical Image Segmentation

Multi-Task Classification and Segmentation for Explicable Capsule Endoscopy Diagnostics

Attention-gated 3D CapsNet for robust hippocampal segmentation

Attentive Octave Convolutional Capsule Network for Medical Image Classification

Feature Amplification Capsule Network for Complex Images.