Journal of Shandong University(Engineering Science) ›› 2019, Vol. 49 ›› Issue (2): 74-79.doi: 10.6040/j.issn.1672-3961.0.2018.273

• Machine Learning & Data Mining • Previous Articles     Next Articles

Computer aided diagnosis method for breast cancer based on AlexNet and ensemble classifiers

Xiaoxiong HOU1,2(),Xinzheng XU1,2,*(),Jiong ZHU1,Yanyan GUO1   

  1. 1. School of Computer Science and Technology, China University of Mining and Technology, Xuzhou 221116, Jiangsu, China
    2. Guangxi High School Key Laboratory of Complex System and Computational Intelligence, Nanning 530006, Guangxi, China
  • Received:2018-07-06 Online:2019-04-20 Published:2019-04-19
  • Contact: Xinzheng XU E-mail:961458517@qq.com;xuxinzh@163.com
  • Supported by:
    国家自然科学基金项目(61672522);广西高校复杂系统与智能计算重点实验室开放课题重点项目(2017CSCI01)

RichHTML

15

PDF (PC)

780

Abstract:

In order to solve the manual feature extraction of medical images in computer aided diagnosis, Alexnet was pre-trained on the ImageNet dataset, and feature extraction was performed on the medical image based on Alexnet with transfer learning. The ensemble learning method was used to train the classifier to classify and obtain a better classification effect than the single classifier. The results showed that the AUC(area under curve) of Alexnet deep learning model and random forest ensemble classifier reached 0.87±0.03, and the effect of the ensemble classifier was better than that of the single classifier in the same network depth.

Key words: medical image analysis, deep learning, convolutional neural network, computer aided diagnosis, ensemble classifiers

CLC Number: 

  • TP391

Fig.1

The workflow chart of CAD"

Fig.2

Steps carried out in preprocessing of mammography"

Fig.3

Structure of Alexnet network"

Table 1

Corresponding test AUC of different depthsCNN with SVM classifier"

特征提取模型 AUC
CNN(2层) 0.76±0.05
CNN(3层) 0.82±0.03
Alexnet 0.84±0.06

Table 2

Corresponding test AUC of Alexnet model withdifferent classifier"

分类器 AUC值
SVM 0.84±0.06
Logistic Regression 0.83±0.03
Softmax 0.85±0.04

Table 3

Test AUC values of Alexnet model withensemble classifier"

