• Home
  • Search
  • General Multi-label Image Classification with Transformers
  • Open Access IconOpen Access
  • Cite Icon335
  • https://doi.org/10.1109/cvpr46437.2021.01621Copy DOI Icon

General Multi-label Image Classification with Transformers

  • Jun 1, 2021
  • Jack Lanchantin +3 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Multi-label image classification is the task of predicting a set of labels corresponding to objects, attributes or other entities present in an image. In this work we propose the Classification Transformer (C-Tran), a general framework for multi-label image classification that leverages Transformers to exploit the complex dependencies among visual features and labels. Our approach consists of a Transformer encoder trained to predict a set of target labels given an input set of masked labels, and visual features from a convolutional neural network. A key ingredient of our method is a label mask training objective that uses a ternary encoding scheme to represent the state of the labels as positive, negative, or unknown during training. Our model shows state-of-the-art performance on challenging datasets such as COCO and Visual Genome. Moreover, because our model explicitly represents the label state during training, it is more general by allowing us to produce improved results for images with partial or extra label annotations during inference. We demonstrate this additional capability in the COCO, Visual Genome, News-500, and CUB image datasets.

Similar Papers
  • Conference Article
  • Citations1305

CNN-RNN: A Unified Framework for Multi-label Image Classification

  • Jun 01, 2016
  • Jiang Wang +5
  • Conference Article
  • Citations3

HSVLT: Hierarchical Scale-Aware Vision-Language Transformer for Multi-Label Image Classification

  • Oct 26, 2023
  • Shuyi Ouyang +8
  • Research Article
  • Citations45

Large-scale multi-label classification using unknown streaming images

  • Nov 01, 2019
  • Pattern Recognition
  • Yu Zhang +4
  • Book Chapter

Multiple Semantic Embedding with Graph Convolutional Networks for Multi-Label Image Classification

  • Jan 01, 2021
  • Tong Zhou +1
  • Research Article
  • Citations4

Multi-label classification approach for quranic verses labeling

  • Oct 01, 2021
  • Indonesian Journal of Electrical Engineering and Computer Science
  • Abdullahi Adeleke +4
  • Research Article

Multi-label fundus disease classification using dual-branch deep learning: an intelligent diagnosis framework inspired by traditional Chinese medicine Five Wheels theory

  • Mar 01, 2026
  • Digital Chinese Medicine
  • He Xin +7
  • Research Article
  • Citations10

Hierarchical GAN-Tree and Bi-Directional Capsules for multi-label image classification

  • Dec 14, 2021
  • Knowledge-Based Systems
  • Boyan Wang +4
  • Conference Article
  • Citations244

Improving Pairwise Ranking for Multi-label Image Classification

  • Jul 01, 2017
  • Yuncheng Li +2
  • Book Chapter
  • Citations2

Robust Multi-label Image Classification with Semi-Supervised Learning and Active Learning

  • Jan 01, 2015
  • Fuming Sun +2
  • Conference Article
  • Citations2

Combining Supervised Learning and Reinforcement Learning for Multi-Label Classification Tasks with Partial Labels

  • Jan 01, 2024
  • Zixia Jia +4
  • Conference Article
  • Citations18

Boosting Multi-Label Image Classification with Complementary Parallel Self-Distillation

  • Jul 01, 2022
  • Jiazhi Xu +5
  • Conference Article
  • Citations8

Multi-View Metric Learning for Multi-Label Image Classification

  • Sep 01, 2019
  • Mengying Zhang +2
  • Conference Article
  • Citations281

Visual Attention Consistency Under Image Transforms for Multi-Label Image Classification

  • Jun 01, 2019
  • Hao Guo +4
  • Conference Article
  • Citations8

Coarse to Fine: Multi-label Image Classification with Global/Local Attention

  • Sep 01, 2018
  • Fan Lyu +5
  • Conference Article
  • Citations2

Region-Awared Transformer with Asymmetric Loss in Multi-Label Classification

  • Jun 04, 2023
  • Lei Zhang +3
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.