Medical Image Retrieval Based on Ensemble Learning using Convolutional Neural Networks and Vision Transformers

The rapid increase in the number of medical image repositories nowadays has led to problems in managing and retrieving medical visual data. This has proved the necessity of Content-Based Image Retrieval (CBIR) with the aim of facilitating the investigation of such medical imagery. One of the most serious challenges that require special attention is the representational quality of the embeddings generated by the retrieval pipelines. These embeddings should include global and local features to obtain more useful information from the input data. To fill this gap, in this paper, we propose a CBIR framework that utilizes the power of deep neural networks to efficiently classify and fetch the most related medical images with respect to a query image. Our proposed model is based on combining Vision Transformers (ViTs) and Convolutional Neural Networks (CNNs) and learns to capture both the locality and also the globality of high-level feature maps. Our method is trained to encode the images in the database and outputs a ranking list containing the most similar image to the least similar one to the query. To conduct our experiments, an intermodal dataset containing ten classes with five different modalities is used to train and assess the proposed framework. The results show an average classification accuracy of ۹۵.۳۲ % and a mean average precision of ۰.۶۱. Our proposed framework can be very effective in retrieving multimodal medical images with the images of different organs in the body.

کلیدواژه ها:

Content-based image retrieval ، medical image retrieval ، Ensemble Learning ، Convolutional neural networks ، vision transformers ، deep learning ، similarity-based visual search