Research on image classification based on Capsnet

Capsnet is a very innovative and innovative deep learning network. This paper uses a capsule network to classify and test the MNIST and CIFAR10 datasets, respectively. And in order to compare with the traditional convolutional neural network, two architectures are designed. Then the experiment is carried out and the results are compared. By comparing and analyzing the experimental results, Capsnet has obvious advantages over the CNN method regardless of the MNIST dataset or the CIFAR10 dataset. Also, Capsnet will greatly reduce the number of parameters, which has the certain guiding significance in the classification of the image.

[1]  Chen Xu,et al.  MS-CapsNet: A Novel Multi-Scale Capsule Network , 2018, IEEE Signal Processing Letters.

[2]  Ulas Bagci,et al.  Capsules for Object Segmentation , 2018, ArXiv.

[3]  Premkumar Natarajan,et al.  CapsuleGAN: Generative Adversarial Capsule Network , 2018, ECCV Workshops.

[4]  Chanki Yu,et al.  Facial expression recognition using feature additive pooling and progressive fine-tuning of CNN , 2018 .

[5]  Shadi Albarqouni,et al.  Capsule Networks against Medical Imaging Data Challenges , 2018, CVII-STENT/LABELS@MICCAI.

[6]  Faliang Huang,et al.  Integrating Local CNN and Global CNN for Script Identification in Natural Scene Images , 2019, IEEE Access.

[7]  Miroslaw Bober,et al.  REMAP: Multi-Layer Entropy-Guided Pooling of Dense CNN Features for Image Retrieval , 2019, IEEE Transactions on Image Processing.

[8]  Jianbiao He,et al.  Cv-CapsNet: Complex-Valued Capsule Network , 2019, IEEE Access.

[9]  Xiangdong Huang,et al.  A Dilated CNN Model for Image Classification , 2019, IEEE Access.

[10]  Roland Vollgraf,et al.  Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms , 2017, ArXiv.

[11]  Sang-Hoon Oh,et al.  Deep CNNs Along the Time Axis With Intermap Pooling for Robustness to Spectral Variations , 2016, IEEE Signal Processing Letters.

[12]  Peng Wang,et al.  A Capsule Network for Traffic Speed Prediction in Complex Road Networks , 2018, 2018 Sensor Data Fusion: Trends, Solutions, Applications (SDF).

[13]  Geoffrey E. Hinton,et al.  Transforming Autoencoders , 2011 .

[14]  Geoffrey E. Hinton,et al.  Matrix capsules with EM routing , 2018, ICLR.

[15]  Hien Van Nguyen,et al.  Fast CapsNet for Lung Cancer Screening , 2018, MICCAI.

[16]  Rinat Mukhometzianov,et al.  CapsNet comparative performance evaluation for image classification , 2018, ArXiv.

[17]  Geoffrey E. Hinton,et al.  Dynamic Routing Between Capsules , 2017, NIPS.

[18]  Alex Krizhevsky,et al.  Learning Multiple Layers of Features from Tiny Images , 2009 .