On Convolutional Neural Networks and Transfer Learning for Classifying Breast Cancer on Histopathological Images Using GPU

被引:3
作者
Silva, D. C. S. E. [1 ]
Cortes, O. A. C. [1 ]
机构
[1] Inst Fed Maranhao, Comp Dept, Av Getulio Vargas 04, Sao Luis, Maranhao, Brazil
来源
XXVII BRAZILIAN CONGRESS ON BIOMEDICAL ENGINEERING, CBEB 2020 | 2022年
关键词
Deep learning; Transfer learning; Breast cancer; Detection; Image; GPU;
D O I
10.1007/978-3-030-70601-2_291
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
This paper presents a study about transfer learning using convolutional neural networks for detecting breast cancer in histopathological images. Transfer learning is a deep learning technique that reuses pre-trained neural network models to perform a particular task, which in this paper is to detect breast cancer in the referred images. Three convolutional architectures were tested: ResNet-18, ResNet-152, and GoogLeNet. The architectures were implemented in Python using PyTorch and trained using GPUs in the Google Colaboratory environment. Moreover, a random image processing stage was used to avoid overfitting, as well. Results indicate that ResNet-152 presents the best results reaching a mean accuracy of 84%, a precision of 90%, and Fl Score of 88%.
引用
收藏
页码:1993 / 1998
页数:6
相关论文
共 26 条
[1]  
American Institute for Cancer Research, 2020, BREAST CANC STAT
[2]  
[Anonymous], 2010, Handbook of Research on Machine Learning Applications and Trends: Algorithms, Methods, and Techniques, DOI DOI 10.4018/978-1-60566-766-9.CH011
[3]  
[Anonymous], 2009, IEEE COMPUTVIS PATTE
[4]   Diagnostic Assessment of Deep Learning Algorithms for Detection of Lymph Node Metastases in Women With Breast Cancer [J].
Bejnordi, Babak Ehteshami ;
Veta, Mitko ;
van Diest, Paul Johannes ;
van Ginneken, Bram ;
Karssemeijer, Nico ;
Litjens, Geert ;
van der Laak, Jeroen A. W. M. .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2017, 318 (22) :2199-2210
[5]  
Beysolow II T, 2017, INTRO DEEP LEARNING
[6]  
Chang J, 2017, 2017 IEEE 19TH INTERNATIONAL CONFERENCE ON E-HEALTH NETWORKING, APPLICATIONS AND SERVICES (HEALTHCOM)
[7]  
Google, 2020, GOOGL COL
[8]   Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs [J].
Gulshan, Varun ;
Peng, Lily ;
Coram, Marc ;
Stumpe, Martin C. ;
Wu, Derek ;
Narayanaswamy, Arunachalam ;
Venugopalan, Subhashini ;
Widner, Kasumi ;
Madams, Tom ;
Cuadros, Jorge ;
Kim, Ramasamy ;
Raman, Rajiv ;
Nelson, Philip C. ;
Mega, Jessica L. ;
Webster, R. .
JAMA-JOURNAL OF THE AMERICAN MEDICAL ASSOCIATION, 2016, 316 (22) :2402-2410
[9]  
He KM, 2015, Arxiv, DOI arXiv:1512.03385
[10]   Breast Cancer Classification in Ultrasound Images using Transfer Learning [J].
Hijab, Ahmed ;
Rushdi, Muhammad A. ;
Gomaa, Mohammed M. ;
Eldeib, Ayman .
2019 FIFTH INTERNATIONAL CONFERENCE ON ADVANCES IN BIOMEDICAL ENGINEERING (ICABME), 2019, :64-67