Back to Search Start Over

Reconstruction Regularized Deep Metric Learning for Multi-Label Image Classification.

Authors :
Li, Changsheng
Liu, Chong
Duan, Lixin
Gao, Peng
Zheng, Kai
Source :
IEEE Transactions on Neural Networks & Learning Systems; Jul2020, Vol. 31 Issue 7, p2294-2303, 10p
Publication Year :
2020

Abstract

In this paper, we present a novel deep metric learning method to tackle the multi-label image classification problem. In order to better learn the correlations among images features, as well as labels, we attempt to explore a latent space, where images and labels are embedded via two unique deep neural networks, respectively. To capture the relationships between image features and labels, we aim to learn a two-way deep distance metric over the embedding space from two different views, i.e., the distance between one image and its labels is not only smaller than those distances between the image and its labels’ nearest neighbors but also smaller than the distances between the labels and other images corresponding to the labels’ nearest neighbors. Moreover, a reconstruction module for recovering correct labels is incorporated into the whole framework as a regularization term, such that the label embedding space is more representative. Our model can be trained in an end-to-end manner. Experimental results on publicly available image data sets corroborate the efficacy of our method compared with the state of the arts. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
2162237X
Volume :
31
Issue :
7
Database :
Complementary Index
Journal :
IEEE Transactions on Neural Networks & Learning Systems
Publication Type :
Periodical
Accession number :
144568146
Full Text :
https://doi.org/10.1109/TNNLS.2019.2924023