Optimizing the transfer-learning with pretrained deep convolutional neural networks for first stage breast tumor diagnosis using breast ultrasound visual images.

Journal: Microscopy research and technique
Published Date:

Abstract

Female accounts for approximately 50% of the total population worldwide and many of them had breast cancer. Computer-aided diagnosis frameworks could reduce the number of needless biopsies and the workload of radiologists. This research aims to detect benign and malignant tumors automatically using breast ultrasound (BUS) images. Accordingly, two pretrained deep convolutional neural network (CNN) models were employed for transfer learning using BUS images like AlexNet and DenseNet201. A total of 697 BUS images containing benign and malignant tumors are preprocessed and performed classification tasks using the transfer learning-based CNN models. The classification accuracy of the benign and malignant tasks is completed and achieved 92.8% accuracy using the DensNet201 model. The results thus achieved compared in state of the art using benchmark data set and concluded proposed model outperforms in accuracy from first stage breast tumor diagnosis. Finally, the proposed model could help radiologists diagnose benign and malignant tumors swiftly by screening suspected patients.

Authors

  • Tanzila Saba
    College of Computer and Information Sciences, Prince Sultan University, Riyadh, 11586, Saudi Arabia.
  • Ibrahim Abunadi
    Information Systems Department, Prince Sultan University, Riyadh, Saudi Arabia.
  • Tariq Sadad
    Department of Computer Science, University of Central Punjab, Lahore, Pakistan.
  • Amjad Rehman Khan
    Artificial Intelligence and Data Analytics Lab, CCIS Prince Sultan University, Riyadh, Saudi Arabia.
  • Saeed Ali Bahaj
    MIS Department College of Business Administration, Prince Sattam bin Abdulaziz University, Alkharj, Saudi Arabia.