One Shot Learning using Memory-Augmented Neural Networks in Tensorflow.
Update: added support for Tensorflow v1*.
Tensorflow implementation of the paper One-shot Learning with Memory-Augmented Neural Networks.
Current Progress of Implementation:
- Utility Functions:
- Image Handler
- Metrics (Accuracy)
- Similarities (Cosine Similarity)
- LSTM Controller and Memory Unit
- Batch Generators
- Omniglot Tester Code
- Unsupervised Feature Learning through Autoencoders
- Cattle/New Born Recognition
The benchmark dataset is Omniglot dataset. All the datasets should be placed in the data/ folder.
Adam Santoro, Sergey Bartunov, Matthew Botvinick, Daan Wierstra, Timothy Lillicrap, One-shot Learning with Memory-Augmented Neural Networks, [arXiv]