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.
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