Browse > Article
http://dx.doi.org/10.22937/IJCSNS.2021.21.6.1

Medical Image Classification using Pre-trained Convolutional Neural Networks and Support Vector Machine  

Ahmed, Ali (King Abdulaziz University-Rabigh)
Publication Information
International Journal of Computer Science & Network Security / v.21, no.6, 2021 , pp. 1-6 More about this Journal
Abstract
Recently, pre-trained convolutional neural network CNNs have been widely used and applied for medical image classification. These models can utilised in three different ways, for feature extraction, to use the architecture of the pre-trained model and to train some layers while freezing others. In this study, the ResNet18 pre-trained CNNs model is used for feature extraction, followed by the support vector machine for multiple classes to classify medical images from multi-classes, which is used as the main classifier. Our proposed classification method was implemented on Kvasir and PH2 medical image datasets. The overall accuracy was 93.38% and 91.67% for Kvasir and PH2 datasets, respectively. The classification results and performance of our proposed method outperformed some of the related similar methods in this area of study.
Keywords
Pre-trained convolution neural networks; medical image classification; support vector machine;
Citations & Related Records
연도 인용수 순위
  • Reference
1 Ahmed, A. and O.B. El Sadig, Heterogeneous multiclassifier method based on weighted voting for Breast Cancer Detection. Int. J. Adv. Sci. Eng. Technol., 2019. 7(4): p. 36-41.
2 Talo, M., O. Yildirim, U. B. Baloglu, G. Aydin and U. R. Acharya., Convolutional neural networks for multi-class brain disease detection using MRI images. Computerized Medical Imaging and Graphics, 2019. 78: p. 101673.   DOI
3 Kang, C., Y. Huo, L. Xin, B. Tian and B. Yu., Feature selection and tumor classification for microarray data using relaxed Lasso and generalized multi-class support vector machine. Journal of theoretical biology, 2019. 463: p. 77-91.   DOI
4 He, K., X. Zhang, S. Ren and J. Sun. Deep residual learning for image recognition. in Proceedings of the IEEE conference on computer vision and pattern recognition. 2016.
5 ImageNet. http://www.image-net.org
6 Mendoncya, T., P. Ferreira, J. Marques, A. Marcyal and J. Rozeira., A dermoscopic image database for research and benchmarking. Presentation in proceedings of PH2 IEEE EMBC, 2013.
7 Jha, D., P. H. Smedsrud, M. A. Riegler, P. Halvorsen, T. de Lange, D. Johansen and H. D. Johansen. Kvasir-seg: A segmented polyp dataset. in International Conference on Multimedia Modeling. 2020. Springer.
8 Ahmed, A. and S.J. Malebary, Query Expansion Based on Top-Ranked Images for Content-Based Medical Image Retrieval. IEEE Access, 2020. 8: p. 194541-194550.   DOI
9 Ahmed, A., Implementing Relevance Feedback for ContentBased Medical Image Retrieval. IEEE Access, 2020. 8: p. 79969-79976.   DOI
10 Ahmed, A. and S. Malebary, Feature selection and the fusion-based method for enhancing the classification accuracy of SVM for breast cancer detection. Int. J. Comput. Sci. Netw. Secur., 2019. 19(11): p. 55.   DOI
11 Mohammed, A., A. Ahmed, W. Mohammed, G. Viju and M. Taha, Mammogram Images Classification using Linear Discriminant Analysis. 2020. International Research Journal of Engineering and Technology (IRJET), 2020. 7(6):p.6656- 6662 .
12 Kandel, I. and M. Castelli, Transfer learning with convolutional neural networks for diabetic retinopathy image classification. A review. Applied Sciences, 2020. 10(6): p. 2021.   DOI
13 Ibrahim, A. O., A. Ahmed, A. Abdu, R. Abd-alaziz, M. A. Alobeed, A. Y. Saleh and A. Elsafi, Classification of Mammogram Images Using Radial Basis Function Neural Network. in International Conference of Reliable Information and Communication Technology. 2019. Springer. : . p.311-320
14 Zhong, J., W. Wang, H. Wu, Z. Wen and J. Qin. PolypSeg: An Efficient Context-Aware Network for Polyp Segmentation from Colonoscopy Videos. in International Conference on Medical Image Computing and Computer-Assisted Intervention. 2020. Springer.
15 Bernal, J., N. Tajkbaksh, F. J. Sanchez, B. J. Matuszewski, H. Chen, L. Yu, Q. Angermann, O. Romain, B. Rustad and I. Balasingham., Comparative validation of polyp detection methods in video colonoscopy: results from the MICCAI 2015 endoscopic vision challenge. IEEE transactions on medical imaging, 2017. 36(6): p. 1231-1249.   DOI
16 Codella, N. C., D. Gutman, M. E. Celebi, B. Helba, M. A. Marchetti, S. W. Dusza, A. Kalloo, K. Liopyris, N. Mishra and H. Kittler. Skin lesion analysis toward melanoma detection: A challenge at the 2017 international symposium on biomedical imaging (isbi), hosted by the international skin imaging collaboration (isic). in 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018). 2018. IEEE.
17 Jianu, S.R.S., L. Ichim, and D. Popescu. Automatic diagnosis of skin cancer using neural networks. in 2019 11th International Symposium on Advanced Topics in Electrical Engineering (ATEE). 2019. IEEE.
18 Sharifrazi, D., R. Alizadehsani, M. Roshanzamir, J. H. Joloudari, A. Shoeibi, M. Jafari, S. Hussain, Z. A. Sani, F. Hasanzadeh and F. Khozeimeh., Fusion of convolution neural network, support vector machine and Sobel filter for accurate detection of COVID-19 patients using X-ray images. Biomedical Signal Processing and Control, 2021: p. 102622.
19 Mahbod, A., G. Schaefer, C. Wang, R. Ecker and I. Ellinge. Skin lesion classification using hybrid deep neural networks. in ICASSP 2019-2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). 2019. IEEE.
20 Zhuang, F., Z. Qi, K. Duan, D. Xi, Y. Zhu, H. Zhu, H. Xiong and Q. He., A comprehensive survey on transfer learning. Proceedings of the IEEE, 2020. 109(1): p. 43-76.
21 Gopalakrishnan, K., S. K. Khaitan, A. Choudhary and A. Agrawal., Deep convolutional neural networks with transfer learning for computer vision-based data-driven pavement distress detection. Construction and Building Materials, 2017. 157: p. 322-330.   DOI
22 Pham, T.D., Classification of COVID-19 chest X-rays with deep learning: new models or fine tuning? Health Information Science and Systems, 2021. 9(1): p. 1-11.
23 Pham, T.D., A comprehensive study on classification of COVID-19 on computed tomography with pretrained convolutional neural networks. Scientific Reports, 2020. 10(1): p. 1-8.   DOI
24 Ou, X., P. Yan, Y. Zhang, B. Tu, G. Zhang, J. Wu and W. Li., Moving object detection method via ResNet-18 with encoder-decoder structure in complex scenes. IEEE Access, 2019. 7: p. 108152-108160.   DOI
25 Islam, M. A., M. N. R. Shuvo, M. Shamsojjaman, S. Hasan, M. S. Hossain and T. Khatun., An Automated Convolutional Neural Network Based Approach for Paddy Leaf Disease Detection. International Journal of Advanced Computer Science and Applications,, 2021. 12(1): p. 280-288.
26 Ayyachamy, S., V. Alex, M. Khened and G. Krishnamurthi. Medical image retrieval using Resnet-18. in Medical Imaging 2019: Imaging Informatics for Healthcare, Research, and Applications. 2019. International Society for Optics and Photonics.
27 Dang-Nguyen, M. Lux and P. T. Schmidt. Kvasir: A multiclass image dataset for computer aided gastrointestinal disease detection. in Proceedings of the 8th ACM on Multimedia Systems Conference. 2017.
28 Ozturk, S. and U. Ozkaya, Residual LSTM layered CNN for classification of gastrointestinal tract diseases. Journal of Biomedical Informatics, 2021. 113: p. 103638.   DOI
29 Yu, L., H. Chen, Q. Dou, J. Qin and P. A. Heng., Integrating online and offline three-dimensional deep learning for automated polyp detection in colonoscopy videos. IEEE journal of biomedical and health informatics, 2016. 21(1): p. 65-75.   DOI
30 Cervantes, J., F. Garcia-Lamont, L. Rodriguez-Mazahua and A. Lopez., A comprehensive survey on support vector machine classification: Applications, challenges and trends. Neurocomputing, 2020. 408: p. 189-215.   DOI
31 Ma, Y., X. Chen, and B. Sun. Polyp detection in colonoscopy videos by bootstrapping via temporal consistency. in 2020 IEEE 17th International Symposium on Biomedical Imaging (ISBI). 2020. IEEE.
32 Zhou, Y., F. Ren, S. Nishide and X. Kang. Facial Sentiment Classification Based on Resnet-18 Model. in 2019 International Conference on Electronic Engineering and Informatics (EEI). 2019. IEEE.
33 Novitasari, D. C. R., R. Hendradi, R. E. Caraka, Y. Rachmawati, N. Z. Fanani, A. Syarifudin, T. Toharudin and R. C. Chen., Detection of covid-19 chest x-ray using support vector machine and convolutional neural network. Commun. Math. Biol. Neurosci., 2020. 2020: p. Article ID 42.
34 Win, K. Y., N. Maneerat, K. Hamamoto and S. Sreng., Hybrid Learning of Hand-Crafted and Deep-Activated Features Using Particle Swarm Optimization and Optimized Support Vector Machine for Tuberculosis Screening. Applied Sciences, 2020. 10(17): p. 5749.   DOI
35 Ebrahimi, A., S. Luo, and R. Chiong. Introducing Transfer Leaming to 3D ResNet-18 for Alzheimer's Disease Detection on MRI Images. in 2020 35th International Conference on Image and Vision Computing New Zealand (IVCNZ). 2020. IEEE.
36 Cuan, K., T. Zhang, J. Huang, C. Fang and Y. Guan., Detection of avian influenza-infected chickens based on a chicken sound convolutional neural network. Computers and Electronics in Agriculture, 2020. 178: p. 105688.   DOI
37 Menegola, A., J. Tavares, M. Fornaciali, L. T. Li, S. Avila and E. Valle., RECOD titans at ISIC challenge 2017. arXiv preprint arXiv:1703.04819, 2017.