StomachNet: Optimal deep learning features fusion for stomach abnormalities classification

Muhammad Attique Khan, Muhammad Shahzad Sarfraz, Majed Alhaisoni, Abdulaziz A. Albesher*, Shuihua Wang, Imran Ashraf*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

80 Citations (Scopus)

Abstract

A fully automated design is proposed in this work employing optimal deep learning features for classifying gastrointestinal infections. Here, three prominent infections – ulcer, bleeding, polyp and a healthy class are considered as class labels. In the initial stage, the contrast is improved by fusing bi-directional histogram equalization with top-hat filtering output. The resultant fusion images are then passed to ResNet101 pre-trained model and trained once again using deep transfer learning. However, there are challenges involved in extracting deep learning features including impertinent information and redundancy. To mitigate this problem, we took advantage of two metaheuristic algorithms – Enhanced Crow Search and Differential Evolution. These algorithms are implemented in parallel to obtain optimal feature vectors. Following this, a maximum correlation-based fusion approach is applied to fuse optimal vectors from the previous step to obtain an enhanced vector. This final vector is given as input to Extreme Learning Machine (ELM) classifier for final classification. The proposed method is evaluated on a combined database. It accomplished an accuracy of 99.46%, which shows significant improvement over preceding techniques and other neural network architectures.

Original languageEnglish
Pages (from-to)197969-197981
Number of pages13
JournalIEEE Access
Volume8
DOIs
Publication statusPublished - 2020
Externally publishedYes

Keywords

  • Contrast stretching
  • Deep learning
  • Fusion
  • Optimization
  • Stomach infections

Fingerprint

Dive into the research topics of 'StomachNet: Optimal deep learning features fusion for stomach abnormalities classification'. Together they form a unique fingerprint.

Cite this