Deep ensemble transfer learning-based framework for mammographic image classification

被引:9
|
作者
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
来源
JOURNAL OF SUPERCOMPUTING | 2023年 / 79卷 / 07期
关键词
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
相关论文
共 50 条
  • [21] A deep learning-based framework for retinal fundus image enhancement
    Lee, Kang Geon
    Song, Su Jeong
    Lee, Soochahn
    Yu, Hyeong Gon
    Kim, Dong Ik
    Lee, Kyoung Mu
    PLOS ONE, 2023, 18 (03):
  • [22] Deep learning-based common skin disease image classification
    Nath, Sudarshan
    Das Gupta, Suparna
    Saha, Soumyabrata
    JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2023, 44 (05) : 7483 - 7499
  • [23] An efficient deep learning-based framework for image distortion correction
    Li, Sicheng
    Chu, Yuhui
    Zhao, Yunpeng
    Zhao, Pengpeng
    VISUAL COMPUTER, 2024, 40 (10): : 6955 - 6967
  • [24] Deep learning-based image classification of turtles imported into Korea
    Jong-Won Baek
    Jung-Il Kim
    Chang-Bae Kim
    Scientific Reports, 13
  • [25] Deep Ensemble Learning for Retinal Image Classification
    Ho, Edward
    Wang, Edward
    Youn, Saerom
    Sivajohan, Asaanth
    Lane, Kevin
    Chun, Jin
    Hutnik, Cindy M. L.
    TRANSLATIONAL VISION SCIENCE & TECHNOLOGY, 2022, 11 (10):
  • [26] A Deep Learning-Based Interactive Medical Image Segmentation Framework
    Mikhailov, Ivan
    Chauveau, Benoit
    Bourdel, Nicolas
    Bartoli, Adrien
    APPLICATIONS OF MEDICAL ARTIFICIAL INTELLIGENCE, AMAI 2022, 2022, 13540 : 98 - 107
  • [27] Impact of image compression on deep learning-based mammogram classification
    Yong-Yeon Jo
    Young Sang Choi
    Hyun Woo Park
    Jae Hyeok Lee
    Hyojung Jung
    Hyo-Eun Kim
    Kyounglan Ko
    Chan Wha Lee
    Hyo Soung Cha
    Yul Hwangbo
    Scientific Reports, 11
  • [28] Selective Inference for Accelerating Deep Learning-based Image Classification
    Lee, HyunYong
    Lee, Byung-Tak
    2016 INTERNATIONAL CONFERENCE ON INFORMATION AND COMMUNICATION TECHNOLOGY CONVERGENCE (ICTC 2016): TOWARDS SMARTER HYPER-CONNECTED WORLD, 2016, : 135 - 137
  • [29] Deep Learning Ensemble for Hyperspectral Image Classification
    Chen, Yushi
    Wang, Ying
    Gu, Yanfeng
    He, Xin
    Ghamisi, Pedram
    Jia, Xiuping
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2019, 12 (06) : 1882 - 1897
  • [30] Deep learning-based image classification of turtles imported into Korea
    Baek, Jong-Won
    Kim, Jung-Il
    Kim, Chang-Bae
    SCIENTIFIC REPORTS, 2023, 13 (01)