arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2007.13547cs.CVcs.LGstat.ML
收藏

Reconstruction Regularized Deep Metric Learning for Multi-label Image Classification

School of Computer Science and Engineering and Big Data Research Center, University of Electronic Science and Technology of China(电子科技大学计算机科学与工程学院与大数据研究中心);PingAn Health Technology Co.,Ltd.(平安健康科技股份有限公司)

Changsheng Li, Chong Liu, Lixin Duan, Peng Gao, Kai Zheng

英文摘要:

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 \emph{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 datasets corroborate the efficacy of our method compared with the state-of-the-arts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