Cargando…

Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer

The existing typical combined query image retrieval methods adopt Euclidean distance as sample distance measurement method, and the model trained by triple loss function blindly pursues absolute distance between samples, resulting in unsatisfactory image retrieval performance. Meanwhile, these metho...

Descripción completa

Detalles Bibliográficos
Autores principales: Zhang, Zhiwei, Wang, Liejun, Cheng, Shuli
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2022
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9718755/
https://www.ncbi.nlm.nih.gov/pubmed/36460827
http://dx.doi.org/10.1038/s41598-022-25340-w
_version_ 1784843161015681024
author Zhang, Zhiwei
Wang, Liejun
Cheng, Shuli
author_facet Zhang, Zhiwei
Wang, Liejun
Cheng, Shuli
author_sort Zhang, Zhiwei
collection PubMed
description The existing typical combined query image retrieval methods adopt Euclidean distance as sample distance measurement method, and the model trained by triple loss function blindly pursues absolute distance between samples, resulting in unsatisfactory image retrieval performance. Meanwhile, these methods singularly adopt Convolutional Neural Network (CNN) to extract reference image features. However, receptive field of convolution operation has the characteristics of locality, which is easy to cause the loss of edge feature information of reference images. In view of shortcomings of these methods, the following improvements are proposed in this paper: (1) We propose Triangle Area Triple Loss Function (TATLF), which adopts Triangle Area (TA) as measurement of sample distance. TA comprehensively considers the absolute distance and included angle between samples, so that the trained model has better retrieval performance; (2) We combine CNN with Transformer to simultaneously extract local and edge features of reference images, which can effectively reduce the loss of reference images information. Specifically, CNN is adopted to extract local feature information of reference images. Transformer is used to pay attention to the edge feature information of reference images. Extensive experiments on two public datasets, Fashion200k and MIT-States, confirm the excellent performance of our proposed method.
format Online
Article
Text
id pubmed-9718755
institution National Center for Biotechnology Information
language English
publishDate 2022
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-97187552022-12-04 Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer Zhang, Zhiwei Wang, Liejun Cheng, Shuli Sci Rep Article The existing typical combined query image retrieval methods adopt Euclidean distance as sample distance measurement method, and the model trained by triple loss function blindly pursues absolute distance between samples, resulting in unsatisfactory image retrieval performance. Meanwhile, these methods singularly adopt Convolutional Neural Network (CNN) to extract reference image features. However, receptive field of convolution operation has the characteristics of locality, which is easy to cause the loss of edge feature information of reference images. In view of shortcomings of these methods, the following improvements are proposed in this paper: (1) We propose Triangle Area Triple Loss Function (TATLF), which adopts Triangle Area (TA) as measurement of sample distance. TA comprehensively considers the absolute distance and included angle between samples, so that the trained model has better retrieval performance; (2) We combine CNN with Transformer to simultaneously extract local and edge features of reference images, which can effectively reduce the loss of reference images information. Specifically, CNN is adopted to extract local feature information of reference images. Transformer is used to pay attention to the edge feature information of reference images. Extensive experiments on two public datasets, Fashion200k and MIT-States, confirm the excellent performance of our proposed method. Nature Publishing Group UK 2022-12-02 /pmc/articles/PMC9718755/ /pubmed/36460827 http://dx.doi.org/10.1038/s41598-022-25340-w Text en © The Author(s) 2022 https://creativecommons.org/licenses/by/4.0/Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Zhang, Zhiwei
Wang, Liejun
Cheng, Shuli
Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer
title Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer
title_full Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer
title_fullStr Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer
title_full_unstemmed Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer
title_short Composed query image retrieval based on triangle area triple loss function and combining CNN with transformer
title_sort composed query image retrieval based on triangle area triple loss function and combining cnn with transformer
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9718755/
https://www.ncbi.nlm.nih.gov/pubmed/36460827
http://dx.doi.org/10.1038/s41598-022-25340-w
work_keys_str_mv AT zhangzhiwei composedqueryimageretrievalbasedontriangleareatriplelossfunctionandcombiningcnnwithtransformer
AT wangliejun composedqueryimageretrievalbasedontriangleareatriplelossfunctionandcombiningcnnwithtransformer
AT chengshuli composedqueryimageretrievalbasedontriangleareatriplelossfunctionandcombiningcnnwithtransformer