Hyperparameter for Deep Learning Applied in Mammogram Image Classification

Deep Learning has become increasingly frequent in the studies and analysis of medical images. Advances relevant to this area of research improve computer-aided diagnostic systems and help physicians' routine when providing a second opinion. Breast cancer is one of the types most common cancer among women worldwide. Early diagnosis of breast cancer can facilitate treatment and help saves lives. Mammography is the most widely used exam in the clinical routine to diagnose breast cancer. The analysis of the mammogram requires a specialist with experience in medical imaging. Deep Learning and Machine Learning techniques can collaborate computationally with this task. Adapting the hyperparameters provided to deep learning architectures helps improve the results in analyzing and classifying mammogram images. This paper presents a deep learning-based approach to classifying mammogram image regions of interest (ROIs). This approach includes transfer learning, hyperparameter and fine-tuning, and an ensemble with the models that showed the best results. The process demonstrated promising results, with the ensemble reaching 92% accuracy in the classification of mammogram ROIs of the test set and the area under the curve (AUC) value of 0.97 for the best model.