Ensemble CNNs for Breast Tumor Classification

To improve the recognition ability of computer-aided breast mass classification among mammographic images, in this work we explore the state-of-the-art classification networks to develop an ensemble mechanism. First, the regions of interest (ROIs) are obtained from the original dataset, and then three models, i.e., XceptionNet, DenseNet, and EfficientNet, are trained individually. After training, we ensemble the mechanism by summing the probabilities outputted from each network which enhances the performance up to 5%. The scheme has been validated on a public dataset and we achieved accuracy, precision, and recall 88%, 85%, and 76% respectively.
Comment: SMA 2021: The 10th International Conference on Smart Media and Applications Gunsan Saemangeum Convention Center and Kunsan National University Gunsan-si, South Korea, September 9-11, 2021