Hyperbolic Contrastive Learning for Visual Representations beyond Objects

Songwei Ge, Shlok Kumar Mishra, Simon Kornblith, Chun-Liang Li, David Jacobs

Keywords: contrastive learning, self-supervised learning, Riemannian geometry, representation learning

Abstract: Despite the rapid progress in visual representation learning driven by self-/un-supervised methods, both objects and scenes have been primarily treated using the same lens. In this paper, we focus on learning representations for objects and scenes explicitly in the same space. Motivated by the observation that visually similar objects are close in the representation space, we argue that the scenes and objects should further follow a hierarchical structure based on their compositionality. To exploit such a structure, we propose a contrastive learning framework where a Euclidean loss is used to learn object representations and a hyperbolic loss is used to regularize scene representations according to the hierarchy. This novel hyperbolic objective encourages the scene-object hypernymy among the representations by optimizing the magnitude of their norms. We show that when pretraining on the COCO and OpenImages datasets, the hyperbolic loss improves downstream performance across multiple datasets and tasks, including image classification, object detection, and semantic segmentation. We also show that the properties of the learned representations allow us to solve various vision tasks that involve the interaction between scenes and objects in a zero-shot way.

TL;DR: We propose a contrastive learning framework that learns visual representation for both objects and scenes in the same representation space with a hierarchical topology preserved.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/hyperbolic-contrastive-learning-for-visual/code)