Enhanced whale optimization algorithm for medical feature selection: A COVID-19 case study

Mohammad H. Nadimi-Shahraki, Hoda Zamani, Seyedali Mirjalili

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

The whale optimization algorithm (WOA) is a prominent problem solver which is broadly applied to solve NP-hard problems such as feature selection. However, it and most of its variants suffer from low population diversity and poor search strategy. Introducing efficient strategies is highly demanded to mitigate these core drawbacks of WOA particularly for dealing with the feature selection problem. Therefore, this paper is devoted to proposing an enhanced whale optimization algorithm named E-WOA using a pooling mechanism and three effective search strategies named migrating, preferential selecting, and enriched encircling prey. The performance of E-WOA is evaluated and compared with well-known WOA variants to solve global optimization problems. The obtained results proved that the E-WOA outperforms WOA's variants. After E-WOA showed a sufficient performance, then, it was used to propose a binary E-WOA named BE-WOA to select effective features, particularly from medical datasets. The BE-WOA is validated using medical diseases datasets and compared with the latest high-performing optimization algorithms in terms of fitness, accuracy, sensitivity, precision, and number of features. Moreover, the BE-WOA is applied to detect coronavirus disease 2019 (COVID-19) disease. The experimental and statistical results prove the efficiency of the BE-WOA in searching the problem space and selecting the most effective features compared to comparative optimization algorithms.

Original languageEnglish
Article number105858
JournalComputers in Biology and Medicine
Volume148
DOIs
Publication statusPublished - Sep 2022

Keywords

  • Binary whale optimization algorithm
  • Classification
  • COVID-19
  • Feature selection
  • Medical data mining
  • Transfer functions

Fingerprint

Dive into the research topics of 'Enhanced whale optimization algorithm for medical feature selection: A COVID-19 case study'. Together they form a unique fingerprint.

Cite this