Breast cancer classification using snapshot ensemble deep learning model and t-distributed stochastic neighbor embedding

Nonita Sharma,K. P. Sharma,Monika Mangla,Rajneesh Rani
DOI: https://doi.org/10.1007/s11042-022-13419-5
IF: 2.577
2022-07-21
Multimedia Tools and Applications
Abstract:The current work aims to analyse the historical data pertaining to breast cancer to detect and predict the disease. For the same, authors employ t-distributed stochastic neighbor embedding (t-SNE), a well-established dimensionality reduction method. The authors also suggest implementing the snapshot ensembling technique to create an efficient model that potentially assists medical professionals in disease diagnosis. Employing t-SNE enables the generation of improved scatter plots in addition to cost optimization. Further, the current manuscript also uses a snapshot ensemble deep learning framework that integrates the predictions through various base models leading to accuracy enhancement. The proposed model is implemented on the Wisconsin Breast Cancer Dataset(WBCD) that is openly accessible at UCI Machine Repository. During the experimental evaluation, proposed model yields an accuracy of 86.6%,higher than the state-of-art models like averaging, weighted averaging, stacked ensemble, and Polyak Rupert that yield an accuracy of 81%, 81.7%, 84.7%, and 82.2% respectively and hence establishes the competence of proposed model. The obtained result is highly encouraging and resultantly opens the avenue for implementing the proposed model in real life at large.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?