Label2Label: A Language Modeling Framework for Multi-Attribute Learning

TL;DR

Label2Label introduces a novel language modeling framework for multi-attribute learning, leveraging attribute correlations by treating attribute labels as words in a sentence, inspired by NLP pre-training techniques.

Contribution

It is the first to apply language modeling to multi-attribute prediction, achieving state-of-the-art results without task-specific modifications.

Findings

Achieves state-of-the-art results on three multi-attribute tasks.

Effectively models attribute correlations through masked language modeling.

Does not rely on task-specific priors or complex network designs.

Abstract

Objects are usually associated with multiple attributes, and these attributes often exhibit high correlations. Modeling complex relationships between attributes poses a great challenge for multi-attribute learning. This paper proposes a simple yet generic framework named Label2Label to exploit the complex attribute correlations. Label2Label is the first attempt for multi-attribute prediction from the perspective of language modeling. Specifically, it treats each attribute label as a "word" describing the sample. As each sample is annotated with multiple attribute labels, these "words" will naturally form an unordered but meaningful "sentence", which depicts the semantic information of the corresponding sample. Inspired by the remarkable success of pre-training language models in NLP, Label2Label introduces an image-conditioned masked language model, which randomly masks some of the "word" tokens from the label "sentence" and aims to recover them based on the masked "sentence" and the context conveyed by image features. Our intuition is that the instance-wise attribute relations are well grasped if the neural net can infer the missing attributes based on the context and the remaining attribute hints. Label2Label is conceptually simple and empirically powerful. Without incorporating task-specific prior knowledge and highly specialized network designs, our approach achieves state-of-the-art results on three different multi-attribute learning tasks, compared to highly customized domain-specific methods. Code is available at https://github.com/Li-Wanhua/Label2Label.