Skip to main content
Article
A Transfer Learning with Structured Filter Pruning Approach for Improved Breast Cancer Classification on Point-Of-Care Devices
Computers in Biology and Medicine
  • Tejalal Choudhary
  • Vipul Mishra
  • Anurag Goswami
  • Jagannathan Sarangapani, Missouri University of Science and Technology
Abstract

Background and objective: A significant progress has been made in automated medical diagnosis with the advent of deep learning methods in recent years. However, deploying a deep learning model for mobile and small-scale, low-cost devices is a major bottleneck. Further, breast cancer is more prevalent currently, and ductal carcinoma being its most common type. Although many machine/deep learning methods have already been investigated, still, there is a need for further improvement. Method: This paper proposes a novel deep convolutional neural network (CNN) based transfer learning approach complemented with structured filter pruning for histopathological image classification, and to bring down the run-time resource requirement of the trained deep learning models. In the proposed method, first, the less important filters are pruned from the convolutional layers and then the pruned models are trained on the histopathological image dataset. Results: We performed extensive experiments using three popular pre-trained CNNs, VGG19, ResNet34, and ResNet50. With VGG19 pruned model, we achieved an accuracy of 91.25% outperforming earlier methods on the same dataset and architecture while reducing 63.46% FLOPs. Whereas, with the ResNet34 pruned model, the accuracy increases to 91.80% with 40.63% fewer FLOPs. Moreover, with the ResNet50 model, we achieved an accuracy of 92.07% with 30.97% less FLOPs. Conclusion: The experimental results reveal that the pre-trained model's performance complemented with filter pruning exceeds original pre-trained models. Another important outcome of the research is that the pruned model with reduced resource requirements can be deployed in point-of-care devices for automated diagnosis applications with ease.

Department(s)
Electrical and Computer Engineering
Keywords and Phrases
  • Breast Cancer,
  • Convolutional Neural Network,
  • Efficient Inference,
  • Filter Pruning,
  • Invasive Ductal Carcinoma
Document Type
Article - Journal
Document Version
Citation
File Type
text
Language(s)
English
Rights
© 2021 Elsevier, All rights reserved.
Publication Date
7-1-2021
Publication Date
01 Jul 2021
PubMed ID
33964737
Citation Information
Tejalal Choudhary, Vipul Mishra, Anurag Goswami and Jagannathan Sarangapani. "A Transfer Learning with Structured Filter Pruning Approach for Improved Breast Cancer Classification on Point-Of-Care Devices" Computers in Biology and Medicine Vol. 134 (2021) ISSN: 0010-4825
Available at: http://works.bepress.com/jagannathan-sarangapani/248/