MobNas ensembled model for breast cancer prediction.

Journal: Scientific reports
Published Date:

Abstract

Breast cancer poses a real and immense threat to humankind, thus a need to develop a way of diagnosing this devastating disease early, accurately, and in a simpler manner. Thus, while substantial progress has been made in developing machine learning algorithms, deep learning, and transfer learning models, issues with diagnostic accuracy and minimizing diagnostic errors persist. This paper introduces MobNAS, a model that uses MobileNetV2 and NASNetLarge to sort breast cancer images into benign, malignant, or normal classes. The study employs a multi-class classification design and uses a publicly available dataset comprising 1,578 ultrasound images, including 891 benign, 421 malignant, and 266 normal cases. By deploying MobileNetV2, it is easy to work well on devices with less computational capability than is used by NASNetLarge, which enhances its applicability and effectiveness in other tasks. The performance of the proposed MobNAS model was tested on the breast cancer image dataset, and the accuracy level achieved was 97%, the Mean Absolute Error (MAE) was 0.05, and the Matthews Correlation Coefficient (MCC) was 95%. From the findings of this research, it is evident that MobNAS can enhance diagnostic accuracy and reduce existing shortcomings in breast cancer detection.

Authors

  • Tariq Shahzad
    Department of Computer Sciences, COMSATS University Islamabad, Sahiwal Campus, Sahiwal, 57000, Pakistan.
  • Sheikh Muhammad Saqib
    Department of Computing and Information Technology, Gomal University, Dera Ismail Khan, 29050, Pakistan. saqibsheikh4@gu.edu.pk.
  • Tehseen Mazhar
    Department of Computer Science, Virtual University of Pakistan, Lahore, Punjab, Pakistan.
  • Muhammad Iqbal
    Department of Computer Science, Abdul Wali Khan University Mardan, Pakistan.
  • Ahmad Almogren
    Chia of Pervasive and Mobile Computing, College of Computer and Information Sciences, King Saud University, Riyadh, 11543, Saudi Arabia.
  • Yazeed Yasin Ghadi
    Department of Computer Science and Software Engineering, Al Ain University, Abu Dhabi, UAE.
  • Mamoon M Saeed
    Department of Communications and Electronics Engineering, Faculty of Engineering, University of Modern Sciences (UMS), Sana'a, 00967, Yemen. dr.mamoon@ums-edu.com.
  • Habib Hamam
    School of Electrical Engineering, Department of Electrical and Electronic Engineering Science, University of Johannesburg, Johannesburg, South Africa.