Breast Cancer Medical Image Analysis Based on Transfer Learning Model

被引:0
作者
Liu, Yi [1 ,2 ]
Zhang, Xiaolong [1 ,2 ]
机构
[1] Wuhan Univ Sci & Technol, Sch Comp Sci & Technol, Wuhan 430065, Hubei, Peoples R China
[2] Intelligent Informat Proc & Real Time Ind Syst Hu, Wuhan 430065, Hubei, Peoples R China
来源
INTELLIGENT COMPUTING THEORIES AND APPLICATION, PT II | 2018年 / 10955卷
基金
中国国家自然科学基金;
关键词
Breast cancer; Transfer learning; Convolutional Neural Network;
D O I
10.1007/978-3-319-95933-7_6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Breast cancer becomes one of the common cancer among women. The computer-aided diagnosis is a very available technology, and it becomes an inevitable trend of modern medicine. There is litter data in the medical field which is not enough to support training model. The paper aims to solve the problem in the case of insufficient data volume, and a transfer learning model combined with Convolutional Neural Network (CNN) is proposed to achieve the goal. This model has three innovations. The first one takes Xavier method to initialize parameter, which can make the training process more stable. The second innovation takes dropout method to discard some network nodes randomly, which can reduce overfitting problem. The third one adds two convolutional layers and one max pooling layer before final fully connected layer. The experimental results have shown that this strategy is suitable for the problem in this paper. The paper indicates that the transfer learning model is an effective method with small-scale data, and it can be combined with deep learning algorithms.
引用
收藏
页码:44 / 53
页数:10
相关论文
共 19 条
[1]  
[Anonymous], 2001, DIGITAL MAMMOGRAPHY
[2]  
[Anonymous], ADV NEURAL INF PROCE
[3]  
[Anonymous], IEEE ACM T COMPUT BI
[4]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[5]   Computer-aided diagnosis in medical imaging: Historical review, current status and future potential [J].
Doi, Kunio .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2007, 31 (4-5) :198-211
[6]  
Glorot X., 2010, P 13 INT C ART INT S, P249
[7]   BREAST-CANCER - MAGNITUDE OF THE PROBLEM AND DESCRIPTIVE EPIDEMIOLOGY - INTRODUCTION [J].
KELSEY, JL ;
HORNROSS, PL .
EPIDEMIOLOGIC REVIEWS, 1993, 15 (01) :7-16
[8]   Deep learning [J].
LeCun, Yann ;
Bengio, Yoshua ;
Hinton, Geoffrey .
NATURE, 2015, 521 (7553) :436-444
[9]   A NOVEL SHAPE FEATURE TO CLASSIFY MICROCALCIFICATIONS [J].
Ma, Yiming ;
Tay, Peter C. ;
Adams, Robert D. ;
Zhang, James Z. .
2010 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, 2010, :2265-2268
[10]   A Survey on Transfer Learning [J].
Pan, Sinno Jialin ;
Yang, Qiang .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2010, 22 (10) :1345-1359