Scaling Open-Vocabulary Image Segmentation with Image-Level Labels

By Golnaz Ghiasi, Xiuye Gu, Yin Cui, Tsung-Yi Lin. 2021

614 citations recorded by the source corpus.

We design an open-vocabulary image segmentation model to organize an image into meaningful regions indicated by arbitrary texts. Recent works (CLIP and ALIGN), despite attaining impressive open-vocabulary classification accuracy with image-level caption labels, are unable to segment visual concepts with pixels. We argue that these models miss an important step of visual grouping, which organizes pixels into groups before learning visual-semantic alignments. We propose OpenSeg to address the above issue while still making use of scalable image-level supervision of captions. First, it learns to propose segmentation masks for possible organizations. Then it learns visual-semantic alignments by aligning each word in a caption to one or a few predicted masks. We find the mask representations are the key to support learning image segmentation from captions, making it possible to scale up the dataset and vocabulary sizes. OpenSeg significantly outperforms the recent open-vocabulary method of LSeg by +19.9 mIoU on PASCAL dataset, thanks to its scalability.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Ziwei Liu
  2. Chen Change Loy
  3. Kai Chen
  4. Josh Tenenbaum
  5. Antonio Torralba
  6. Shuang Li
  7. Jiangmiao Pang
  8. Wenwei Zhang

In Computer Vision and Pattern Recognition

  1. Chen Change Loy
  2. Luc Van Gool
  3. Xiaogang Wang
  4. Ming-Hsuan Yang
  5. Dacheng Tao
  6. Jitendra Malik
  7. Andrea Vedaldi
  8. Trevor Darrell

Citations in the corpus