Automated Classification and Segmentation and Feature Extraction from Breast Imaging Data

Abstract

Breast cancer is the most common type of cancer in women and poses a significant health risk to women globally. Developments in computer-aided diagnosis (CAD) systems are focused on specific tasks of classification and segmentation, but few studies involve a completely integrated system. In this study, a comprehensive CAD system was proposed to screen ultrasound, mammograms and magnetic resonance imaging (MRI) of breast cancer, including image preprocessing, breast cancer classification, and tumour segmentation. First, the total variation filter was used for image denoising. Second, an optimised XGBoost machine learning model using EfficicnetB0 as feature extraction was proposed to classify breast images into normal and tumour. Third, after classifying the tumour images, a hybrid CNN deep learning model integrating the strengths of MobileNet and InceptionV3 was proposed to categorise tumour images into benign and malignant. Finally, Attention U-Net was used to segment tumours in annotated datasets while classical image segmentation methods were used for the others. The proposed models in the designed CAD system achieved an accuracy of 96.14% on the abnormal classification and 94.81% on tumour classification on the BUSI dataset, improving the effectiveness of automatic breast cancer diagnosis.