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

Sensors

Usman Tariq¹  Majed Alhaisoni²  Artūras Mickus³  Robertas Damaševičius³  Ameer Hamza⁴  Muhammad Attique Khan⁴  Kiran Jabeen⁴  Yu-Dong Zhang⁵ 

[1] College of Computer Engineering and Science, Prince Sattam Bin Abdulaziz University, Al-Kharaj 11942, Saudi Arabia;College of Computer Science and Engineering, University of Ha’il, Ha’il 55211, Saudi Arabia;Department of Applied Informatics, Vytautas Magnus University, LT-44404 Kaunas, Lithuania;Department of Computer Science, HITEC University Taxila, Taxila 47080, Pakistan;Department of Informatics, University of Leicester, Leicester LE1 7RH, UK;

关键词: breast cancer;    data augmentation;    deep learning;    feature optimization;    classification;   

DOI  :  10.3390/s22030807

来源: DOAJ

【 摘 要 】

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.

【 授权许可】

Unknown