Breast cancer histology images classification: Training from scratch or transfer learning?

被引:107
作者
Shallu [1 ]
Mehra, Rajesh [1 ]
机构
[1] Natl Inst Tech Teachers Training & Res, Dept Elect & Commun, Chandigarh 160019, India
关键词
Breast cancer; Histopathological images; Convolutional neural network; Full training; Transfer learning; CONVOLUTIONAL NEURAL-NETWORKS; FEATURES; DATASET; RISK;
D O I
10.1016/j.icte.2018.10.007
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
We demonstrated the ability of transfer learning in comparison with the fully-trained network on the histopathological imaging modality by considering three pre-trained networks: VGG16, VGG19, and ResNet50 and analyzed their behavior for magnification independent breast cancer classification. Concurrently, we examined the effect of training-testing data size on the performance of considered networks. A fine-tuned pre-trained VGG16 with logistic regression classifier yielded the best performance with 92.60% accuracy, 95.65% area under ROC curve (AUC), and 95.95% accuracy precision score (APS) for 90%-10% training-testing data splitting. Layer-wise fine-tuning and different weight initialization schemes can be a future aspect of this study. (C) 2018 The Korean Institute of Communications and Information Sciences (KICS). Publishing Services by Elsevier B.V.
引用
收藏
页码:247 / 254
页数:8
相关论文
共 35 条
[1]   Early detection: a long road ahead [J].
不详 .
NATURE REVIEWS CANCER, 2018, 18 (07) :401-401
[2]  
[Anonymous], IEEE INT C NAN INSTR
[3]  
[Anonymous], 2014, ARXIV14065774
[4]  
[Anonymous], 2015, C COMPUTER VISION PA
[5]  
[Anonymous], IEEE T BIOMED ENG
[6]  
[Anonymous], LUNG PATTERN CLASSIF
[7]  
[Anonymous], INT C LEARN REPR
[8]  
[Anonymous], ARXIV180402491
[9]  
[Anonymous], COMPUT INTEL NEUROSC
[10]   Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network [J].
Anthimopoulos, Marios ;
Christodoulidis, Stergios ;
Ebner, Lukas ;
Christe, Andreas ;
Mougiakakou, Stavroula .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1207-1216