Open Access iconOpen Access

ARTICLE

crossmark

Gastrointestinal Diseases Classification Using Deep Transfer Learning and Features Optimization

Mousa Alhajlah1, Muhammad Nouman Noor2, Muhammad Nazir2, Awais Mahmood1,*, Imran Ashraf3, Tehmina Karamat4

1 College of Applied Computer Science, King Saud University (Almuzahmiyah Campus), Riyadh, 11543, Saudi Arabia
2 Department of Computer Science, HITEC University, Taxila, 47080, Pakistan
3 Department of Computer Engineering, HITEC University, Taxila, 47080, Pakistan
4 Department of Software Engineering, Foundation University Islamabad, Islamabad, 44000, Pakistan

* Corresponding Author: Awais Mahmood. Email: email

Computers, Materials & Continua 2023, 75(1), 2227-2245. https://doi.org/10.32604/cmc.2023.031890

Abstract

Gastrointestinal diseases like ulcers, polyps’, and bleeding are increasing rapidly in the world over the last decade. On average 0.7 million cases are reported worldwide every year. The main cause of gastrointestinal diseases is a Helicobacter Pylori (H. Pylori) bacterium that presents in more than 50% of people around the globe. Many researchers have proposed different methods for gastrointestinal disease using computer vision techniques. Few of them focused on the detection process and the rest of them performed classification. The major challenges that they faced are the similarity of infected and healthy regions that misleads the correct classification accuracy. In this work, we proposed a technique based on Mask Recurrent-Convolutional Neural Network (R-CNN) and fine-tuned pre-trained ResNet-50 and ResNet-152 networks for feature extraction. Initially, the region of interest is detected using Mask R-CNN which is later utilized for the training of fine-tuned models through transfer learning. Features are extracted from fine-tuned models that are later fused using a serial approach. Moreover, an Improved Ant Colony Optimization (ACO) algorithm has also opted for the best feature selection from the fused feature vector. The best-selected features are finally classified using machine learning techniques. The experimental process was conducted on the publicly available dataset and obtained an improved accuracy of 96.43%. In comparison with state-of-the-art techniques, it is observed that the proposed accuracy is improved.

Keywords


Cite This Article

M. Alhajlah, M. N. Noor, M. Nazir, A. Mahmood, I. Ashraf et al., "Gastrointestinal diseases classification using deep transfer learning and features optimization," Computers, Materials & Continua, vol. 75, no.1, pp. 2227–2245, 2023.



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 748

    View

  • 330

    Download

  • 1

    Like

Share Link