Breast Cancer Classification from Ultrasound Images Using Probability-Based Optimal Deep Learning Feature Fusion

Sensors - Tập 22 Số 3 - Trang 807
Kiran Jabeen1, Muhammad Attique Khan1, Majed Alhaisoni2, Usman Tariq3, Yudong Zhang4, Ameer Hamza1, Artūras Mickus5, Robertas Damaševičius5
1Department of Computer Science, HITEC University Taxila, Taxila 47080, Pakistan
2College of Computer Science and Engineering, University of Ha’il, Ha’il 55211, Saudi Arabia
3College of Computer Engineering and Science, Prince Sattam Bin Abdulaziz University, Al-Kharaj 11942, Saudi Arabia
4Department of Informatics, University of Leicester, Leicester, LE1 7RH, UK
5Department of Applied Informatics, Vytautas Magnus University, LT-44404 Kaunas, Lithuania

Tóm tắt

After lung cancer, breast cancer is the second leading cause of death in women. If breast cancer is detected early, mortality rates in women can be reduced. Because manual breast cancer diagnosis takes a long time, an automated system is required for early cancer detection. This paper proposes a new framework for breast cancer classification from ultrasound images that employs deep learning and the fusion of the best selected features. The proposed framework is divided into five major steps: (i) data augmentation is performed to increase the size of the original dataset for better learning of Convolutional Neural Network (CNN) models; (ii) a pre-trained DarkNet-53 model is considered and the output layer is modified based on the augmented dataset classes; (iii) the modified model is trained using transfer learning and features are extracted from the global average pooling layer; (iv) the best features are selected using two improved optimization algorithms known as reformed differential evaluation (RDE) and reformed gray wolf (RGW); and (v) the best selected features are fused using a new probability-based serial approach and classified using machine learning algorithms. The experiment was conducted on an augmented Breast Ultrasound Images (BUSI) dataset, and the best accuracy was 99.1%. When compared with recent techniques, the proposed framework outperforms them.

Từ khóa


Tài liệu tham khảo

Yu, K., Chen, S., and Chen, Y. (2021). Tumor Segmentation in Breast Ultrasound Image by Means of Res Path Combined with Dense Connection Neural Network. Diagnostics, 11.

Feng, 2018, Breast cancer development and progression: Risk factors, cancer stem cells, signaling pathways, genomics, and molecular pathogenesis, Genes Dis., 5, 77, 10.1016/j.gendis.2018.05.001

Badawy, S.M., Mohamed, A.E.-N.A., Hefnawy, A.A., Zidan, H.E., GadAllah, M.T., and El-Banby, G.M. (2021). Automatic semantic segmentation of breast tumors in ultrasound images based on combining fuzzy logic and deep learning—A feasibility study. PLoS ONE, 16.

Zhang, 2019, Clinical implications of tumor-infiltrating immune cells in breast cancer, J. Cancer, 10, 6175, 10.7150/jca.35901

Irfan, R., Almazroi, A.A., Rauf, H.T., Damaševičius, R., Nasr, E., and Abdelgawad, A. (2021). Dilated Semantic Segmentation for Breast Ultrasonic Lesion Detection Using Parallel Feature Fusion. Diagnostics, 11.

Faust, 2018, Comparative assessment of texture features for the identification of cancer in ultrasound images: A review, Biocybern. Biomed. Eng., 38, 275, 10.1016/j.bbe.2018.01.001

Pourasad, Y., Zarouri, E., Salemizadeh Parizi, M., and Salih Mohammed, A. (2021). Presentation of Novel Architecture for Diagnosis and Identifying Breast Cancer Location Based on Ultrasound Images Using Machine Learning. Diagnostics, 11.

Sainsbury, 2000, Breast cancer, BMJ, 321, 745, 10.1136/bmj.321.7263.745

Sun, 2020, Deep learning vs. radiomics for predicting axillary lymph node metastasis of breast cancer using ultrasound images: Don’t forget the peritumoral region, Front. Oncol., 10, 53, 10.3389/fonc.2020.00053

