Tehnički vjesnik, Vol. 29 No. 4, 2022.
Izvorni znanstveni članak
https://doi.org/10.17559/TV-20210925093644
Image Retrieval Method Combining Bayes and SVM Classifier Based on Relevance Feedback with Application to Small-scale Datasets
Siyu Lai
; Department of Medical Imaging, North Sichuan Medical College, China
Qinghua Yang
; Department of Medical Imaging, North Sichuan Medical College, China
Wenjin He
; Department of Medical Imaging, North Sichuan Medical College, China
Yuanzhong Zhu
; Department of Medical Imaging, North Sichuan Medical College, China
Juan Wang
orcid.org/0000-0003-2635-8804
; 1) College of Computer Science, China West Normal University, China 2) Department of Computer and Information Sciences, Temple University, USA College of Computer Science, China West Normal University, Shida Road, Nanchong, 637002, Sichuan Province, China
Sažetak
A vast amount of images has been generated due to the diversity and digitalization of devices for image acquisition. However, the gap between low-level visual features and high-level semantic representations has been a major concern that hinders retrieval accuracy. A retrieval method based on the transfer learning model and the relevance feedback technique was formulated in this study to optimize the dynamic trade-off between the structural complexity and retrieval performance of the small- and medium-scale content-based image retrieval (CBIR) system. First, the pretrained deep learning model was fine-tuned to extract features from target datasets. Then, the target dataset was clustered into the relative and irrelative image library by exploring the Bayes classifier. Next, the support vector machine (SVM) classifier was used to retrieve similar images in the relative library. Finally, the relevance feedback technique was employed to update the parameters of both classifiers iteratively until the request for the retrieval was met. Results demonstrate that the proposed method achieves 95.87% in classification index F1 - Score, which surpasses that of the suboptimal approach DCNN-BSVM by 6.76%. The performance of the proposed method is superior to that of other approaches considering retrieval criteria as average precision, average recall, and mean average precision. The study indicates that the Bayes + SVM combined classifier accomplishes the optimal quantities more efficiently than only either Bayes or SVM classifier under the transfer learning framework. Transfer learning skillfully excels training from scratch considering the feature extraction modes. This study provides a certain reference for other insights on applications of small- and medium-scale CBIR systems with inadequate samples.
Ključne riječi
Bayes classifier; image retrieval; relevance feedback; support vector machine classifier; transfer learning
Hrčak ID:
279471
URI
Datum izdavanja:
17.6.2022.
Posjeta: 501 *