Nowe podejście do generowania deskryptorów do wyszukiwania obrazów poprzez analizę aktywacji warstw głębokiej sieci neuronowej

Autorzy

Szczegóły publikacji

Paweł Staszewski, Maciej Jaworski, Jinde Cao, Leszek Rutkowski: A New Approach to Descriptors Generation for Image Retrieval by Analyzing Activations of Deep Neural Network Layers. In: IEEE Transactions on Neural Networks and Learning Systems, vol. 33, no. 12, pp. 7913-7920, 2022.

Streszczenie

Streszczenie publikacji

Abstract

In this brief, we consider the problem of descriptors construction for the task of content-based image retrieval using deep neural networks. The idea of neural codes, based on fully connected layers’ activations, is extended by incorporating the information contained in convolutional layers. It is known that the total number of neurons in the convolutional part of the network is large and the majority of them have little influence on the final classification decision. Therefore, in this brief, we propose a novel algorithm that allows us to extract the most significant neuron activations and utilize this information to construct effective descriptors. The descriptors consisting of values taken from both the fully connected and convolutional layers perfectly represent the whole image content. The images retrieved using these descriptors match semantically very well to the query image, and also, they are similar in other secondary image characteristics, such as background, textures, or color distribution. These features of the proposed descriptors are verified experimentally based on the IMAGENET1M dataset using the VGG16 neural network. For comparison, we also test the proposed approach on the ResNet50 network.

Bezpośredni link