Capsule Nets for Complex Medical Image Segmentation Tasks

Total Page:16

File Type:pdf, Size:1020Kb

Capsule Nets for Complex Medical Image Segmentation Tasks Capsule Nets for Complex Medical Image Segmentation Tasks Shanmugapriya Survarachakan1, Jenny Stange Johansen1, Mathias Aarseth Pedersen1, Mahdi Amani1;2, and Frank Lindseth1 1 Norwegian University of Science and Technology, Department of Computer Science, Trondheim, Norway fshanmugapriya.survarachakan, mahdi.amani, [email protected] fjennysj, [email protected] 2 SimulaMET, Simula Research Laboratory, Oslo, Norway Abstract. In recent years, automatic image segmentation using deep learning has shown great potential. U-Net is commonly applied to the medical image segmentation task. Further, a new architecture called Seg- Caps, based on capsules networks has been introduced. Even though medical image segmentation has made great strides, it is still a complex task and continued research in this area is important. In this paper, the method from the Segcaps paper was implemented and trained on the same dataset and achieved a similar result. In addition, a Multi-SegCaps model, an EM-routing SegCaps model and a U-Net model able to seg- ment an arbitrary number of classes were developed. The models were implemented for the segmentation of single 2D slices using two of its neighboring slices on each side (five in total). These models were trained to perform left atrium and hippocampus (anterior & posterior) segmenta- tion on the MSD datasets. The performance of the Multi-SegCaps model and the EM-routing SegCaps model were compared to the U-Net model. The 2.5D U-Net model had an overall better performance on all the datasets achieving a Dice score of 91.39% for the left atrium segmenta- tion, and the Dice scores of 85.18% and 83.90% for the anterior and pos- terior hippocampus segmentation. The multi-class SegCaps model was also applied to two different datasets, showing the ability to segment sev- eral classes reasonably well. A Dice score of 68.2% was achieved for the left atrium and the Dice scores of 72.42% and 70.49% were achieved for the anterior and posterior hippocampus, respectively. The EM-SegCaps model was applied to the hippocampus dataset and it achieved a Dice score of 54.50% for the whole hippocampus and the Dice scores of 18.67% and 24.52% for the anterior and posterior hippocampus, respectively. Keywords: Medical Image Segmentation · U-Net · SegCaps · Capsule Network. Copyright c 2020 for this paper by its authors. Use permitted under Creative Com- mons License Attribution 4.0 International (CC BY 4.0). Colour and Visual Com- puting Symposium 2020, Gjøvik, Norway, September 16-17, 2020. 2 S. Survarachakan et al. 1 Introduction Medical image segmentation methods attempt to extract and locate the precise location of organs, tumors and other structures of interest, with the intention of aiding health professionals in making accurate diagnoses in a shorter amount of time. In recent years, convolutional neural networks (CNNs) have been com- monly used for image analysis tasks. A successful application of CNNs to the task of image segmentation was shown by [9] when they introduced fully convo- lutional networks where the skip connections were introduced to directly connect the opposing convolutional layers in the contraction and expansion paths. The network can work regardless of the original image size, without requiring any fixed number of units. The performance was improved by the elegant fully con- volutional U-Net architecture, published by [10]. The architecture is symmetric and the skip connections between the downsampling path and the upsampling path apply a concatenation operator instead of a sum. The network combines the location information from the downsampling path with the contextual informa- tion in the upsampling path and preserves the spatial resolution of the output. SegNet, an encoder-decoder network similar to FCN was introduced in [1]. It uses max unpooling in the upsampling path which eliminates the need for the network to learn the upsampling and provides a more efficient way to achieve segmentations than FCN. [6] created a new U-Net type architecture, showing state of the art performance on the task of multimodal brain tumor segmenta- tion using the BraTS dataset. The architecture was residual and worked directly with 3D patches of medical images. Though CNNs are popularly used they have some limitations due to its properties of rotational invariance and lack of instan- tiation parameters. So, many images with different views are needed to train the network. To overcome this issue, [11] presented a brand new capsule neural network architecture. Unlike CNNs, capsule networks are equivariant which can detect the objects if they are rotated and how many degrees it is rotated. This reduces the number of images required to train the network. [11] showed that capsule networks gave state of the art performance on the task of classifying highly overlapping digits in the MNIST dataset, proving CapsNet to be an effi- cient architecture for learning this type of problem. In [3], the capsule network consisting of two convolutional layers, two capsule layers, one fully connected layer, and two de-convolutional layers employing dynamic routing mechanism was proposed to segment Left ventricle (LV). Building on the work of capsule networks, [8] presented a method for performing binary image segmentation. The number of parameters needed by the architecture was reduced substantially by constraining the dynamic routing to capsules only in a defined kernel. The network showed good results for lung segmentation using the LUNA16 dataset. [5] claimed that the EM-routing algorithm solves most of the issues with the routing algorithm from [11]. It reduces the number of parameters needed, as well as using a metric of agreement that does not saturate with highly confi- dent predictions. It gave state of the art performance at detecting objects at novel viewpoints on the smallNORB dataset. In [2], a novel capsule is intro- duced which combines pose and appearance information encoded as capsules, Capsule Nets for Complex Medical Image Segmentation Tasks 3 named Matwo-Caps. Additionally, a new routing mechanism, i.e. dual routing, combining this two information was proposed. The method was evaluated for the semantic segmentation of the JSRT chest X-ray dataset. In [4], Fourier analysis and the circular Hough transform methods were applied to indicate the approxi- mate location of the LV and the capsule network with dynamic routing was used to precisely segment the LV. Thresholding and morphological processing were used as postprocessing methods to increase the accuracy of LV segmentation. In our work, the method and the results from [8] were reproduced and trained to adapt to other medical datasets and different modalities. The architecture was, as reported in the paper, capable of segmenting images of lungs. How- ever, the experiment also revealed that the architecture struggles to adapt to another medical dataset. A SegCaps architecture is further developed and eval- uated for multi-label image segmentation. It was experimentally shown to be able to perform segmentation of datasets with multiple label classes but strug- gles to segment structures in imbalanced datasets. Additionally, a framework for image segmentation using capsule networks and the EM-routing algorithm was developed and evaluated. As far as the authors know, an attempt of this has never been reported in the literature so far. Although the model did not give very good results, it showed that it was capable of segmenting some simple structures from the hippocampus dataset. A 2.5D architecture based on a U-Net variant was implemented and compared with the performance of the SegCaps, the Multi-SegCaps and the EM-SegCaps models on the cardiac and the hip- pocampus datasets from the Medical Segmentation Decathlon (MSD) datasets. This paper consists of five sections. The first section gives an introduction to the paper and the motivation behind it. The various methods and the datasets used in the paper are presented in the second section. In the third section, the experiments comparing different methods and their corresponding results are presented. The fourth section summarizes the discussion of the results and the fifth section presents the conclusion and suggests future work. 2 Methodology This section presents the different models developed and the datasets used in the experiments. 2.1 SegCaps The SegCaps architecture presented in [8] was used to perform binary segmen- tation of medical images (see Fig. 1). The architecture used is called SegCaps R3, referring to the use of three iterations of the dynamic routing algorithm in each capsule. A regular 2D convolution is performed on an input image to produce 16 fea- ture maps. The tensor consisting of 16 feature maps is reshaped in order to give it a new dimension of length one, such that the reshaped tensor now represents a single 16D capsule. The 16D capsule is forwarded to the primary capsule layer, 4 S. Survarachakan et al. Fig. 1. SegCaps architecture which is a regular convolutional capsule layer with one routing iteration, which returns predictions of two 16D capsules. Following the primary capsule layer, there are sets of two convolutional capsule layers at every layer for the remain- der of the contracting path. The first of the two operations is a 5x5 convolutional capsule layer with stride two, which also doubles the number of feature maps that are output. The second layer consists of 5x5 convolutional capsules without a stride. Output features from the latter operation being forwarded to the next level of the contracting
Recommended publications
  • RCNX: RESIDUAL CAPSULE NEXT by Arjun Narukkanchira Anilkumar
    RCNX: RESIDUAL CAPSULE NEXT by Arjun Narukkanchira Anilkumar A Thesis Submitted to the Faculty of Purdue University In Partial Fulfillment of the Requirements for the degree of Master of Science Department of Electrical and Computer Engineering Indianapolis, Indiana May 2021 THE PURDUE UNIVERSITY GRADUATE SCHOOL STATEMENT OF COMMITTEE APPROVAL Dr. Mohamed El-Sharkawy, Chair Department of Electrical and Computer Engineering Dr. Brian King Department of Electrical and Computer Engineering Dr. Maher Rizkalla Department of Electrical and Computer Engineering Approved by: Dr. Brian King 2 Dedicated to My Parents: Anilkumar N C and Swapna K A, My grandparents, teachers, friends, family, colleagues and all well wishers. 3 ACKNOWLEDGMENTS I would like to begin by thanking my parents and my teachers for everything. My thesis advisor, Dr. Mohamed El-Sharkawy, whose guidance and wisdom has been crucial throughout my thesis. I am grateful to have had Dr. Brian King and Dr. Maher Rizkalla for serving on my graduate committee and lending their expertise to my research. Additionally, I would like to thank Ms. Sherrie Tucker, who patiently assisted me throughout my graduate studies. I am grateful to Purdue School of Engineering & Technology for help fund my Thesis development. I am grateful to the IoT Collaboratory at IUPUI for accepting me as a member. The lab provided me with most of the hardware and software support for this research. I would like to thank my colleagues for their insights. I would like to acknowledge Lilly Endowment Inc. for their support to Indiana University Pervasive Technology Institute. 4 TABLE OF CONTENTS LIST OF TABLES ....................................
    [Show full text]
  • Capsule Network Performance with Autonomous Navigation
    International Journal of Artificial Intelligence and Applications (IJAIA), Vol.11, No.1, January 2020 CAPSULE NETWORK PERFORMANCE WITH AUTONOMOUS NAVIGATION Thomas Molnar and Eugenio Culurciello ECE, Purdue University, 610 Purdue Mall, West Lafayette, 47907, IN, USA ABSTRACT Capsule Networks (CapsNets) have been proposed as an alternative to Convolutional Neural Networks (CNNs). This paper showcases how CapsNets are more capable than CNNs for autonomous agent exploration of realistic scenarios. In real world navigation, rewards external to agents may be rare. In turn, reinforcement learning algorithms can struggle to form meaningful policy functions. This paper’s approach Capsules Exploration Module (Caps-EM) pairs a CapsNets architecture with an Advantage Actor Critic algorithm. Other approaches for navigating sparse environments require intrinsic reward generators, such as the Intrinsic Curiosity Module (ICM) and Augmented Curiosity Modules (ACM). Caps- EM uses a more compact architecture without need for intrinsic rewards. Tested using ViZDoom, the Caps- EM uses 44% and 83% fewer trainable network parameters than the ICM and Depth-Augmented Curiosity Module (D-ACM), respectively, for 1141% and 437% average time improvement over the ICM and D- ACM, respectively, for converging to a policy function across "My Way Home" scenarios. KEYWORDS Neural Networks, Autonomous, Navigation, Capsules Networks 1. INTRODUCTION Capsule Networks(CapsNets) were first presented by [1] to addressshortcomings ofConvolutionalNeural Networks (CNNs). The max pooling operation used with CNNs reduces thespatial size ofdata flowing through a network and thuslosesinformation. Thisleadsto the problem that “[i]nternaldata representation of a convolutional neural network does not take into account important spatial hierarchies between simple and complex objects" [1]. CapsNets resolve this by encoding the probability of detection of a feature as the length of their output vector.
    [Show full text]
  • Analysis of Capsule Networks for Image Classification
    International Conferences Computer Graphics, Visualization, Computer Vision and Image Processing 2021; Connected Smart Cities 2021; and Big Data Analytics, Data Mining and Computational Intelligence 2021 ANALYSIS OF CAPSULE NETWORKS FOR IMAGE CLASSIFICATION Evgin Goceri Biomedical Engineering Department, Engineering Faculty Akdeniz University, Dumlupinar Boulevard, 07058, Antalya, Turkey ABSTRACT Recently, the interest in convolutional neural networks have grown exponentially for image classification. Their success is based on the ability to learn hierarchical and meaningful image representation that results in a feature extraction technique which is general, flexible and can encode complex patterns. However, these networks have some drawbacks. For example, they need a large number of labeled data, lose valuable information (about the internal properties, such as shape, location, pose and orientation in an image and their relationships) in the pooling and are not able to encode deformation information. Therefore, capsule based networks have been introduced as an alternative to convolutional neural networks. Capsules are a group of neurons (logistic units) representing the presence of an entity and the vector indicating the relationship between features by encoding instantiation parameters. Unlike convolutional neural networks, maximum pooling layers are not employed in a capsule network, but a dynamic routing mechanism is applied iteratively in order to decide the coupling of capsules between successive layers. In other words, training between capsule layers is provided with a routing-by-agreement method. However, capsule networks and their properties to provide high accuracy for image classification have not been sufficiently investigated. Therefore, this paper aims (i) to point out drawbacks of convolutional networks, (ii) to examine capsule networks, (iii) to present advantages, weaknesses and strengths of capsule networks proposed for image classification.
    [Show full text]
  • Capsule Network Implementation on FPGA FPGA
    Journal of Pure & Applied Sciences www.Suj.sebhau.edu.ly ISSN 2521-9200 Received 23/03/2020 Revised 16/08/2020 Published online 05/10/2020 Capsule Network Implementation On FPGA *Salim A. Adrees , Ala A. Abdulrazeg Department of Computer Engineering, College of Engineering, Omar Al-Mukhtar University, Libya *Corresponding author: [email protected] Abstract A capsule neural network (CapsNet) is a new approach in artificial neural network (ANN) that produces a better hierarchical relationship. The performance of CapsNet on graphics processing unit (GPU) is considerably better than convolutional neural network (CNN) at recognizing highly overlapping digits in images. Nevertheless, this new method has not been designed as an accelerator on field programmable gate array (FPGA) to measure the speedup performance and compare it with the GPU. This paper aims to design the CapsNet module (accelerator) on FPGA. The performance between FPGA and GPU will be compared, mainly in terms of speedup and accuracy. The results show that training time on GPU using MATLAB is 789.091 s. Model evaluation accuracy is 99.79% and the validation accuracy is 98.53%. The time required to finish one routing algorithm iteration in MATLAB is 0.043622 s and in FPGA it takes 0000056s which means FPGA module is 67 times faster than GPU. Keywords: Artificial intelligent, Capsule neural network, Deep learning, FPGA, Image recognition. FPGA [email protected] ANN (CapsNet) CNN GPU FPGA MATLAB 70063 77097 9070071 0000056 00033522 MATLAB 59 FPGA I. Introduction Nowadays, Machine learning powers different fields between inputs and outputs. Unfortunately, CNN such as websites, cameras, and smartphones.
    [Show full text]
  • A Survey of Convolutional Neural Networks: Analysis, Applications, and Prospects
    > REPLACE THIS LINE WITH YOUR PAPER IDENTIFICATION NUMBER (DOUBLE-CLICK HERE TO EDIT) < 1 A Survey of Convolutional Neural Networks: Analysis, Applications, and Prospects Zewen Li, Wenjie Yang, Shouheng Peng, Fan Liu, Member, IEEE perceptron network cannot handle linear inseparable problems Abstract—Convolutional Neural Network (CNN) is one of the (such as XOR problems). In 1986, Hinton et al. [4] proposed a most significant networks in the deep learning field. Since CNN multi-layer feedforward network trained by the error back- made impressive achievements in many areas, including but not propagation algorithm—Back Propagation Network (BP limited to computer vision and natural language processing, it attracted much attention both of industry and academia in the past Network), which addressed some problems that single-layer few years. The existing reviews mainly focus on the applications of perceptron could not solve. In 1987, Waibel et al. [5] proposed CNN in different scenarios without considering CNN from a Time Delay Neural Network (TDNN) for speech recognition, general perspective, and some novel ideas proposed recently are which can be viewed as a one-dimensional convolutional neural not covered. In this review, we aim to provide novel ideas and network. Then, Zhang [6] proposed the first two-dimensional prospects in this fast-growing field as much as possible. Besides, convolutional neural network—Shift-invariant Artificial not only two-dimensional convolution but also one-dimensional Neural Network (SIANN). LeCun et al. [7] also constructed a and multi-dimensional ones are involved. First, this review starts with a brief introduction to the history of CNN. Second, we convolutional neural network for handwritten zip code provide an overview of CNN.
    [Show full text]
  • A Capsule Neural Network Based Model for Structural Damage Localization and Quantification Using Transmissibility Data
    UNIVERSIDAD DE CHILE FACULTAD DE CIENCIAS FÍSICAS Y MATEMÁTICAS DEPARTAMENTO DE INGENIERÍA MECÁNICA A CAPSULE NEURAL NETWORK BASED MODEL FOR STRUCTURAL DAMAGE LOCALIZATION AND QUANTIFICATION USING TRANSMISSIBILITY DATA TESIS PARA OPTAR AL GRADO DE MAGISTER EN CIENCIAS DE LA INGENIERÍA MENCIÓN MECÁNICA MEMORIA PARA OPTAR AL TÍTULO DE INGENIERO CIVIL MECÁNICO JOAQUÍN EDUARDO FIGUEROA BARRAZA PROFESOR GUÍA ENRIQUE LÓPEZ DROGUETT MIEMBROS DE LA COMISIÓN VIVIANA MERUANE NARANJO RUBÉN BOROSCHEK KRAUSKOPF SANTIAGO DE CHILE 2019 RESUMEN DE LA TESIS PARA OPTAR AL TÍTULO DE: Ingeniero Civil Mecánico y grado de Magister en Ingeniería Mecánica POR: Joaquín Eduardo Figueroa Barraza FECHA: Enero de 2019 PROF. GUÍA: Enrique López Droguett A CAPSULE NEURAL NETWORK BASED MODEL FOR STRUCTURAL DAMAGE LOCALIZATION AND QUANTIFICATION USING TRANSMISSIBILITY DATA Dentro de la ingeniería estructural, el monitoreo de condición usando diferentes tipos de sensores ha sido importante en la prevención de fallas y diagnóstico del estado de salud. El desafío actual es aprovechar al máximo las grandes cantidades de datos para entregar mediciones y predicciones precisas. Los algoritmos de aprendizaje profundo abordan estos problemas mediante el uso de datos para encontrar relaciones complejas entre ellos. Entre estos algoritmos, las redes neuronales convolucionales (CNN) han logrado resultados de vanguardia, especialmente cuando se trabaja con imágenes. Sin embargo, existen dos problemas principales: la incapacidad de reconocer imágenes rotadas como tales, y la inexistencia de jerarquías dentro de las imágenes. Para resolver estos problemas, se desarrollaron las redes de cápsulas (Capsule Networks), logrando resultados prometedores en problemas de tipo benchmark. En esta tesis, las Capsule Networks se modifican para localizar y cuantificar daños estructurales.
    [Show full text]
  • Examining the Benefits of Capsule Neural Networks
    1 Examining the Benefits of Capsule Neural Networks Arjun Punjabi1*, Jonas Schmid2*, and Aggelos K. Katsaggelos1 1Electrical and Computer Engineering, McCormick School of Engineering, Northwestern University 2Electrical Engineering, Institute for Communication Systems, HSR University of Applied Science Rapperswil *Both authors contributed equally to this work Abstract—Capsule networks are a recently developed class of applied to all classes of neural networks, not just those with neural networks that potentially address some of the deficiencies capsules. The interpretability of neural networks has always with traditional convolutional neural networks. By replacing the been a problem, and it is difficult to examine the benefit of standard scalar activations with vectors, and by connecting the artificial neurons in a new way, capsule networks aim to be capsules without a comparison to traditional CNN features. the next great development for computer vision applications. In an attempt to elucidate these capsules, this investiga- However, in order to determine whether these networks truly tion will begin by employing a deep visualization technique operate differently than traditional networks, one must look to generate images that visually represent the information at the differences in the capsule features. To this end, we contained in a capsule. This image can then be compared perform several analyses with the purpose of elucidating capsule features and determining whether they perform as described in to an image created in a similar fashion from a traditional the initial publication. First, we perform a deep visualization CNN, and the discrepancies between them can provide vi- analysis to visually compare capsule features and convolutional sual justification for the hypothesized benefits of capsule neural network features.
    [Show full text]
  • Handwritten Text Recognition Based on CNN Classification
    International Journal of Advanced Science and Technology Vol. 29, No. 9s, (2020), pp. 6870-6880 Handwritten Text Recognition Based On CNN Classification Kasturi Madineni, Department of Computer Science and Engineering, Saveetha School of Engineering, Saveetha Institute of Medical and Technical Sciences. [email protected] Prasanna Vasudevan, Assistant Professor, Department of Computer Science and Engineering, Saveetha School of Engineering, Saveetha Institute of Medical and Technical Sciences. [email protected] Abstract Handwritten images for characters, digits and special symbols has acknowledged more consideration in research community of pattern recognition due to vast tenders and uncertainty in learning new methods. Primarily, it includes two steps like character recognition and feature extraction, it mainly based on the classification of algorithm. The growth of handwritten charactersareapplicable mainly in cheques, medical prescriptions and tax returns etc., but identifying the handwritten characters are much difficult than the printed characters because each person have different handwriting style. The handwritten recognition mainly based on the convolutional neural network because it provides high accuracy and faster data processing. When compare to capsule neural network and Convolutional neural network the capsule neural network will work faster. The present research working on capsule neural network as classifier, MNIST as dataset with suitable parameters for testing and training for handwritten text recognition. Keywords: Handwritten recognition, convolutional neural network, capsule neural network, MNIST dataset, feature extraction and classification. INTRODUCTION In this paper written characters, digits and symbols have completely different vogue to acknowledge. Neural network is applied in several areas to acknowledge characters and gain additional accuracy. Capsule neural network square measure new sensational in Machine learning and therefore the superior performance on the Kaggle dataset.
    [Show full text]
  • A Comparative Study of Routing Methods in Capsule Networks
    Master of Science Thesis in Electrical Engineering Department of Electrical Engineering, Linköping University, 2019 A Comparative Study of Routing Methods in Capsule Networks Christoffer Malmgren Master of Science Thesis in Electrical Engineering A Comparative Study of Routing Methods in Capsule Networks Christoffer Malmgren LiTH-ISY-EX--19/5188--SE Supervisor: M.Sc Gustav Häger isy, Linköpings universitet Ph.D. Erik Ringaby SICK Linköping Examiner: Prof. Michael Felsberg isy, Linköpings universitet Computer Vision Laboratory Department of Electrical Engineering Linköping University SE-581 83 Linköping, Sweden Copyright © 2019 Christoffer Malmgren Abstract Recently, the deep neural network structure caps-net was proposed by Sabour et al. [11]. Capsule networks are designed to learn relative geometry between the features of a layer and the features of the next layer. The Capsule network’s main building blocks are capsules, which are represented by vectors. The idea is that each capsule will represent a feature as well as traits or subfeatures of that feature. This allows for smart information routing. Capsules traits are used to predict the traits of the capsules in the next layer, and information is sent to to next layer capsules on which the predictions agree. This is called routing by agreement. This thesis investigates theoretical support of new and existing routing al- gorithms as well as evaluates their performance on the MNIST [16] and CIFAR- 10 [8] datasets. A variation of the dynamic routing algorithm presented in the original paper [11] achieved the highest accuracy and fastest execution time. iii Acknowledgments I want to thank SICK Linköping for the opportunity to make this thesis.
    [Show full text]
  • Devanagari Script Recognition Using Capsule Neural Network
    International Journal of Computer Sciences and Engineering Open Access Research Paper Vol.-7, Issue-1, Jan 2019 E-ISSN: 2347-2693 Devanagari Script Recognition using Capsule Neural Network U.M. Sawant1*, R.K. Parkar2, S.L. Shitole3, S.P. Deore4 1,2,3Department of Computer Science, Modern Education Society’s College of Engineering, Pune, India 4Modern Education Society’s College of Engineering, Pune, India *Corresponding Author: [email protected], Tel.: +91-8975583271 Available online at: www.ijcseonline.org Accepted: 19/Jan/2019, Published: 31/Jan/2019 Abstract— Handwritten Devanagari Script Recognition has a lot of applications in the field of document processing, automation of postal services, automated cheque processing and so on. Several approaches have been proposed and experimented in the past depending on the type of features extracted and the ways of extracting them. In this paper, we proposed the use of Capsule Neural Networks (CapsNet) for the recognition of Handwritten Devanagari script, which is an advancement over the Convolutional Neural Networks (CNN) in terms of spatial relationships between the features. Capsule Neural Networks follow the principle of equivariance unlike the convolutional neural networks which follow the invariance property. CapsNet uses the dynamic routing by agreement method for passing data to higher capsules. CapsNet uses vector format for data representation. It can recognize similar characters in a more efficient manner as compared to CNN. Thus by using the advantages of CapsNet we are aiming to achieve better classification rate. We collected 100 samples of each of the 48 Devanagari characters and 10 Devanagari digits, and performed scaling, rotation and mirroring operations on these images.
    [Show full text]
  • Deformable Capsules for Object Detection
    Under review as a conference paper at ICLR 2021 DEFORMABLE CAPSULES FOR OBJECT DETECTION Anonymous authors Paper under double-blind review ABSTRACT Capsule networks promise significant benefits over convolutional networks by stor- ing stronger internal representations, and routing information based on the agree- ment between intermediate representations’ projections. Despite this, their success has been mostly limited to small-scale classification datasets due to their computa- tionally expensive nature. Recent studies have partially overcome this burden by locally-constraining the dynamic routing of features with convolutional capsules. Though memory efficient, convolutional capsules impose geometric constraints which fundamentally limit the ability of capsules to model the pose/deformation of objects. Further, they do not address the bigger memory concern of class-capsules scaling-up to bigger tasks such as detection or large-scale classification. In this study, we introduce deformable capsules (DeformCaps), a new capsule structure (SplitCaps), and a novel dynamic routing algorithm (SE-Routing) to balance compu- tational efficiency with the need for modeling a large number of objects and classes. We demonstrate that the proposed methods allow capsules to efficiently scale-up to large-scale computer vision tasks for the first time, and create the first-ever capsule network for object detection in the literature. Our proposed architecture is a one-stage detection framework and obtains results on MS COCO which are on-par with state-of-the-art one-stage
    [Show full text]
  • Face Identification Using Capsule Network with Small Data Set
    TALLINN UNIVERSITY OF TECHNOLOGY SCHOOL OF ENGINEERING Department of Electrical power Engineering and Mechatronics FACE IDENTIFICATION USING CAPSULE NETWORK WITH SMALL DATA SET NÄOTUVASTUS VÄIKESE ANDMEHULGAGA KAPSELVÕRGU ABIL MASTER THESIS Üliõpilane: Ahmed Amin Osman Üliõpilaskood: 165596 MAHM Juhendaja: Saleh Ragheb Saleh Alsaleh, Eng, Tallinn 2020 AUTHOR’S DECLARATION Hereby I declare, that I have written this thesis independently. No academic degree has been applied for based on this material. All works, major viewpoints and data of the other authors used in this thesis have been referenced. “.......” .................... 20….. Author: .............................. /signature / Thesis is in accordance with terms and requirements “.......” .................... 20…. Supervisor: …......................... /signature/ Accepted for defence “.......”....................20… . Chairman of theses defence commission: ................................................. /name and signature/ 2 Non-exclusive Licence for Publication and Reproduction of GraduationTthesis¹ I, Ahmed Amin Osman (12/20/1992 ) hereby 1. grant Tallinn University of Technology (TalTech) a non-exclusive license for my thesis FACE IDENTIFICATION USING CAPSULE NETWORK WITH SMALL DATA-SET supervised by Prof.Mart Tamre and Eng, Saleh Ragheb Saleh Alsaleh ____________________________________________________________, 1.1 reproduced for the purposes of preservation and electronic publication, incl. to be entered in the digital collection of TalTech library until the expiry of the term of copyright; 1.2 published via the web of TalTech, incl. to be entered in the digital collection of TalTech library until the expiry of the term of copyright. 1.3 I am aware that the author also retains the rights specified in clause 1 of this license. 2. I confirm that granting the non-exclusive license does not infringe third persons' intellectual property rights, the rights arising from the Personal Data Protection Act or rights arising from other legislation.
    [Show full text]