A Classifier Model Using Fine-Tuned Convolutional Neural Network and Transfer Learning Approaches for Prostate Cancer Detection

被引:1
作者
Sariates, Murat [1 ]
Ozbay, Erdal [1 ]
机构
[1] Firat Univ, Dept Comp Engn, TR-23119 Elazig, Turkiye
来源
APPLIED SCIENCES-BASEL | 2025年 / 15卷 / 01期
关键词
prostate cancer; fine tuning; CNN; transfer learning; classification; deep learning;
D O I
10.3390/app15010225
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Background: Accurate and reliable classification models play a major role in clinical decision-making processes for prostate cancer (PCa) diagnosis. However, existing methods often demonstrate limited performance, particularly when applied to small datasets and binary classification problems. Objectives: This study aims to design a fine-tuned deep learning (DL) model capable of classifying PCa MRI images with high accuracy and to evaluate its performance by comparing it with various DL architectures. Methods: In this study, a basic convolutional neural network (CNN) model was developed and subsequently optimized using techniques such as L2 regularization, Tanh activation, dropout, and early stopping to enhance its performance. Additionally, a pyramid-type CNN architecture was designed to simultaneously evaluate both fine details and broader structures by combining low- and high-resolution information through feature maps extracted from different CNN layers. This approach enabled the model to learn complex features more effectively. For performance comparison, the developed fine-tuned enhanced pyramid network (FT-EPN) model was benchmarked against models such as Vgg16, Vgg19, Resnet50, InceptionV3, Densenet121, and Xception, which were trained using transfer learning (TL) techniques. It was also compared to next-generation models such as vision transformer (ViT) and MaxViT-v2. Results: The developed fine-tuned model achieved an accuracy rate of 96.77%, outperforming pre-trained TL models and next-generation models like ViT and MaxViT-v2. Among the TL models, Vgg19 achieved the highest accuracy rate at 92.74%. In comparison, ViT achieved an accuracy of 93.55%, while MaxViT-v2 achieved an accuracy of 95.16%. Conclusions: This study presents an optimized FT-EPN model to enhance the performance of DL models for PCa classification, offering a reference solution for future research. This model provides significant advantages in terms of classification accuracy and simplicity and has been evaluated as an effective solution in clinical applications.
引用
收藏
页数:19
相关论文
共 34 条
[1]   Detecting prostate cancer using deep learning convolution neural network with transfer learning approach [J].
Abbasi, Adeel Ahmed ;
Hussain, Lal ;
Awan, Imtiaz Ahmed ;
Abbasi, Imran ;
Majid, Abdul ;
Nadeem, Malik Sajjad Ahmed ;
Chaudhary, Quratul-Ain .
COGNITIVE NEURODYNAMICS, 2020, 14 (04) :523-533
[2]   Precise Identification of Prostate Cancer from DWI Using Transfer Learning [J].
Abdelmaksoud, Islam R. ;
Shalaby, Ahmed ;
Mahmoud, Ali ;
Elmogy, Mohammed ;
Aboelfetouh, Ahmed ;
Abou El-Ghar, Mohamed ;
El-Melegy, Moumen ;
Alghamdi, Norah Saleh ;
El-Baz, Ayman .
SENSORS, 2021, 21 (11)
[3]   Computer-aided classification of prostate cancer grade groups from MRI images using texture features and stacked sparse autoencoder [J].
Abraham, Bejoy ;
Nair, Madhu S. .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2018, 69 :60-68
[4]   A Deep Learning-Based Approach for the Detection and Localization of Prostate Cancer in T2 Magnetic Resonance Images [J].
Alkadi, Ruba ;
Taher, Fatma ;
El-baz, Ayman ;
Werghi, Naoufel .
JOURNAL OF DIGITAL IMAGING, 2019, 32 (05) :793-807
[5]   Prostate cancer grading framework based on deep transfer learning and Aquila optimizer [J].
Balaha, Hossam Magdy ;
Shaban, Ahmed Osama ;
El-Gendy, Eman M. ;
Saafan, Mahmoud M. .
NEURAL COMPUTING & APPLICATIONS, 2024, 36 (14) :7877-7902
[6]  
Bygari R., 2023, Machine Learning, Image Processing, Network Security and Data Sciences: Select Proceedings of 3rd International Conference on MIND 2021, P271, DOI 10.1007/978-981-19-5868-7_21
[7]   A Transfer Learning Approach for Malignant Prostate Lesion Detection on Multiparametric MRI [J].
Chen, Quan ;
Hu, Shiliang ;
Long, Peiran ;
Lu, Fang ;
Shi, Yujie ;
Li, Yunpeng .
TECHNOLOGY IN CANCER RESEARCH & TREATMENT, 2019, 18
[8]   Transfer Learning-Based Multi-Scale Denoising Convolutional Neural Network for Prostate Cancer Detection [J].
Chui, Kwok Tai ;
Gupta, Brij B. ;
Chi, Hao Ran ;
Arya, Varsha ;
Alhalabi, Wadee ;
Ruiz, Miguel Torres ;
Shen, Chien-Wen .
CANCERS, 2022, 14 (15)
[9]   Superpixel-based deep convolutional neural networks and active contour model for automatic prostate segmentation on 3D MRI scans [J].
da Silva, Giovanni L. F. ;
Diniz, Petterson S. ;
Ferreira, Jonnison L. ;
Franca, Joao V. F. ;
Silva, Aristofanes C. ;
de Paiva, Anselmo C. ;
de Cavalcanti, Elton A. A. .
MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2020, 58 (09) :1947-1964
[10]   Comparison of fine-tuning strategies for transfer learning in medical image classification [J].
Davila, Ana ;
Colan, Jacinto ;
Hasegawa, Yasuhisa .
IMAGE AND VISION COMPUTING, 2024, 146