2 July 2021 Deep capsule network regularization based on generative adversarial network framework
Kun Sun, Haixia Xu, Liming Yuan, Xianbin Wen
Author Affiliations +
Abstract

Deep capsule networks have more capsule layers, which makes their performance better on complex images. However, with the increase of layers, overfitting will become more serious. Image reconstruction is an effective regularization method for capsule networks. To improve it, we propose an adversarial decoder that introduces the generative adversarial network framework into the reconstruction process to implement learnable reconstruction losses. This architecture consists of three parts: a deep capsule network, a decoder, and a discriminator. The deep capsule network extracts feature capsules from input images, which are then reconstructed by the decoder. The discriminator is the learnable reconstruction loss function that evaluates the similarity between reconstructed images and input images. Minimizing this learnable reconstruction loss and mean square error of images provides a regularization effect for the deep capsule network. Experimental results show that our models have a competitive performance of regularization on CIFAR10, CIFAR100, and FMNIST.

© 2021 SPIE and IS&T 1017-9909/2021/$28.00© 2021 SPIE and IS&T
Kun Sun, Haixia Xu, Liming Yuan, and Xianbin Wen "Deep capsule network regularization based on generative adversarial network framework," Journal of Electronic Imaging 30(4), 040501 (2 July 2021). https://doi.org/10.1117/1.JEI.30.4.040501
Received: 13 April 2021; Accepted: 22 June 2021; Published: 2 July 2021
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Gallium nitride

Network architectures

Image classification

Image restoration

Lawrencium

Performance modeling

Convolution

RELATED CONTENT

Analysis of adversarial attack based on FGSM
Proceedings of SPIE (January 12 2023)
Improved SRGAN model
Proceedings of SPIE (August 09 2023)

Back to Top