Open Access iconOpen Access



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.


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.


Cite This Article

APA Style
Alhajlah, M., Noor, M.N., Nazir, M., Mahmood, A., Ashraf, I. et al. (2023). Gastrointestinal diseases classification using deep transfer learning and features optimization. Computers, Materials & Continua, 75(1), 2227-2245.
Vancouver Style
Alhajlah M, Noor MN, Nazir M, Mahmood A, Ashraf I, Karamat T. Gastrointestinal diseases classification using deep transfer learning and features optimization. Comput Mater Contin. 2023;75(1):2227-2245
IEEE Style
M. Alhajlah, M.N. Noor, M. Nazir, A. Mahmood, I. Ashraf, and T. Karamat "Gastrointestinal Diseases Classification Using Deep Transfer Learning and Features Optimization," Comput. Mater. Contin., 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.
  • 851


  • 427


  • 1


Share Link