Tencent ML-Images: A Large-Scale Multi-Label Image Database for Visual Representation Learning

By Baoyuan Wu, Weidong Chen, Yanbo Fan, Yong Zhang, Jinlong Hou, T. Zhang. 2019

95 citations recorded by the source corpus.

In existing visual representation learning tasks, deep convolutional neural networks (CNNs) are often trained on images annotated with single tag, such as ImageNet. However, single tag annotation cannot describe all important contents of one image, and some useful visual information may be wasted during training. In this work, we propose to train CNNs from images annotated with multiple tags, to enhance the quality of visual representation of the trained CNN model. To this end, we build a large-scale multi-label image database with 18M images and 11K categories, dubbed Tencent ML-Images. We efficiently train the ResNet-101 model with multi-label outputs on Tencent ML-Images, taking 90 hours for 60 epochs, based on a large-scale distributed deep learning framework, i.e., TFplus. The good quality of the visual representation of the Tencent ML-Images checkpoint is verified through three transfer learning tasks, including single-label image classification on ImageNet and Caltech-256, object detection on PASCAL VOC 2007, and semantic segmentation on PASCAL VOC 2012. The Tencent ML-Images database, the checkpoints of ResNet-101, and all the training codes have been released at https://github.com/Tencent/tencent-ml-images. It is expected to promote other vision tasks in the research and industry community.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Evaluation & Benchmarks

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. Mohit Bansal
  2. Quoc V. Le
  3. Jianfeng Gao
  4. Dacheng Tao
  5. Trevor Darrell
  6. Ling Shao
  7. Ishan Misra
  8. Sergey Levine

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Ming-Hsuan Yang
  4. Trevor Darrell
  5. Xiaogang Wang
  6. Ling Shao
  7. Andrew Zisserman
  8. Alan L. Yuille

In Evaluation & Benchmarks

  1. Soravit Changpinyo
  2. Radu Soricut
  3. Bo Pang
  4. Piyush Sharma
  5. Baoyuan Wu
  6. Ling Shao
  7. Fumin Shen
  8. Fan Zhu

Citations in the corpus