Gaussian Dropout Based Stacked Ensemble CNN for Classification of Breast Tumor in Ultrasound Images

Abstract

ObjectiveBreast cancer and breast tumors have been considered to be the most pervasive form of cancer in medical practice. Breast tumors are life-threatening to women, and their early detection could save lives with the proper treatment. Physical methods for detection of Breast Cancer are time-consuming and often prone to a misdiagnosis at classifying tumors. Recent trends in radiological imaging have significantly improved the efficiency and veracity of breast tumor classification. Artificial intelligence techniques could be used as an automated detection and classification system. Materials and methodsIn this research, we propose a novel configuration of a Stacking Ensemble with custom Convolutional Neural Network architectures to classify breast tumors from ultrasound images into ‘Normal’, ‘Benign’, and ‘Malignant’ categories. ResultsAfter thorough experimentation, our ensemble has performed with an accuracy, f1-score, precision, and recall of 92.15%, 92.21%, 92.26%, 92.17% respectively. ConclusionThe presented ensemble leverages three Stacked Feature Extractors coupled with a characteristic meta-learner to provide an overall balanced classification performance, with better accuracy and lower false positives. The architecture works in association with gaussian dropout layers to improve the computation and an alternative pooling scheme to retain essential features.