Open Access Open Access  Restricted Access Subscription Access

BREAST CANCER CLASSIFICATION: A COMPARISON OF MACHINE LEARNING MODELS

Aitha Sreevarsha Sreevarsha, Dr. CRK Reddy, Ms.K. Sunitha, Ms. Musrat Sultana

Abstract


Breast cancer remains one of the most common cancers affecting women globally, and early detection is vital for effective treatment and improved survival rates. This paper presents a comparative analysis of three supervised machine learning algorithms—Logistic Regression, Support Vector Machine (SVM), and Random Forest—for classifying breast tumors as benign or malignant. The experiments were conducted on the Breast Cancer Wisconsin Diagnostic Dataset, containing 30 numerical features derived from medical imaging. Each model was trained and tested using an 80:20 train- test split. Among the models evaluated, the Support Vector Machine (SVM) achieved the highest test accuracy of 94.7%, outperforming both Logistic Regression and Random Forest in terms of generalization performance. A user- friendly interface was also developed using Gradio to facilitate real-time predictions. The results confirm that SVM is a highly effective algorithm for breast cancer classification and demonstrates strong potential for deployment in clinical decision-support systems.


Full Text:

PDF

References


Ramzan et al. (2024). “Comparative analysis of ML algorithms for breast cancer classification.” Journal of Computing & Biomedical Informatics. https://jcbi.org/index.php/Main/article/vi ew/564

Yasmeen et al. (2024). “Study of ensemble techniques for breast cancer classification.” Journal of Computing & Biomedical Informatics. https://jcbi.org/index.php/Main/article/vi ew/648

Chatterjee et al. (2024). “Breast cancer classification using machine learning.” JDAIC.

https://doi.org/10.31181/jdaic100280120 24c

Kumar et al. (2024). “Efficient breast cancer detection using classification models.” IJOE.

https://doi.org/10.3991/ijoe.v20i13.5028 9

D. Houfani, S. Slatnia, O. Kazar, N. Zerhouni, et al.,“Breast cancer classification using machine learning techniques: A comparative study,” Medical Technologies Journal, vol. 4, no. 2, pp. 535–544, Nov. 2020.

http://dx.doi.org/10.26415/2572-004X- vol4iss2p535-544

Wu J, Hicks C. Breast cancer type classification using machine learning. J Personalized Med 2021;11(2):61. https://doi.org/10.3390/jpm11020061.


Refbacks

  • There are currently no refbacks.