Adversarial Learning of Label Dependency: A Novel Framework for Multi-class Classification

Abstract

Recent work has shown that exploiting relations between labels improves the performance of multi-label classification. We propose a novel framework based on generative adversarial networks (GANs) to model label dependency. The discriminator learns to model label dependency by discriminating real and generated label sets. To fool the discriminator, the classifier, or generator, learns to generate label sets with dependencies close to real data. Extensive experiments and comparisons on two large-scale image classification benchmark datasets (MS-COCO and NUS-WIDE) show that the discriminator improves generalization ability for different kinds of models

References in corpus (11)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Categorical Reparameterization with Gumbel-Softmax
• Generative Adversarial Text to Image Synthesis
• Improved Training of Wasserstein GANs
• T-CNN: Tubelets with Convolutional Neural Networks for Object Detection from Videos
• Towards Good Practices for Very Deep Two-Stream ConvNets
• Deep Convolutional Ranking for Multilabel Image Annotation
• Learning Spatial Regularization with Image-level Supervisions for Multi-label Image Classification
• Multi-label Image Recognition by Recurrently Discovering Attentional Regions
• Towards Binary-Valued Gates for Robust LSTM Training
• Recurrent Attentional Reinforcement Learning for Multi-label Image Recognition