Automatic machine learning model for enhanced partition and identification of breast disorders in breast MRI scan

Abstract

The rapid identification and categorisation of breast cancers using low-contrast MRI images presents a significant challenge due to the disease’s prevalence among women of all ages. Nowadays, it is very difficult to classify and generalise models from low-contrast MRI datasets due to the prevalence of class imbalance caused by a wide range of symptoms and untrustworthy data sources. Using ensemble learning with optimised k-means helps with these problems; it reduces overfitting and improves reliability by using methods like k-means clustering, frog feap algorithm (FLA), boosting, and bagging to balance datasets and show the complexity of symptoms. In this research, we present a new strategy that makes use of a hybrid optimised K-Means algorithm combined with the FLA, along with thresholding and morphological approaches, and ensemble learning. There are two main stages to our work. In the initial phase, we preprocess low-contrast MRI images and delineate the breast tumour area from the segmented MRI images. Furthermore, we employ Discrete Wavelet Transformation (DWT) to extract the most salient features. In the next step, the retrieved features will be used as input parameters for an ensemble-learned breast tumour classifier. To determine whether a low-contrast breast tumour MRI picture has a benign or malignant tumour after model training, we use a variety of classifiers, such as k-nearest neighbours (KNN), decision trees (DT), gradient boosting (GB), random forests (RF), and artificial neural networks (ANN). Through a voting mechanism that averages the accuracy of all models, our suggested framework achieves a remarkable 98.8% accuracy rate. In addition, proposed model is quite efficient & reliable to detect the kind of breast tumour or cancer with a sensitivity of 98.02% and a specificity of 97.5%.

Keywords

• Breast cancer
• ensemble-learning
• frog leap algorithm
• feature extraction
• Optimized k-means clustering