Transformer-inspired training principles based breast cancer prediction: combining EfficientNetB0 and ResNet50

被引:0
|
作者
Tariq Shahzad [1 ]
Tehseen Mazhar [2 ]
Sheikh Muhammad Saqib [4 ]
Khmaies Ouahada [3 ]
机构
[1] University of Johannesburg,Department of Electrical and Electronic Engineering Science
[2] National College of Business Administration and Economics,School of Computer Science
[3] Gomal University,Institute of Computing and Information Technology
[4] Government of Punjab,Department of Computer Science, School Education Department
关键词
Invasive ductal carcinoma (IDC) and Non-IDC categories; EfficientNetB0; ResNet50; Mean absolute error (MAE); Matthews correlation coefficient (MCC);
D O I
10.1038/s41598-025-98523-w
中图分类号
学科分类号
摘要
Breast cancer is a leading killer and has been deepened by COVID-19, which affected diagnosis and treatment services. The absence of a rapid, efficient, accurate diagnostic tool remains a pressing issue for this severe disease. Thus, it is still possible to encounter issues concerning diagnostic accuracy and utilization of errors in the sphere of machine learning, deep learning, and transfer learning models. This paper presents a new model combining EfficientNetB0 and ResNet50 to improve the classification of breast histopathology images into IDC and non-IDC classes. The implementation steps, it include resizing all the images to be of a standard size of 128*128 pixels and then performing normalization to enhance the learning model. EfficientNetB0 is selected for its efficient yet effective performance while ResNet50 employs deep residual connections to overcome the vanishing gradient problem. The proposed model that incorporates some of the characteristics from both architectures turns out to be very resilient and accurate in classification. The model demonstrates superior performance with an accuracy of 94%, a Mean Absolute Error (MAE) of 0.0628, and a Matthews Correlation Coefficient (MCC) of 0.8690. These results outperform previous baselines and show that the model performs well in achieving a good trade-off between precision and recall. The comparison with the related works demonstrates the superiority of the proposed ensemble approach in terms of accuracy and complexity, which makes it efficient for practical breast cancer diagnosis and screening.
引用
收藏
相关论文
共 4 条
  • [1] A New Breast Cancer Diagnosis Application Based on ResNet50
    Le, Anjie
    Li, Zhenghao
    Tang, Haoyun
    Yang, Haobo
    SECOND IYSF ACADEMIC SYMPOSIUM ON ARTIFICIAL INTELLIGENCE AND COMPUTER ENGINEERING, 2021, 12079
  • [2] VGG-16, VGG-16 With Random Forest, Resnet50 With SVM, and EfficientNetB0 With XGBoost-Enhancing Bone Fracture Classification in X-Ray Using Deep Learning Models
    Torne, Spoorthy
    Shetty, Dasharathraj K.
    Makkithaya, Krishnamoorthi
    Hegde, Prasiddh
    Sudhi, Manu
    Kumar Pullela, Phani
    Tamil Eniyan, T.
    Kamath, Ritesh
    Salu, Staissy
    Bhat, Pranav
    Girisha, S.
    Priya, P. S.
    IEEE ACCESS, 2025, 13 : 25568 - 25577
  • [3] Optimizing Breast Cancer Mammogram Classification Through a Dual Approach: A Deep Learning Framework Combining ResNet50, SMOTE, and Fully Connected Layers for Balanced and Imbalanced Data
    Alshamrani, Abdullah Fahad A.
    Alshomrani, Faisal Saleh Zuhair
    IEEE ACCESS, 2025, 13 : 4815 - 4826
  • [4] Deep Learning-based Automatic Diagnosis of Breast Cancer on MRI Using Mask R-CNN for Detection Followed by ResNet50 for Classification
    Zhang, Yang
    Liu, Yan-Lin
    Nie, Ke
    Zhou, Jiejie
    Chen, Zhongwei
    Chen, Jeon-Hor
    Wang, Xiao
    Kim, Bomi
    Parajuli, Ritesh
    Mehta, Rita S.
    Wang, Meihao
    Su, Min-Ying
    ACADEMIC RADIOLOGY, 2023, 30 : S161 - S171