Online Unsupervised Learning of Visual Representations and Categories

Abstract

Real world learning scenarios involve a nonstationary distribution of classes with sequential dependencies among the samples, in contrast to the standard machine learning formulation of drawing samples independently from a fixed, typically uniform distribution. Furthermore, real world interactions demand learning on-the-fly from few or no class labels. In this work, we propose an unsupervised model that simultaneously performs online visual representation learning and few-shot learning of new categories without relying on any class labels. Our model is a prototype-based memory network with a control component that determines when to form a new class prototype. We formulate it as an online mixture model, where components are created with only a single new example, and assignments do not have to be balanced, which permits an approximation to natural imbalanced distributions from uncurated raw data. Learning includes a contrastive loss that encourages different views of the same image to be assigned to the same prototype. The result is a mechanism that forms categorical representations of objects in nonstationary environments. Experiments show that our method can learn from an online stream of visual input data and its learned representations are significantly better at category recognition compared to state-of-the-art self-supervised learning methods.

Technical report, 32 pages

References in corpus (16)

• Distilling the Knowledge in a Neural Network
• Prototypical Networks for Few-shot Learning
• Improved Baselines with Momentum Contrastive Learning
• Unsupervised Learning of Visual Features by Contrasting Cluster Assignments
• Prototypical Contrastive Learning of Unsupervised Representations
• Matterport3D: Learning from RGB-D Data in Indoor Environments
• Stochastic Video Generation with a Learned Prior
• Unsupervised Learning via Meta-Learning
• Continual Unsupervised Representation Learning
• Infinite Mixture Prototypes for Few-Shot Learning
• Unsupervised Meta-Learning For Few-Shot Image Classification
• Assume, Augment and Learn: Unsupervised Few-Shot Meta-Learning via Random Labels and Data Augmentation
• Self-Supervised Prototypical Transfer Learning for Few-Shot Classification
• S3VAE: Self-Supervised Sequential VAE for Representation Disentanglement and Data Generation
• Online Deep Clustering for Unsupervised Representation Learning
• Self-Supervised Representation Learning from Flow Equivariance