Deep ensemble transfer learning-based framework for mammographic image classification

被引:14
作者
Oza, Parita [1 ,2 ]
Sharma, Paawan [1 ]
Patel, Samir [1 ]
机构
[1] Pandit Deendayal Energy Univ, Gandhinagar 382007, Gujarat, India
[2] Nirma Univ, Ahmadabad 382481, Gujarat, India
关键词
Breast cancer; Mammograms; Deep learning; Ensemble learning; BREAST; MODEL;
D O I
10.1007/s11227-022-04992-5
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
This research intends to provide a method for clinical decision support systems that can accurately classify benign and malignant mass from breast X-ray images. The model was initially trained and assessed using distinct convolutional neural network (CNN) models. Based on the comparative analysis, the best models were then selected and used for further implementation. The work employs an average, weighted average and concatenation strategy that seeks to merge pre-trained CNN, relying on the transfer learning technique to construct a highly accurate ensemble model. The models make it possible to save and use the information gained from a pre-trained CNN for a new task, namely breast mammogram classification. A benchmark datasets such as MIAS, CBIS-DDSM and a private dataset with two classes, benign and malignant, were used in the proposed approach. To assess the proposed model's efficacy, several generic assessment techniques were employed. Our ensemble model outperforms other state-of-the-art methods with an overall accuracy (99.76%), Sensitivity (99.84%), Specificity(99.76%), Precision(99.76%) and F1-Score(99.76%).
引用
收藏
页码:8048 / 8069
页数:22
相关论文
共 46 条
[1]   Comparison of clinical outcomes between luminal invasive ductal carcinoma and luminal invasive lobular carcinoma [J].
Adachi, Yayoi ;
Ishiguro, Junko ;
Kotani, Haruru ;
Hisada, Tomoka ;
Ichikawa, Mari ;
Gondo, Naomi ;
Yoshimura, Akiyo ;
Kondo, Naoto ;
Hattori, Masaya ;
Sawaki, Masataka ;
Fujita, Takashi ;
Kikumori, Toyone ;
Yatabe, Yasushi ;
Kodera, Yasuhiro ;
Iwata, Hiroji .
BMC CANCER, 2016, 16
[2]   Performance Evaluation of Deep Learning Models on Mammogram Classification Using Small Dataset [J].
Adedigba, Adeyinka P. ;
Adeshina, Steve A. ;
Aibinu, Abiodun M. .
BIOENGINEERING-BASEL, 2022, 9 (04)
[3]   The performances of iterative type-2 fuzzy C-mean on GPU for image segmentation [J].
Ali, Noureddine Ait ;
El Abbassi, Ahmed ;
Cherradi, Bouchaib .
JOURNAL OF SUPERCOMPUTING, 2022, 78 (02) :1583-1601
[4]   Automated Breast Cancer Detection Models Based on Transfer Learning [J].
Alruwaili, Madallah ;
Gouda, Walaa .
SENSORS, 2022, 22 (03)
[5]   Detecting Rumors on Social Media Based on a CNN Deep Learning Technique [J].
Alsaeedi, Abdullah ;
Al-Sarem, Mohammed .
ARABIAN JOURNAL FOR SCIENCE AND ENGINEERING, 2020, 45 (12) :10813-10844
[6]   A hybrid deep learning model for breast cancer diagnosis based on transfer learning and pulse-coupled neural networks [J].
Altaf, Meteb M. .
MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2021, 18 (05) :5029-5046
[7]   Breast Tumor Classification Using an Ensemble Machine Learning Method [J].
Assiri, Adel S. ;
Nazir, Saima ;
Velastin, Sergio A. .
JOURNAL OF IMAGING, 2020, 6 (06)
[8]   An integrated framework for breast mass classification and diagnosis using stacked ensemble of residual neural networks [J].
Baccouche, Asma ;
Garcia-Zapirain, Begonya ;
Elmaghraby, Adel S. .
SCIENTIFIC REPORTS, 2022, 12 (01)
[9]   Invasive lobular and ductal breast carcinoma differ in immune response, protein translation efficiency and metabolism [J].
Du, Tian ;
Zhu, Li ;
Levine, Kevin M. ;
Tasdemir, Nilgun ;
Lee, Adrian V. ;
Vignali, Dario A. A. ;
Van Houten, Bennett ;
Tseng, George C. ;
Oesterreich, Steffi .
SCIENTIFIC REPORTS, 2018, 8
[10]   Ensemble deep learning: A review [J].
Ganaie, M. A. ;
Hu, Minghui ;
Malik, A. K. ;
Tanveer, M. ;
Suganthan, P. N. .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2022, 115