Diagnose Breast Cancer Using Two Machine Learning Methods andComparing and Analyzing Them

Authors

  • Chengjun Hou

DOI:

https://doi.org/10.61173/fv3z4j30

Keywords:

Breast Cancer, Machine Learning, Decision Tree, SVM

Abstract

The breast cancer is currently the number one cancer in the world and has a high early treatment rate. Therefore, early screening and diagnosis of breast cancer is very important. Machine learning is very strong in analyzing and processing massive data. In this paper, we train and optimize decision tree and SVM models to identify malignant breast cancer and compare and analyze the performance of the two models. The accuracy of the earliest optimized models reaches about 98%, and it is found that the two models focus on different classification effects for the two samples, which can be further optimized by stacking and other ways. The dataset used to train the models in this paper is from the Wisconsin breast cancer database (WBCD), a benchmark dataset commonly used to compare different algorithms, which is of some significance for exploring the use of machine learning in the field of medical diagnosis.

References

[1] Bingfeng Han, Rongshou Zheng, Hongmei Zeng, Shaoming Wang, Kexin Sun, Ru Chen, Li Li, Wenqiang Wei, et al. Cancer incidence and mortality in China, 2022. Journal of the National Cancer Center, Vol 4, Issue 1, 2024; 47-53.

[2] Gabriel N. Hortobagyi, et al. The Global Breast Cancer Burden: Variations in Epidemiology and Survival. Clinical Breast Cancer, Vol. 6, No. 5, 2005; 391-401.

[3] Taylor C, McGale P, Probert J, et al. Breast cancer mortality in 500000 women with early invasive breast cancer diagnosed in England, 1993-2015: population based observational cohort study. BMJ. 2023 Jun 13; 381:e074684.

[4] Warner E, et al. Systematic review: using magnetic resonance imaging to screen women at high risk for breast cancer. Ann Intern Med. 2008 May 6; 148(9):671-9.

[5] A. Osareh, B. Shadgar, Machine learning techniques to diagnose breast cancer, 2010 5th International Symposium on Health Informatics and Bioinformatics, Ankara, Turkey, 2010, pp. 114-120.

[6] Benson J, Jatoi I, Keisch M, Esteva F, Makris A, Jordan VC, Early breast cancer. Lancet. 2009; 373: 1463-79

Downloads

Published

2024-08-14