Deep feature selection using local search embedded social ski-driver optimization algorithm for breast cancer detection in mammograms

Loading...
Thumbnail Image
File version

Version of Record (VoR)

Author(s)
Pramanik, Payel
Mukhopadhyay, Souradeep
Mirjalili, Seyedali
Sarkar, Ram
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2022
Size
File type(s)
Location
Abstract

Breast cancer has become a common malignancy in women. However, early detection and identification of this disease can save many lives. As computer-aided detection helps radiologists in detecting abnormalities efficiently, researchers across the world are striving to develop reliable models to deal with. One of the common approaches to identifying breast cancer is through breast mammograms. However, the identification of malignant breasts from mass lesions is a challenging research problem. In the current work, we propose a method for the classification of breast mass using mammograms which consists of two main stages. At first, we extract deep features from the input mammograms using the well-known VGG16 model while incorporating an attention mechanism into this model. Next, we apply a meta-heuristic called Social Ski-Driver (SSD) algorithm embedded with Adaptive Beta Hill Climbing based local search to obtain an optimal features subset. The optimal features subset is fed to the K-nearest neighbors (KNN) classifier for the classification. The proposed model is demonstrated to be very useful for identifying and differentiating malignant and healthy breasts successfully. For experimentation, we evaluate our model on the digital database for screening mammography (DDSM) database and achieve 96.07% accuracy using only 25% of features extracted by the attention-aided VGG16 model. The Python code of our research work is publicly available at: https://github.com/Ppayel/BreastLocalSearchSSD.

Journal Title

Neural Computing and Applications

Conference Title
Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement

© The Author(s) 2022. Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made.

Item Access Status
Note

This publication has been entered in Griffith Research Online as an advanced online version.

Access the data
Related item(s)
Subject

Oncology and carcinogenesis

Data structures and algorithms

Biomedical imaging

Artificial intelligence

Computer vision and multimedia computation

Machine learning

Science & Technology

Technology

Computer Science, Artificial Intelligence

Computer Science

Mammogram images

Persistent link to this record
Citation

Pramanik, P; Mukhopadhyay, S; Mirjalili, S; Sarkar, R, Deep feature selection using local search embedded social ski-driver optimization algorithm for breast cancer detection in mammograms, Neural Computing and Applications, 2022

Collections