Multi-objective hyperparameter optimization on gradient-boosting for breast cancer detection

Priya Singh (), Swayam Gupta () and Vasu Gupta ()
Additional contact information
Priya Singh: Delhi Technological University
Swayam Gupta: Delhi Technological University
Vasu Gupta: Delhi Technological University

International Journal of System Assurance Engineering and Management, 2024, vol. 15, issue 5, No 6, 1676-1686

Abstract: Abstract The most commonly occurring cancer among women, breast cancer, causes lakhs of deaths annually, which can be prevented by early detection and treatment. Detection can be done by using machine learning models on histopathological images which are affordable, reliable, and accurate. Previous studies in this regard have focused on transfer learning methods combining feature selection using Convolutional Neural Networks (CNNs) and an ensemble of gradient-boosting algorithms. However, none of the state-of-the-art techniques capture the multi-objective nature of Breast Cancer Detection (BCD) and tend to improve a single performance measure such as Accuracy and F1 score, which fail to capture certain essential aspects of the problem as the cost of misclassification varies greatly depending on its type. In this study, a multi-objective hyperparameter optimization technique for Breast Cancer Prediction is proposed by comparing random search, Non-Dominated Sorting Genetic Algorithm (NSGA-II) and Bayesian optimization. This approach is applied to an ensemble of three popular gradient-boosting techniques: extreme gradient-boosting, light gradient-boosting machine and categorical boosting on features obtained from Inception-ResNet-v2 CNN model applied on the benchmark BreakHis dataset to optimize Precision, Recall, Accuracy, and AUC simultaneously. The novel NSGA2-IRv2-CXL model proposed in this study achieves maximum Accuracy of 94.40%, AUC of 98.16, Precision of 95.77%, and Recall of 99.29% for 100 $$\times$$ × magnification. The study also establishes trade-offs between performance metrics thereby opening avenues for further research in multi-objective approaches to BCD which can provide a larger view of the strengths and weaknesses of the classification model.

Keywords: Multi-objective optimization; Hyperparameter tuning; Gradient-boosting; Breast cancer detection (search for similar items in EconPapers)
Date: 2024
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
http://link.springer.com/10.1007/s13198-023-01955-8 Abstract (text/html)
Access to the full text of the articles in this series is restricted.

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:spr:ijsaem:v:15:y:2024:i:5:d:10.1007_s13198-023-01955-8

Ordering information: This journal article can be ordered from
http://www.springer.com/engineering/journal/13198

DOI: 10.1007/s13198-023-01955-8

Access Statistics for this article

International Journal of System Assurance Engineering and Management is currently edited by P.K. Kapur, A.K. Verma and U. Kumar

More articles in International Journal of System Assurance Engineering and Management from Springer, The Society for Reliability, Engineering Quality and Operations Management (SREQOM),India, and Division of Operation and Maintenance, Lulea University of Technology, Sweden
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().