Deep Learning-based Metaheuristic Weighted K-Nearest Neighbor Algorithm for the Severity Classification of Breast Cancer

Elsevier

Available online 3 January 2023, 100749

IRBMAuthor links open overlay panelHighlights•

A novel deep learning-based classification framework is proposed for medical data.

Transfer learning with a novel feature transformation technique is implemented.

The metaheuristic algorithms are utilized for the feature transformation process.

The approach is evaluated for breast cancer classification using different datasets.

The results revealed that the approach is supreme over state-of-the-art techniques.

AbstractObjective

The most widespread and intrusive cancer type among women is breast cancer. Globally, this type of cancer causes more mortality among women, next to lung cancer. This made the researchers to focus more on developing effective Computer-Aided Detection (CAD) methodologies for the classification of such deadly cancer types. In order to improve the rate of survival and earlier diagnosis, an optimistic research methodology is required in the classification of breast cancer. Consequently, an improved methodology that integrates the principle of deep learning with metaheuristic and classification algorithms is proposed for the severity classification of breast cancer. Hence to enhance the recent findings, an improved CAD methodology is proposed for redressing the healthcare problem.

Material and Methods

The work intends to cast a light-of-research towards classifying the severities present in digital mammogram images. For evaluating the work, the publicly available MIAS, INbreast, and WDBC databases are utilized. The proposed work employs transfer learning for extricating the features. The novelty of the work lies in improving the classification performance of the weighted k-nearest neighbor (wKNN) algorithm using particle swarm optimization (PSO), dragon-fly optimization algorithm (DFOA), and crow-search optimization algorithm (CSOA) as a transformation technique i.e., transforming non-linear input features into minimal linear separable feature vectors.

Results

The results obtained for the proposed work are compared then with the Gaussian Naïve Bayes and linear Support Vector Machine algorithms, where the highest accuracy for classification is attained for the proposed work (CSOA-wKNN) with 84.35% for MIAS, 83.19% for INbreast, and 97.36% for WDBC datasets respectively.

Conclusion

The obtained results reveal that the proposed Computer-Aided-Diagnosis (CAD) tool is robust for the severity classification of breast cancer.

Graphical abstractDownload : Download high-res image (140KB)Download : Download full-size imageKeywords

Breast cancer

Mammograms

deep-learning

particle swarm optimization

dragonfly

crow-search

and KNN

View full text

© 2022 AGBM. Published by Elsevier Masson SAS. All rights reserved.

留言 (0)

沒有登入
gif