Affiliations: [a] Laboratoire de Recherche en Électronique de Skikda and Department of Computer Science, University of Skikda 20 Août 1955, Skikda, Algeria | [b] Department of Computer Languages and Computer Science, University of Málaga, Málaga, Spain | [c] Biomedic Research Institute of Málaga, Málaga, Spain
Correspondence:
[*]
Corresponding author: Safa Hamreras, Laboratoire de Recherche en Électronique de Skikda and Department of Computer Science, University of Skikda 20 Août 1955, Skikda, Algeria. E-mail: [email protected].
Abstract: Ensemble learning has demonstrated its efficiency in many computer vision tasks. In this paper, we address this paradigm within content based image retrieval (CBIR). We propose to build an ensemble of convolutional neural networks (CNNs), either by training the CNNs on different bags of images, or by using CNNs trained on the same dataset, but having different architectures. Each network is used to extract the class probability vectors from images to use them as representations. The final image representation is then generated by combining the extracted class probability vectors from the built ensemble. We show that the use of CNN ensembles is very efficient in generating a powerful image representation compared to individual CNNs. Moreover, we propose an Averarge Query Expansion technique for our proposal to enhance the retrieval results. Several experiments were conducted to extensively evaluate the application of ensemble learning in CBIR. Results in terms of precision, recall, and mean average precision show the outperformance of our proposal compared to the state of the art.
Keywords: Content based image retrieval, ensemble learning, convolutional neural networks