Almajalid, R., Shan, J., Du, Y., and Zhang, M. (2018, January 17–20). Development of a deep-learning-based method for breast ultrasound image segmentation. Proceedings of the 17th IEEE International Conference on Machine Learning and Applications (ICMLA), Orlando, FL, USA.

Ouahabi, A. (2012). Signal and Image Multiresolution Analysis, John Wiley & Sons.

Ahmed, 2015, Nonparametric denoising methods based on contourlet transform with sharp frequency localization: Application to low exposure time electron microscopy images, Entropy, 17, 3461, 10.3390/e17053461

Sood, 2019, Ultrasound for breast cancer detection globally: A systematic review and meta-analysis, J. Glob. Oncol., 5, 1

Byra, 2021, Breast mass classification with transfer learning based on scaling of deep representations, Biomed. Signal Process. Control, 69, 102828, 10.1016/j.bspc.2021.102828

Chen, 2008, Computer-aided diagnosis in breast ultrasound, J. Med. Ultrasound, 16, 46, 10.1016/S0929-6441(08)60005-3

Moustafa, A.F., Cary, T.W., Sultan, L.R., Schultz, S.M., Conant, E.F., Venkatesh, S.S., and Sehgal, C.M. (2020). Color doppler ultrasound improves machine learning diagnosis of breast cancer. Diagnostics, 10.

Shen, 2007, Breast ultrasound computer-aided diagnosis using BI-RADS features, Acad. Radiol., 14, 928, 10.1016/j.acra.2007.04.016

Lee, J.-H., Seong, Y.K., Chang, C.-H., Park, J., Park, M., Woo, K.-G., and Ko, E.Y. (September, January 28). Fourier-based shape feature extraction technique for computer-aided b-mode ultrasound diagnosis of breast tumor. Proceedings of the 2012 Annual International Conference of the IEEE Engineering in Medicine and Biology Society, San Diego, CA, USA.

Ding, 2012, Breast ultrasound image classification based on multiple-instance learning, J. Digit. Imaging, 25, 620, 10.1007/s10278-012-9499-x

Bing, 2017, Sparse representation based multi-instance learning for breast ultrasound image classification, Comput. Math. Methods Med., 2017, 7894705, 10.1155/2017/7894705

Prabhakar, T., and Poonguzhali, S. (September, January 31). Automatic detection and classification of benign and malignant lesions in breast ultrasound images using texture morphological and fractal features. Proceedings of the 2017 10th Biomedical Engineering International Conference (BMEiCON), Hokkaido, Japan.

Zhang, 2017, Dual-modal computer-assisted evaluation of axillary lymph node metastasis in breast cancer patients on both real-time elastography and B-mode ultrasound, Eur. J. Radiol., 95, 66, 10.1016/j.ejrad.2017.07.027

Gao, 2019, New frontiers: An update on computer-aided diagnosis for breast imaging in the age of artificial intelligence, Am. J. Roentgenol., 212, 300, 10.2214/AJR.18.20392

Geras, 2019, Artificial intelligence for mammography and digital breast tomosynthesis: Current concepts and future perspectives, Radiology, 293, 246, 10.1148/radiol.2019182627

Fujioka, T., Mori, M., Kubota, K., Oyama, J., Yamaga, E., Yashima, Y., Katsuta, L., Nomura, K., Nara, M., and Oda, G. (2020). The utility of deep learning in breast ultrasonic imaging: A review. Diagnostics, 10.

Zahoor, 2020, Breast cancer detection and classification using traditional computer vision techniques: A comprehensive review, Curr. Med. Imaging, 16, 1187, 10.2174/1573405616666200406110547

Kadry, S., Rajinikanth, V., Taniar, D., Damaševičius, R., and Valencia, X.P.B. (2021). Automated segmentation of leukocyte from hematological images—A study using various CNN schemes. J. Supercomput., 1–21.

Misra, 2021, Malignant skin melanoma detection using image augmentation by oversampling in nonlinear lower-dimensional embedding manifold, Turk. J. Electr. Eng. Comput. Sci., 29, 2600, 10.3906/elk-2101-133

