HyperAIHyperAI

Command Palette

Search for a command to run...

a month ago

Multi-task CNN Model for Attribute Prediction

Abdulnabi Abrar H. Wang Gang Lu Jiwen Jia Kui

Multi-task CNN Model for Attribute Prediction

Abstract

This paper proposes a joint multi-task learning algorithm to better predictattributes in images using deep convolutional neural networks (CNN). Weconsider learning binary semantic attributes through a multi-task CNN model,where each CNN will predict one binary attribute. The multi-task learningallows CNN models to simultaneously share visual knowledge among differentattribute categories. Each CNN will generate attribute-specific featurerepresentations, and then we apply multi-task learning on the features topredict their attributes. In our multi-task framework, we propose a method todecompose the overall model's parameters into a latent task matrix andcombination matrix. Furthermore, under-sampled classifiers can leverage sharedstatistics from other classifiers to improve their performance. Naturalgrouping of attributes is applied such that attributes in the same group areencouraged to share more knowledge. Meanwhile, attributes in different groupswill generally compete with each other, and consequently share less knowledge.We show the effectiveness of our method on two popular attribute datasets.

Benchmarks

BenchmarkMethodologyMetrics
clothing-attribute-recognition-on-clothingMG-CNN
Accuracy: 92.82
clothing-attribute-recognition-on-clothingS-CNN
Accuracy: 90.43

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Multi-task CNN Model for Attribute Prediction | Papers | HyperAI