Mathematical Biosciences and Engineering (Apr 2023)

Deep transfer learning for IDC breast cancer detection using fast AI technique and Sqeezenet architecture

  • Sushovan Chaudhury,
  • Kartik Sau ,
  • Muhammad Attique Khan,
  • Mohammad Shabaz

DOI
https://doi.org/10.3934/mbe.2023457
Journal volume & issue
Vol. 20, no. 6
pp. 10404 – 10427

Abstract

Read online

One of the most effective approaches for identifying breast cancer is histology, which is the meticulous inspection of tissues under a microscope. The kind of cancer cells, or whether they are cancerous (malignant) or non-cancerous, is typically determined by the type of tissue that is analyzed by the test performed by the technician (benign). The goal of this study was to automate IDC classification within breast cancer histology samples using a transfer learning technique. To improve our outcomes, we combined a Gradient Color Activation Mapping (Grad CAM) and image coloring mechanism with a discriminative fine-tuning methodology employing a one-cycle strategy using FastAI techniques. There have been lots of research studies related to deep transfer learning which use the same mechanism, but this report uses a transfer learning mechanism based on lightweight Squeeze Net architecture, a variant of CNN (Convolution neural network). This strategy demonstrates that fine-tuning on Squeeze Net makes it possible to achieve satisfactory results when transitioning generic features from natural images to medical images.

Keywords