Maqsood, S., Damaševičius, R., and Maskeliūnas, R. (2021). Hemorrhage detection based on 3d cnn deep learning framework and feature fusion for evaluating retinal abnormality in diabetic patients. Sensors, 21.

Hussain, 2021, Intelligent Deep Learning and Improved Whale Optimization Algorithm Based Framework for Object Recognition, Hum. Cent. Comput. Inf. Sci., 11, 34

Kadry, S., Parwekar, P., Damaševičius, R., Mehmood, A., Khan, J.A., Naqvi, S.R., and Khan, M.A. (2021). Human gait analysis for osteoarthritis prediction: A framework of deep learning and kernel extreme learning machine. Complex Intell. Syst., 1–19.

Dhungel, N., Carneiro, G., and Bradley, A.P. (2016). The automated learning of deep features for breast mass classification from mammograms. Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention, Athens, Greece, 17–21 October 2016, Springer.

Alhaisoni, 2021, COVID-19 Case Recognition from Chest CT Images by Deep Learning, Entropy-Controlled Firefly Optimization, and Parallel Feature Fusion, Sensors, 21, 7286, 10.3390/s21217286

Odusami, M., Maskeliūnas, R., Damaševičius, R., and Krilavičius, T. (2021). Analysis of features of alzheimer’s disease: Detection of early stage from functional brain changes in magnetic resonance images using a finetuned resnet18 network. Diagnostics, 11.

Nawaz, M., Nazir, T., Masood, M., Mehmood, A., Mahum, R., Kadry, S., and Thinnukool, O. (2021). Analysis of Brain MRI Images Using Improved CornerNet Approach. Diagnostics, 11.

Farzaneh, N., Williamson, C.A., Jiang, C., Srinivasan, A., Bapuraj, J.R., Gryak, J., Najarian, K., and Soroushmehr, S. (2020). Automated segmentation and severity analysis of subdural hematoma for patients with traumatic brain injuries. Diagnostics, 10.

Meng, L., Zhang, Q., and Bu, S. (2021). Two-Stage Liver and Tumor Segmentation Algorithm Based on Convolutional Neural Network. Diagnostics, 11.

Khaldi, 2021, Ear recognition based on deep unsupervised active learning, IEEE Sens. J., 21, 20704, 10.1109/JSEN.2021.3100151

Majid, 2021, COVID19 classification using CT images via ensembles of deep learning models, Comput. Mater. Contin., 69, 319

Sharif, M.I., Alhussein, M., Aurangzeb, K., and Raza, M. (2021). A decision support system for multimodal brain tumor classification using deep learning. Complex Intell. Syst., 1–14.

Liu, 2019, Fusion of handcrafted and deep features for medical image classification, J. Phys. Conf. Ser., 1345, 022052, 10.1088/1742-6596/1345/2/022052

Alinsaif, 2021, 3D shearlet-based descriptors combined with deep features for the classification of Alzheimer’s disease based on MRI data, Comput. Biol. Med., 138, 104879, 10.1016/j.compbiomed.2021.104879

Khan, 2021, Multi-Class Skin Lesion Detection and Classification via Teledermatology, IEEE J. Biomed. Health Inform., 25, 4267, 10.1109/JBHI.2021.3067789

Masud, M., Rashed, A.E.E., and Hossain, M.S. (2020). Convolutional neural network-based models for diagnosis of breast cancer. Neural Comput. Appl., 1–12.

Jiménez-Gaona, Y., Rodríguez-Álvarez, M.J., and Lakshminarayanan, V. (2020). Deep-Learning-Based Computer-Aided Systems for Breast Cancer Imaging: A Critical Review. Appl. Sci., 10.

Zeebaree, 2020, A Review on Region of Interest Segmentation Based on Clustering Techniques for Breast Cancer Ultrasound Images, J. Appl. Sci. Technol. Trends, 1, 78

Huang, K., Zhang, Y., Cheng, H., and Xing, P. (2021, January 5–9). Shape-adaptive convolutional operator for breast ultrasound image segmentation. Proceedings of the 2021 IEEE International Conference on Multimedia and Expo (ICME), Shenzhen, China.