分类器 AUC值
Softmax 0.85±0.04
Random Forest 0.87±0.03
1 BUNGE R E , HERMAN C L . Usage of diagnostic imaging procedures: a nationwide hospital study[J]. Radiology, 1987, 163 (2): 569- 573.
doi: 10.1148/radiology.163.2.3550886
2 SLUIMER I , SCHILHAM A , PROKOP M , et al. Computer analysis of computed tomography scans of the lung: a survey[J]. IEEE Trans Med Imaging, 2006, 25 (4): 385- 405.
doi: 10.1109/TMI.2005.862753
3 QUEKEL L G , KESSELS A G , GOEI R , et al. Miss rate of lung cancer on the chest radiograph in clinical practice[J]. Chest, 1999, 115 (3): 720- 724.
doi: 10.1378/chest.115.3.720
4 LI F , SONE S , ABE H , et al. Lung cancers missed at low-dose helical CT screening in a general population: comparison of clinical, histopathologic, and imaging findings[J]. Radiology, 2002, 225 (3): 673- 683.
doi: 10.1148/radiol.2253011375
5 DOI K , MACMAHON H , KATSURAGAWA S , et al. Computer-aided diagnosis in radiology: potential and pitfalls[J]. European Journal of Radiology, 1999, 31 (2): 97- 109.
doi: 10.1016/S0720-048X(99)00016-9
6 GIGER M L , CHAN H P , BOONE J . Anniversary paper: history and status of CAD and quantitative image analysis: the role of medical physics and AAPM[J]. Medical Physics, 2008, 35 (12): 5799- 5820.
doi: 10.1118/1.3013555
7 LI Q , LI F , SUZUKI K , et al. Computer-aided diagnosis in thoracic CT[J]. Seminars in Ultrasound, CT and MRI, 2005, 26 (5): 357- 363.
doi: 10.1053/j.sult.2005.07.001
8 HINTON G E , OSINDERO S , TEH Y W . A fast learning algorithm for deep belief nets[J]. Neural Computation, 2006, 18 (7): 1527- 1554.
doi: 10.1162/neco.2006.18.7.1527
9 MCLOUGHLIN K J , BONES P J , KARSSEMEIJER N . Noise equalization for detection of microcalcification clusters in direct digital mammogram images[J]. IEEE Transactions on Medical Imaging, 2004, 23 (3): 313- 320.
doi: 10.1109/TMI.2004.824240
10 TALHA M , SULONG G B , JAFFAR A . Preprocessing digital breast mammograms using adaptive weighted frost filter[J]. Biomedical Research, 2016, 27 (4): 1407- 1412.
11 PONRAJ D N , JENIFER M E , POONGODI P , et al. A survey on the preprocessing techniques of mammogram for the detection of breast cancer[J]. Journal of Emerging Trends in Computing & Information Sciences, 2011, 2 (12): 656- 664.
12 MAKANDAR A , HALALLI B . Pre-processing of mammography image for early detection of breast bancer[J]. International Journal of Computer Applications, 2016, 144 (3): 11- 15.
doi: 10.5120/ijca2016910153
13 RAMANI R , VANITHA N S , VALARMATHY S . The pre-processing techniques for breast cancer detection in mammography images[J]. International Journal of Image Graphics & Signal Processing, 2013, 5 (5): 47- 54.
14 MUSTRA M , GRGIC M , RANGAYYAN R M . Review of recent advances in segmentation of the breast boundary and the pectoral muscle in mammograms[J]. Medical & Biological Engineering & Computing, 2016, 54 (7): 1003- 1024.
15 MAKANDAR A , HALALLI B . Breast cancer image enhancement using median filter and clahe[J]. International Journal of Scientific & Engineering Research, 2015, 6 (4): 462- 465.
16 KRIZHEVSKY A, SUTSKEVER I, HINTON G E. ImageNet classification with deep convolutional neural networks[C]//International Conference on Neural Information Processing Systems. Vancouver, Canada: Neural Information Processing Systems Foundation, 2012: 1097-1105.
17 DONAHUE J, JIA Y, VINYALS O, et al. DeCAF: a deep convolutional activation feature for generic visual recognition[J]. arXiv: 1310.1531 [cs.CV], 2013: 1-10.
18 RUSSAKOVSKY O , DENG J , SU H , et al. ImageNet large scale visual recognition challenge[J]. International Journal of Computer Vision, 2014, 115 (3): 211- 252.
19 BREIMAN L . Random Forest[J]. Machine Learning, 2001, 45 (1): 5- 32.
20 BREIMAN L . Bagging predictors[J]. Machine Learning, 1996, 24 (2): 123- 140.
21 AREVALO J , GONZ LEZ F A , RAMOS-POLL N R , et al. Representation learning for mammography mass lesion classification with convolutional neural networks[J]. Computer Methods & Programs in Biomedicine, 2016, 127 (C): 248- 257.
[1] LI Changgang, LI Baoliang, CAO Yongji, WANG Jiaying. Review and prospect on artificial intelligence application in power system power flow calculation [J]. Journal of Shandong University(Engineering Science), 2025, 55(5): 1-17.
[2] ZHOU Qunying, SUI Jiacheng, ZHANG Ji, WANG Hongyuan. Industrial product surface defect detection based on self supervised convolution and parameter free attention mechanism [J]. Journal of Shandong University(Engineering Science), 2025, 55(4): 40-47.
[3] XUE Bingbing, WANG Yong, YANG Weihao, WANG Chuan, YU Di, WANG Xu. Real-time expressway traffic data imputation and state prediction based on ETC system data [J]. Journal of Shandong University(Engineering Science), 2025, 55(3): 58-71.
[4] DONG Mingshu, CHEN Liqi, MA Chuanyi, ZHANG Zhuhao, SUN Renjuan, GUAN Yanhua, ZHUANG Peizhi. Deep learning-based intelligent judgment for radar detection of pavement cracks [J]. Journal of Shandong University(Engineering Science), 2025, 55(3): 72-79.
[5] Jiachun LI,Bowen LI,Jianbo CHANG. An efficient and lightweight RGB frame-level face anti-spoofing model [J]. Journal of Shandong University(Engineering Science), 2023, 53(6): 1-7.
[6] Yue YUAN,Yanli WANG,Kan LIU. Named entity recognition model based on dilated convolutional block architecture [J]. Journal of Shandong University(Engineering Science), 2022, 52(6): 105-114.
[7] Tongyu JIANG, Fan CHEN, Hongjie HE. Lightweight face super-resolution network based on asymmetric U-pyramid reconstruction [J]. Journal of Shandong University(Engineering Science), 2022, 52(1): 1-8.
[8] Jianqing WU,Xiuguang SONG. Review on development of simultaneous localization and mapping technology [J]. Journal of Shandong University(Engineering Science), 2021, 51(5): 16-31.
[9] YANG Xiuyuan, PENG Tao, YANG Liang, LIN Hongfei. Adaptive multi-domain sentiment analysis based on knowledge distillation [J]. Journal of Shandong University(Engineering Science), 2021, 51(3): 15-21.
[10] Qingfa CHAI,Shoujing SUN,Jifu QIU,Ming CHEN,Zhen WEI,Wei CONG. Prediction method of power grid emergency supplies under meteorological disasters [J]. Journal of Shandong University(Engineering Science), 2021, 51(3): 76-83.
[11] LIAO Jinping, MO Yuchang, YAN Ke. Model and application of short-term electricity consumption forecast based on C-LSTM [J]. Journal of Shandong University(Engineering Science), 2021, 51(2): 90-97.
[12] Guoyong CAI,Xinhao HE,Yangyang CHU. Visual sentiment analysis based on spatial attention mechanism and convolutional neural network [J]. Journal of Shandong University(Engineering Science), 2020, 50(4): 8-13.
[13] LIAO Nanxing, ZHOU Shibin, ZHANG Guopeng, CHENG Deqiang. Image caption generation method based on class activation mapping and attention mechanism [J]. Journal of Shandong University(Engineering Science), 2020, 50(4): 28-34.
[14] LIU Shuai, WANG Lei, DING Xutao. Emotional EEG recognition based on Bi-LSTM [J]. Journal of Shandong University(Engineering Science), 2020, 50(4): 35-39.
[15] Shiqi SONG,Yan PIAO,Zexin JIANG. Vehicle classification and tracking for complex scenes based on improved YOLOv3 [J]. Journal of Shandong University(Engineering Science), 2020, 50(2): 27-33.
Viewed
Full text


