Abstract
Breast cancer poses a grave threat, being a primary cause of cancer-related fatalities for women globally. Early detection and precise diagnosis are pivotal in enhancing the likelihood of survival. Computer-aided detection systems can aid radiologists in detecting breast cancer on mammograms. This paper proposes a CAD system using Vector Field Convolution (VFC) and active model deformation for mass segmentation and Random Forest (RF) for breast cancer detection. The CBIS-DDSM database provides the mammogram images. Mammograms are processed with two focuses: the entire mammogram and the region having the mass. Preprocessing encompasses noise reduction, binarization, contour-based masking, erosion dilation and Hough transform for the mammogram, and Rough Set (RS) theory-based noise filtering and interference discernment for the region of interest. Segmentation is achieved using an active model deformation approach that minimizes internal and external energy to capture the desired features. Our research results reveal compelling evidence, suggesting that mammogram segmentation utilizing VFC exhibits superior accuracy compared to manual segmentation. Furthermore, comparing various machine learning algorithms, including Random Forest (RF), Support Vector Machine (SVM), and K-Nearest Neighbors (KNN), provides evidence supporting the superior performance of RF over SVM and KNN. Specifically, RF achieved, on its best performance, an accuracy of 83%, proving its effectiveness in mammogram classification tasks using 19 features from segmented mass and the full mammogram.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Estadísticas a propósito del día internacional de la lucha contra el cáncer de mama (19 de octubre). INEGI, 10 2023
WHO. Breast cancer. Available: https://www.who.int/news-room/fact-sheets/detail/breast-cancer
Vijayarajeswari, R., Parthasarathy, P., Vivekanandan, S., Basha, A.A.: Classification of mammogram for early detection of breast cancer using SVM classifier and Hough transform. Measure: J. Int. Measure. Confed. 146, 800–805, 11 (2019)
Lopez, V.R.: Análisis de imágenes de mamografía para la detección de cáncer de mama 2012
M. Dong, X. Lu, Y. Ma, Y. Guo, Y. Ma, and K. Wang, “An efficient approach for automated mass segmentation and classification in mammograms,” J. Digit. Imaging, vol. 28, pp. 613–625, 10 2015. https://doi.org/10.1007/s10278-015-9778-4
Yong, Y., Wang, B., Zhang, W., Peng, Z.: Low-contrast small target image enhancement based on rough set theory. In: Zhou, L., Li, C.-S., Yeung, M.M. (eds.), vol. 11, p. 68332I (2007)
Salem, N., Malik, H., Shams, A.: Medical image enhancement based on histogram algorithms. Procedia Computer Science. 163, 300–311, 1 (2019)
Li, B., Acton, S.T.: Active contour external force using vector field convolution for image segmentation. In: IEEE Transactions on Image Processing, vol. 16, pp. 2096–2106, 8 (2007)
Shen, L., Margolies, L.R., Rothstein, J.H., Fluder, E., McBride, R., Sieh, W.: Deep learning to improve breast cancer detection on screening mammography. Sci. Rep. 9, 12 (2019)
Khalid, A., Mehmood, A., Alabrah, A., Alkhamees, B.F., Amin, F., AlSalman, H., Choi, G.S.: Breast cancer detection and prevention using machine learning. Diagnostics. 13 (2023)
Lee, R.S., Gimenez, F., Hoogi, A., Miyake, K.K., Gorovoy, M., Rubin, D.L.: A curated mammography data set for use in computer-aided detection and diagnosis research. Sci. Data. 4, Article number: 170177 (2017)
L. Breiman, “Random forests,” Mach. Learn., vol. 45, pp. 5–32, 10, 2001. https://doi.org/10.1023/A:1010933404324
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2025 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Salinas-Cortés, H.A., Martínez-Serrato, J.M., Muñoz-Jiménez, V., Ramos, M. (2025). A Machine Learning Approach to Breast Cancer Detection in Mammograms. In: Flores Cuautle, J.d.J.A., et al. XLVII Mexican Conference on Biomedical Engineering. CNIB 2024. IFMBE Proceedings, vol 116. Springer, Cham. https://doi.org/10.1007/978-3-031-82123-3_4
Download citation
DOI: https://doi.org/10.1007/978-3-031-82123-3_4
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-82122-6
Online ISBN: 978-3-031-82123-3
eBook Packages: EngineeringEngineering (R0)