Sadad, T., Hussain, A., Munir, A., Habib, M., Ali Khan, S., Hussain, S., Yang, S., and Alawairdhi, M. (2020). Identification of breast malignancy by marker-controlled watershed transformation and hybrid feature set for healthcare. Appl. Sci., 10.

Mishra, 2021, Breast ultrasound tumour classification: A Machine Learning—Radiomics based approach, Expert Syst., 38, e12713, 10.1111/exsy.12713

Hussain, 2020, Contextual level-set method for breast tumor segmentation, IEEE Access, 8, 189343, 10.1109/ACCESS.2020.3029684

Xiangmin, H., Jun, W., Weijun, Z., Cai, C., Shihui, Y., and Jun, S. (2020). Deep Doubly Supervised Transfer Network for Diagnosis of Breast Cancer with Imbalanced Ultrasound Imaging Modalities. arXiv.

Moon, 2020, Computer-aided diagnosis of breast ultrasound images using ensemble learning from convolutional neural networks, Comput. Methods Programs Biomed., 190, 105361, 10.1016/j.cmpb.2020.105361

Byra, 2020, Breast mass segmentation in ultrasound with selective kernel U-Net convolutional neural network, Biomed. Signal Process. Control, 61, 102027, 10.1016/j.bspc.2020.102027

Kadry, S., Damaševičius, R., Taniar, D., Rajinikanth, V., and Lawal, I.A. (2021, January 25–27). Extraction of tumour in breast MRI using joint thresholding and segmentation–A study. Proceedings of the 2021 Seventh International conference on Bio Signals, Images, and Instrumentation (ICBSII), Chennai, India.

Lahoura, V., Singh, H., Aggarwal, A., Sharma, B., Mohammed, M., Damaševičius, R., Kadry, S., and Cengiz, K. (2021). Cloud computing-based framework for breast cancer diagnosis using extreme learning machine. Diagnostics, 11.

Maqsood, S., Damasevicius, R., and Shah, F.M. (2021). An Efficient Approach for the Detection of Brain Tumor Using Fuzzy Logic and U-NET CNN Classification, International Conference on Computational Science and Its Applications, Springer.

Rajinikanth, V., Kadry, S., Taniar, D., Damasevicius, R., and Rauf, H.T. (2021, January 26–27). Breast-cancer detection using thermal images with marine-predators-algorithm selected features. Proceedings of the 2021 Seventh International conference on Bio Signals, Images, and Instrumentation (ICBSII), Noida, India.

Ouahabi, 2021, Deep learning for real-time semantic segmentation: Application in ultrasound imaging, Pattern Recognit. Lett., 144, 27, 10.1016/j.patrec.2021.01.010

Gomaa, 2020, Dataset of breast ultrasound images, Data Brief, 28, 104863, 10.1016/j.dib.2019.104863

Khan, 2021, Prediction of COVID-19-pneumonia based on selected deep features and one class kernel extreme learning machine, Comput. Electr. Eng., 90, 106960, 10.1016/j.compeleceng.2020.106960

Khan, M.A., Sharif, M.I., Raza, M., Anjum, A., Saba, T., and Shad, S.A. (2019). Skin lesion segmentation and classification: A unified framework of deep neural network features fusion and selection. Expert Syst., e12497.

2006, Statistical comparisons of classifiers over multiple data sets, J. Mach. Learn. Res., 7, 1

Cao, 2020, Breast tumor classification through learning from noisy labeled ultrasound images, Med. Phys., 47, 1048, 10.1002/mp.13966

Ilesanmi, 2021, A method for segmentation of tumors in breast ultrasound images using the variant enhanced deep learning, Biocybern. Biomed. Eng., 41, 802, 10.1016/j.bbe.2021.05.007

Zhuang, 2021, Breast ultrasound tumor image classification using image decomposition and fusion based on adaptive multi-model spatial feature fusion, Comput. Methods Programs Biomed., 208, 106221, 10.1016/j.cmpb.2021.106221

Pang, 2021, Semi-supervised GAN-based Radiomics Model for Data Augmentation in Breast Ultrasound Mass Classification, Comput. Methods Programs Biomed., 203, 106018, 10.1016/j.cmpb.2021.106018