Machine Learning Algorithms for Breast Cancer Detection in Mammography Images: A Comparative Study

Rhaylander Almeida, Dehua Chen, Agnaldo Filho, Wladmir Brandão

2021

Abstract

Breast tumor is the most common type of cancer in women worldwide, representing approximately 12% of reported new cases and 6.5% of cancer deaths in 2018. Mammography screening are extremely important for early detection of breast cancer. The assessment of mammograms is a complex task with significant variability due to professional experience and human errors, an opportunity for assisting tools to improve both reliability and accuracy. The usage of deep learning in medical image analysis have increased, assisting specialists in early detection, diagnosis, treatment or prognosis of diseases. In this article, we compare the performance of XGBoost and VGG16 in the task of breast cancer detection by using digital mammograms from CBIS-DDSM dataset. In addition, we perform a comparison of prediction accuracy between full mammogram images and patches extracted from original images based on ROI annotated by experts. Moreover, we also perform experiments with transfer learning and data augmentation to exploit data diversity, and the ability to extract features and learn from raw unprocessed data. Experimental results show that XGBoost achieves 68.29% in AUC, while VGG16 achieves approximately the same performance of 68.24% in AUC.

Paper Citation

in Harvard Style

Almeida R., Chen D., Filho A. and Brandão W. (2021). Machine Learning Algorithms for Breast Cancer Detection in Mammography Images: A Comparative Study. In Proceedings of the 23rd International Conference on Enterprise Information Systems - Volume 1: ICEIS, ISBN 978-989-758-509-8, pages 660-667. DOI: 10.5220/0010440906600667

in Bibtex Style

@conference{iceis21,
author={Rhaylander Almeida and Dehua Chen and Agnaldo Filho and Wladmir Brandão},
title={Machine Learning Algorithms for Breast Cancer Detection in Mammography Images: A Comparative Study},
booktitle={Proceedings of the 23rd International Conference on Enterprise Information Systems - Volume 1: ICEIS,},
year={2021},
pages={660-667},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0010440906600667},
isbn={978-989-758-509-8},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 23rd International Conference on Enterprise Information Systems - Volume 1: ICEIS,
TI - Machine Learning Algorithms for Breast Cancer Detection in Mammography Images: A Comparative Study
SN - 978-989-758-509-8
AU - Almeida R.
AU - Chen D.
AU - Filho A.
AU - Brandão W.
PY - 2021
SP - 660
EP - 667
DO - 10.5220/0010440906600667