English

Capsule networks for low-data transfer learning

Computer Vision and Pattern Recognition 2018-04-27 v1

Abstract

We propose a capsule network-based architecture for generalizing learning to new data with few examples. Using both generative and non-generative capsule networks with intermediate routing, we are able to generalize to new information over 25 times faster than a similar convolutional neural network. We train the networks on the multiMNIST dataset lacking one digit. After the networks reach their maximum accuracy, we inject 1-100 examples of the missing digit into the training set, and measure the number of batches needed to return to a comparable level of accuracy. We then discuss the improvement in low-data transfer learning that capsule networks bring, and propose future directions for capsule research.

Keywords

Cite

@article{arxiv.1804.10172,
  title  = {Capsule networks for low-data transfer learning},
  author = {Andrew Gritsevskiy and Maksym Korablyov},
  journal= {arXiv preprint arXiv:1804.10172},
  year   = {2018}
}

Comments

11 pages, 10 figures

R2 v1 2026-06-23T01:37:15.004Z