Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification
Convolutional neural networks and deep learning models represent the gold standard in medical image classification. Their innovative architectures have led to notable breakthroughs in image classification and feature extraction performance. However, these advancements often remain underutilized in t...
Main Authors: | , , |
---|---|
Format: | Article |
Language: | English |
Published: |
MDPI AG
2023-08-01
|
Series: | Applied Sciences |
Subjects: | |
Online Access: | https://www.mdpi.com/2076-3417/13/17/9639 |
_version_ | 1797582902135881728 |
---|---|
author | Mayra C. Berrones-Reyes M. Angélica Salazar-Aguilar Cristian Castillo-Olea |
author_facet | Mayra C. Berrones-Reyes M. Angélica Salazar-Aguilar Cristian Castillo-Olea |
author_sort | Mayra C. Berrones-Reyes |
collection | DOAJ |
description | Convolutional neural networks and deep learning models represent the gold standard in medical image classification. Their innovative architectures have led to notable breakthroughs in image classification and feature extraction performance. However, these advancements often remain underutilized in the medical imaging field due to the scarcity of sufficient labeled data which are needed to leverage these new features fully. While many methodologies exhibit stellar performance on benchmark data sets like DDSM or Minimias, their efficacy drastically decreases when applied to real-world data sets. This study aims to develop a tool to streamline mammogram classification that maintains high reliability across different data sources. We use images from the DDSM data set and a proprietary data set, YERAL, which comprises 943 mammograms from Mexican patients. We evaluate the performance of ensemble learning algorithms combined with prevalent deep learning models such as Alexnet, VGG-16, and Inception. The computational results demonstrate the effectiveness of the proposed methodology, with models achieving 82% accuracy without overtaxing our hardware capabilities, and they also highlight the efficiency of ensemble algorithms in enhancing accuracy across all test cases. |
first_indexed | 2024-03-10T23:28:07Z |
format | Article |
id | doaj.art-de1976c997a04a3c8938de8a1a1ddda2 |
institution | Directory Open Access Journal |
issn | 2076-3417 |
language | English |
last_indexed | 2024-03-10T23:28:07Z |
publishDate | 2023-08-01 |
publisher | MDPI AG |
record_format | Article |
series | Applied Sciences |
spelling | doaj.art-de1976c997a04a3c8938de8a1a1ddda22023-11-19T07:49:49ZengMDPI AGApplied Sciences2076-34172023-08-011317963910.3390/app13179639Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography ClassificationMayra C. Berrones-Reyes0M. Angélica Salazar-Aguilar1Cristian Castillo-Olea2Facultad de Ingeniería Mecánica y Eléctrica, Universidad Autónoma de Nuevo León, Av. Universidad s/n, Cd. Universitaria, San Nicolás de los Garza 66455, MexicoFacultad de Ingeniería Mecánica y Eléctrica, Universidad Autónoma de Nuevo León, Av. Universidad s/n, Cd. Universitaria, San Nicolás de los Garza 66455, MexicoFacultad de Ingeniería, CETYS Universidad Campus Mexicali, Calzada CETYS s/n, Colonia Rivera, Mexicali 21259, MexicoConvolutional neural networks and deep learning models represent the gold standard in medical image classification. Their innovative architectures have led to notable breakthroughs in image classification and feature extraction performance. However, these advancements often remain underutilized in the medical imaging field due to the scarcity of sufficient labeled data which are needed to leverage these new features fully. While many methodologies exhibit stellar performance on benchmark data sets like DDSM or Minimias, their efficacy drastically decreases when applied to real-world data sets. This study aims to develop a tool to streamline mammogram classification that maintains high reliability across different data sources. We use images from the DDSM data set and a proprietary data set, YERAL, which comprises 943 mammograms from Mexican patients. We evaluate the performance of ensemble learning algorithms combined with prevalent deep learning models such as Alexnet, VGG-16, and Inception. The computational results demonstrate the effectiveness of the proposed methodology, with models achieving 82% accuracy without overtaxing our hardware capabilities, and they also highlight the efficiency of ensemble algorithms in enhancing accuracy across all test cases.https://www.mdpi.com/2076-3417/13/17/9639convolutional neural networksensemble learningdeep learningtransfer learningimage classificationmedical imaging |
spellingShingle | Mayra C. Berrones-Reyes M. Angélica Salazar-Aguilar Cristian Castillo-Olea Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification Applied Sciences convolutional neural networks ensemble learning deep learning transfer learning image classification medical imaging |
title | Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification |
title_full | Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification |
title_fullStr | Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification |
title_full_unstemmed | Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification |
title_short | Use of Ensemble Learning to Improve Performance of Known Convolutional Neural Networks for Mammography Classification |
title_sort | use of ensemble learning to improve performance of known convolutional neural networks for mammography classification |
topic | convolutional neural networks ensemble learning deep learning transfer learning image classification medical imaging |
url | https://www.mdpi.com/2076-3417/13/17/9639 |
work_keys_str_mv | AT mayracberronesreyes useofensemblelearningtoimproveperformanceofknownconvolutionalneuralnetworksformammographyclassification AT mangelicasalazaraguilar useofensemblelearningtoimproveperformanceofknownconvolutionalneuralnetworksformammographyclassification AT cristiancastilloolea useofensemblelearningtoimproveperformanceofknownconvolutionalneuralnetworksformammographyclassification |