Abstract

Cited
  Shared   
  Discussed   
[1] SHI Lai-shun,WAN Zhong-yi . Synthesis and performance evaluation of a novel betaine-type asphalt emulsifier[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2008, 38(4): 112 -115 .
[2] CHEN Rui, LI Hongwei, TIAN Jing. The relationship between the number of magnetic poles and the bearing capacity of radial magnetic bearing[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2018, 48(2): 81 -85 .
[3] YUE Yuan-Zheng. Relaxation in glasses far from equilibrium[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(5): 1 -20 .
[4] SUN Cong-zheng,GUAN Cong-sheng,QIN Jing-yu,CHENG Chuan . The structure and performances of the electroless Ni-P alloy coating on aluminum alloy[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2007, 37(5): 108 -112 .
[5] LIU Xin 1, SONG Sili 1, WANG Xinhong 2. [J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(2): 98 -100 .
[6] . [J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(2): 108 -112 .
[7] HU Tian-liang,LI Peng,ZHANG Cheng-rui,ZUO Yi . Design of a QEP decode counter based on VHDL[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2008, 38(3): 10 -13 .
[8] . [J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(2): 104 -107 .
[9] . [J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(2): 131 -136 .
[10] GAO Hou-Lei, TIAN Jia, DU Jiang, WU Zhi-Gang, LIU Chu-Min. Distributed generation—new technology in energy development[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(5): 106 -110 .
Copyright 2018 © Editorial office of Journal of Shandong University (Engineering Science)
Supported by Beijing Magtech Co.Ltd