Classification of Multi-view mammogram images using a parallel pre-trained models system
Abdelhafidh Kacher, Medjeded Merati, Saïd Mahmoudi · 2024
Breast cancer is the main lead of women’s cancer-related mortalities. Therefore, early detection is imperative for preventing breast cancer from developing to advanced stages. Moreover, the emergence of Computer-aided diagnosis systems combined with Deep Learning techniques improve breast cancer diagnosis at its early stages. This study proposes a system for classifying mammogram images scanned from different views based on Deep learning techniques. In this paper, we propose combining four pre-trained models in one system employing the Transfer Learning technique with different input images of the same lesion. This proposed system provides a binary classification of the lesion into malignant or benign classes. The Mini-DDSM dataset images were used in this study; the images were pre-processed and augmented to achieve the best results. Several well-known pre-trained models for image classification (ResNet, Xception, DenseNet, MobileNet, and Inception) were experimented with for the proposed model construction. The study results showed that our proposed model achieved an accuracy of 94.18%, a precision of 94.26%, a recall of 94.16%, and an F1-Score of 94.21%, outperforming the pre-trained model on classifying mammogram images of the same dataset.