Hybrid bio-inspired algorithm and convolutional neural network for automatic lung tumor detection

Surbhi Vijh, Prashant Gaurav, Hari Mohan Pandey*

*Corresponding author for this work

Research output: Contribution to journalArticle (journal)peer-review

28 Citations (Scopus)
224 Downloads (Pure)

Abstract

In this paper, we have proposed a hybrid bio-inspired algorithm which takes the merits of whale optimization algorithm (WOA) and adaptive particle swarm optimization (APSO). The proposed algorithm is referred as the hybrid WOA_APSO algorithm. We utilize a convolutional neural network (CNN) for classification purposes. Extensive experiments are performed to evaluate the performance of the proposed model. Here, pre-processing and segmentation are performed on 120 lung CT images for obtaining the segmented tumored and non-tumored region nodule. The statistical, texture, geometrical and structural features are extracted from the processed image using different techniques. The optimized feature selection plays a crucial role in determining the accuracy of the classification algorithm. The novel variant of whale optimization algorithm and adaptive particle swarm optimization, hybrid bio-inspired WOA_APSO, is proposed for selecting optimized features. The feature selection grouping is applied by embedding linear discriminant analysis which helps in determining the reduced dimensions of subsets. Twofold performance comparisons are done. First, we compare the performance against the different classification techniques such as support vector machine, artificial neural network (ANN) and CNN. Second, the computational cost of the hybrid WOA_APSO is compared with the standard WOA and APSO algorithms. The experimental result reveals that the proposed algorithm is capable of automatic lung tumor detection and it outperforms the other state-of-the-art methods on standard quality measures such as accuracy (97.18%), sensitivity (97%) and specificity (98.66%). The results reported in this paper are encouraging; hence, these results will motivate other researchers to explore more in this direction.

Original languageEnglish
Article numberNCAA-D-20-02162R1
JournalNeural Computing and Applications
Early online date19 Sept 2020
DOIs
Publication statusE-pub ahead of print - 19 Sept 2020

Keywords

  • Artificial intelligence
  • Convolutional neural network
  • Feature extraction
  • Hybrid WOA_APSO
  • Medical imaging

Fingerprint

Dive into the research topics of 'Hybrid bio-inspired algorithm and convolutional neural network for automatic lung tumor detection'. Together they form a unique fingerprint.

Cite this