Few-Shot Learning with Graph Neural Networks
View Researcher's Other CodesDisclaimer: The provided code links for this paper are external links. Science Nest has no responsibility for the accuracy, legality or content of these links. Also, by downloading this code(s), you agree to comply with the terms of use as set out by the author(s) of the code(s).
Please contact us in case of a broken link from here
Authors | Victor Garcia, Joan Bruna |
Journal/Conference Name | 6th International Conference on Learning Representations, ICLR 2018 - Conference Track Proceedings |
Paper Category | Artificial Intelligence |
Paper Abstract | We propose to study the problem of few-shot learning with the prism of inference on a partially observed graphical model, constructed from a collection of input images whose label can be either observed or not. By assimilating generic message-passing inference algorithms with their neural-network counterparts, we define a graph neural network architecture that generalizes several of the recently proposed few-shot learning models. Besides providing improved numerical performance, our framework is easily extended to variants of few-shot learning, such as semi-supervised or active learning, demonstrating the ability of graph-based models to operate well on 'relational' tasks. |
Date of publication | 2017 |
Code Programming Language | Multiple |
Comment |