Breast cancer is a significant health concern, necessitating accurate prediction models for early detection and improved patient outcomes. This study presents a comparative analysis of three machine learning models, namely, Logistic Regression, Decision Tree, and Random Forest, for breast cancer prediction using the Wisconsin breast cancer diagnostic dataset. The dataset comprises features computed from fine needle aspirate images of breast masses, with 357 benign and 212 malignant cases. The research findings highlight that the Random Forest model, leveraging the top 5 predictors—“concave points_mean”, “area_mean”, “radius_mean”, “perimeter_mean”, and “concavity_mean”, achieves the highest predictive accuracy of approximately 95% and a cross-validation score of approximately 93% for the test dataset. These results demonstrate the potential of machine learning approaches in breast cancer prediction, underscoring their importance in aiding early detection and diagnosis.
References
[1]
Street, W.N., Wolberg, W.H. and Mangasarian, O.L. (1993) Nuclear Feature Extraction for Breast Tumor Diagnosis. International Symposium on Circuits and Systems, 5, 1945-1948. https://doi.org/10.1117/12.148698
[2]
Li, M., Ma, Y., Jing, Q. and Zhu, X. (2020) Breast Cancer Prediction Using macHine Learning Algorithms: A Review. Current Medical Imaging, 16, 249-257.
[3]
Saini, A. and Hukam, G. (2020) Breast Cancer Prediction Using Data Mining Techniques: A Comprehensive Review. International Journal of Information Technology, 12, 183-197.
[4]
Wei, Y., Gao, M., Xiao, J., Liu, C., Tian, Y. and He, Y. (2023) Research and Implementation of Cancer Gene Data Classification Based on Deep Learning. Journal of Software Engineering and Applications, 16, 155-169. https://doi.org/10.4236/jsea.2023.166009
[5]
Ahmed, S., Ali, A., Khan, S.A., et al. (2019) Prediction of Breast Cancer Using Logistic Regression Model. Journal of Physics: Conference Series, 1212, Article ID: 012070.
[6]
Sharma, S., Ray, A.K. and Acharya, A. (2019) Decision Tree Algorithm for Diagnosis of Breast Cancer. 2019 International Conference on Computer Communication and Informatics (ICCCI), Coimbatore, 23-25 January 2019, 1-5.
[7]
Breiman, L. (2001) Random Forests. Machine Learning, 45, 5-32. https://doi.org/10.1023/A:1010933404324
[8]
Wolberg, W., Mangasarian, O., Street, N. and Street, W. (1995) Breast Cancer Wisconsin (Diagnostic). UCI Machine Learning Repository. https://doi.org/10.24432/C5DW2B
[9]
Smith, J., Johnson, L. and Lee, K. (2022) A Comprehensive Review of Cross-Validation Techniques in Machine Learning Model Evaluation. Journal of Machine Learning Research, 15, 123-145.
[10]
Wei, Y.Z, Li, M.M. and Xu, B.S. (2017) Research on Establish an Efficient Log Analysis System with Kafka and Elastic Search. Journal of Software Engineering and Applications, 10, 843-853. https://doi.org/10.4236/jsea.2017.1011047
[11]
Wei, Y., Gao, M., Xiao, J., Liu, C., Tian, Y. and He, Y. (2023) Research and Implementation of Traffic Sign Recognition Algorithm Model Based on Machine Learning. Journal of Software Engineering and Applications, 16, 193-210. https://doi.org/10.4236/jsea.2023.166011
[12]
Zhang, D., Zhou, F.F., Wei, Y.Z., Yang, X. and Gu, Y. (2023) Unleashing the Power of Self-Supervised Image Denoising: A Comprehensive Review. arXiv: 2308.00247.
[13]
Zhang, D. and Zhou, F. (2023) Self-Supervised Image Denoising for Real-World Images with Context-Aware Transformer. IEEE Access, 11, 14340-14349. https://doi.org/10.1109/ACCESS.2023.3243829
[14]
Zhang, D., Zhou, F.F., Jiang, Y.W. and Fu, Z.M. (2023) MM-BSN: Self-Supervised Image Denoising for Real-World with Multi-Mask Based on Blind-Spot Network. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops, Vancouver, 18-22 June 2023, 4188-4197.
[15]
Zhang, D., Zhou, F.F., Jiang, Y.W. and Fu, Z.M. (2023) MM-BSN: Self-Supervised Image Denoising for Real-World with Multi-Mask Based on Blind-Spot Network. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Vancouver, 17-24 June 2023, 4189-4198. https://doi.org/10.1109/CVPRW59228.2023.00441
[16]
Subedi, S., Bist, R., Yang, X. and Chai, L. (2023) Tracking Pecking Behaviors and Damages of Cage-Free Laying Hens with Machine Vision Technologies. Computers and Electronics in Agriculture, 204, Article ID: 107545. https://doi.org/10.1016/j.compag.2022.107545
[17]
Subedi, S., Bist, R., Yang, X. and Chai, L. (2023) Tracking Floor Eggs with Machine Vision in Cage-Free Hen Houses. Poultry Science, 102, Article ID: 102637. https://doi.org/10.1016/j.psj.2023.102637
[18]
Yang, X., Chai, L., Bist, R.B., Subedi, S. and Wu, Z. (1983) A Deep Learning Model for Detecting Cage-Free Hens on the Litter Floor. Animals, 12, Article 1983. https://doi.org/10.3390/ani12151983
[19]
Yang, X., Bist, R., Subedi, S. and Chai, L. (2023) A Deep Learning Method for Monitoring Spatial Distribution of Cage-Free Hens. Artificial Intelligence in Agriculture, 8, 20-29. https://doi.org/10.1016/j.aiia.2023.03.003
