Title :
Weakly Semi-Supervised Deep Learning for Multi-Label Image Annotation
Author :
Fei Wu ; Zhuhao Wang ; Zhongfei Zhang ; Yi Yang ; Jiebo Luo ; Wenwu Zhu ; Yueting Zhuang
Author_Institution :
Coll. of Comput. Sci., Zhejiang Univ., Hangzhou, China
Abstract :
In this paper, we study leveraging both weakly labeled images and unlabeled images for multi-label image annotation. Motivated by the recent advance in deep learning, we propose an approach called weakly semi-supervised deep learning for multi-label image annotation (WeSed). In WeSed, a novel weakly weighted pairwise ranking loss is effectively utilized to handle weakly labeled images, while a triplet similarity loss is employed to harness unlabeled images. WeSed enables us to train deep convolutional neural network (CNN) with images from social networks where images are either only weakly labeled with several labels or unlabeled. We also design an efficient algorithm to sample high-quality image triplets from large image datasets to fine-tune the CNN. WeSed is evaluated on benchmark datasets for multi-label annotation. The experiments demonstrate the effectiveness of our proposed approach and show that the leverage of the weakly labeled images and unlabeled images leads to a significantly better performance.
Keywords :
image retrieval; image sampling; learning (artificial intelligence); neural nets; visual databases; CNN training; WeSed; benchmark datasets; deep-convolutional neural network training; high-quality image triplet sampling; large image datasets; multilabel image annotation; social network images; triplet similarity loss; unlabeled image leveraging; weakly-labeled image leveraging; weakly-semisupervised deep-learning; weakly-weighted pairwise ranking loss; Big data; Machine learning; Neural networks; Semantics; Social network services; Training; Visualization; Weakly labeled image; deep learning; ranking loss; unlabeled image;
Journal_Title :
Big Data, IEEE Transactions on
DOI :
10.1109/TBDATA.2015.2497270