Research and realization of commodity image retrieval system based on deep learning

Cen Chen, Rui Yang, Chongwen Wang*

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

3 Citations (Scopus)

Abstract

This paper proposed a commodity image retrieval system based on CNN and ListNet sort learning method. CNN contained two convolutional layers, two pooling layers and two innerproduct layers. ReLu function was used as the activation function after the convolutional layer, achieving the sparsity and preventing the disappearance of the gradient. The pooling layer used stochastic pooling method and improved the generalization ability of the model. In addition, softmax regression was used for classification. Innerproduct layer adopted dropconnect method, which is more powerful than the generalization of dropout, and it can effectively prevent the occurrence of the overfitting. What’s more, the feature extraction of the network was optimized by stochastic gradient descent (SGD) algorithm. And we combined the learn to rank algorithm of the text retrieval domain. We used ListNet algorithm to combine a variety of feature vectors, solving the problem of the image retrieval.

Original languageEnglish
Title of host publicationParallel Architecture, Algorithm and Programming - 8th International Symposium, PAAP 2017, Proceedings
EditorsHong Shen, Guoliang Chen, Mingrui Chen
PublisherSpringer Verlag
Pages376-385
Number of pages10
ISBN (Print)9789811064418
DOIs
Publication statusPublished - 2017
Event8th International Symposium on Parallel Architectures, Algorithms, and Programming, PAAP 2017 - Haikou, China
Duration: 17 Jun 201718 Jun 2017

Publication series

NameCommunications in Computer and Information Science
Volume729
ISSN (Print)1865-0929

Conference

Conference8th International Symposium on Parallel Architectures, Algorithms, and Programming, PAAP 2017
Country/TerritoryChina
CityHaikou
Period17/06/1718/06/17

Keywords

  • Convolutional network
  • Deep learning
  • Learn to rank
  • ListNet

Fingerprint

Dive into the research topics of 'Research and realization of commodity image retrieval system based on deep learning'. Together they form a unique fingerprint.

Cite this