Boosted efficientnet: Detection of lymph node metastases in breast cancer using convolutional neural networks

Jun Wang, Qianying Liu, Haotian Xie, Zhaogang Yang, Hefeng Zhou

Research output: Contribution to journalArticlepeer-review

64 Scopus citations

Abstract

(1) Purpose: To improve the capability of EfficientNet, including developing a cropping method called Random Center Cropping (RCC) to retain the original image resolution and significant features on the images’ center area, reducing the downsampling scale of EfficientNet to facilitate the small resolution images of RPCam datasets, and integrating attention and Feature Fusion (FF) mechanisms with EfficientNet to obtain features containing rich semantic information. (2) Methods: We adopt the Convolutional Neural Network (CNN) to detect and classify lymph node metastasis in breast cancer. (3) Results: Experiments illustrate that our methods significantly boost performance of basic CNN architectures, where the best-performed method achieves an accuracy of 97.96% ± 0.03% and an Area Under the Curve (AUC) of 99.68% ± 0.01% on RPCam datasets, respectively. (4) Conclu-sions: (1) To our limited knowledge, we are the only study to explore the power of EfficientNet on Metastatic Breast Cancer (MBC) classification, and elaborate experiments are conducted to compare the performance of EfficientNet with other state-of-the-art CNN models. It might provide inspiration for researchers who are interested in image-based diagnosis using Deep Learning (DL). (2) We design a novel data augmentation method named RCC to promote the data enrichment of small resolution datasets. (3) All of our four technological improvements boost the performance of the original EfficientNet.

Original languageEnglish (US)
Article number661
Pages (from-to)1-14
Number of pages14
JournalCancers
Volume13
Issue number4
DOIs
StatePublished - Feb 2 2021

Keywords

  • CNN
  • Diagnosis
  • Feature fusion
  • Image classification
  • Metastatic breast cancer

ASJC Scopus subject areas

  • Oncology
  • Cancer Research

Fingerprint

Dive into the research topics of 'Boosted efficientnet: Detection of lymph node metastases in breast cancer using convolutional neural networks'. Together they form a unique fingerprint.

Cite this