One-shot learning of object categories
Li Fei-FeiR. FergusP. Perona
Develops a Bayesian framework that leverages prior knowledge from previously learned classes to recognize novel visual object categories from as little as a single training image.
Learning visual categories of objects has long required hundreds or thousands of labeled training images per category, along with manual alignment of examples, making it impractical to build systems that recognize the tens of thousands of categories humans handle. The paper addresses this bottleneck by testing whether prior knowledge from a small number of already-learned categories can enable useful models of new categories from just one or a few images.
The work set out to evaluate a Bayesian method that represents object categories as probabilistic constellation models of shape and appearance and encodes knowledge from earlier categories as a prior distribution over model parameters. The prior is updated with limited new observations to produce a posterior that supports detection and recognition. Experiments used weakly supervised training on four standard categories and then on a 101-category collection assembled from web images, with performance measured by equal-error rates on held-out test images against background scenes. The Bayesian approach was compared directly with maximum-likelihood and maximum-a-posteriori baselines using identical data splits and repeated random trials.
With one training image the Bayesian method reached roughly 70–85 percent detection on several categories where the other two methods could not produce a model at all; with three to six images it typically reached 80–95 percent on well-represented categories and remained above chance on most of the 101. The same prior, derived from only three unrelated categories, improved results across widely varying objects, and full shape-plus-appearance models outperformed either cue alone. Discrimination among all 101 categories rose from about 10 percent correct with three examples to roughly 18 percent with fifteen—still far above the 1 percent random baseline.
These results show that prior knowledge can substitute for large training sets, lowering the cost and supervision needed to expand visual recognition systems and bringing machine performance closer to the rapid category acquisition observed in children. Because the method works with unaligned web images, it also reduces the practical barrier to scaling recognition to thousands of categories.
The main limitations are the use of a simple prior built from only three source categories, a four-part single-viewpoint model that omits occlusion, and uneven feature detection quality across categories. Performance therefore varies and remains below the level needed for many applications. Further gains are likely from richer priors learned from dozens of categories, incremental updating of the prior as new categories arrive, and modest increases in model complexity.
- Paper: A Bayesian hierarchical model for learning natural scene categories, Li Fei-Fei et al. (2005). This earlier hierarchical scene categorization work establishes the foundational Bayesian modeling framework and codebook ideas adapted for one-shot learning across diverse categories.
- Paper: Matching Networks for One Shot Learning, Oriol Vinyals et al. (2016). Building directly on the one-shot learning paradigm, this work advances the field by introducing metric spaces and attention mechanisms for rapid classification without parameter updates.
- Paper: Prototypical Networks for Few-shot Learning, Jake Snell et al. (2017). This paper extends few-shot classification concepts into simpler, highly effective embedding spaces defined by class prototypes.
- Paper: DeViSE: A Deep Visual-Semantic Embedding Model, Andrea Frome et al. (2013). Extending beyond visual-only priors, this research applies embedding models to bridge visual and semantic spaces for zero-shot object recognition.
