A Classifier Model Using Fine-Tuned Convolutional Neural Network and Transfer Learning Approaches for Prostate Cancer Detection
Source abstract
Abstract
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.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: AI in cancer detection · Medical Imaging and Analysis · Spectroscopy Techniques in Biomedical and Chemical Research
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Computer Engineering, Firat University, 23119 Elazig, Türkiye
Funding
Scientific Research Projects Management Unit of Firat University
References (34)
Show source-provided references
1. Transfer Learning with Pretrained Convolutional Neural Network for Automated Gleason Grading of Prostate Cancer Tissue Microarrays10.4103/jmss.jmss_42_22 ↗
2. Cancer statistics, 202410.3322/caac.21820 ↗
3. Chui, K.T., Gupta, B.B., Chi, H.R., Arya, V., Alhalabi, W., Ruiz, M.T., and Shen, C.W. (2022). Transfer learning-based multi-scale denoising convolutional neural network for prostate cancer detection. Cancers, 14.10.3390/cancers14153687 ↗
4. Prostate cancer grading framework based on deep transfer learning and Aquila optimizer10.1007/s00521-024-09499-z ↗
5. Abdelmaksoud, I.R., Shalaby, A., Mahmoud, A., Elmogy, M., Aboelfetouh, A., El-Ghar, M.A., El-Melegy, M., Alghamdi, N.S., and El-Baz, A. (2021). Precise identification of prostate cancer from DWI using transfer learning. Sensors, 21.10.3390/s21113664 ↗
6. Mehmood, M., Abbasi, S.H., Aurangzeb, K., Majeed, M.F., Anwar, M.S., and Alhussein, M. (2023). A classifier model for prostate cancer diagnosis using CNNs and transfer learning with multi-parametric MRI. Front. Oncol., 13.10.3389/fonc.2023.1225490 ↗
7. Transfer learning; powerful and fast segmentation and classification prostate cancer from MRI scans, in the development set10.3233/JIFS-224274 ↗
8. Comparison of fine-tuning strategies for transfer learning in medical image classification10.1016/j.imavis.2024.105012 ↗
9. El-Melegy, M., Mamdouh, A., Ali, S., Badawy, M., El-Ghar, M.A., Alghamdi, N.S., and El-Baz, A. (2024). Prostate Cancer Diagnosis via Visual Representation of Tabular Data and Deep Transfer Learning. Bioengineering, 11.10.3390/bioengineering11070635 ↗
10. Rippa, M., Schulze, R., Kenyon, G., Himstedt, M., Kwiatkowski, M., Grobholz, R., Wyler, S., Cornelius, A., Schindera, S., and Burn, F. (2024). Evaluation of Machine Learning Classification Models for False-Positive Reduction in Prostate Cancer Detection Using MRI Data. Diagnostics, 14.10.3390/diagnostics14151677 ↗
11. Dondi, F., Albano, D., Bertagna, F., and Treglia, G. (2022). Bone scintigraphy versus PSMA-targeted PET/CT or PET/MRI in prostate cancer: Lessons learned from recent systematic reviews and meta-analyses. Cancers, 14.10.3390/cancers14184470 ↗
12. PSMA PET/CT Accuracy in Diagnosing Prostate Cancer Nodes Metastases10.21873/invivo.13769 ↗
13. Fiorentino, V., Martini, M., Dell’aquila, M., Musarra, T., Orticelli, E., Larocca, L.M., Rossi, E., Totaro, A., Pinto, F., and Lenci, N. (2020). Histopathological ratios to predict Gleason score agreement between biopsy and radical prostatectomy. Diagnostics, 11.10.3390/diagnostics11010010 ↗
14. Sekhoacha, M., Riet, K., Motloung, P., Gumenku, L., Adegoke, A., and Mashele, S. (2022). Prostate cancer review: Genetics, diagnosis, treatment options, and alternative approaches. Molecules, 27.10.3390/molecules27175730 ↗
15. Detecting prostate cancer using deep learning convolution neural network with transfer learning approach10.1007/s11571-020-09587-5 ↗
16. A deep learning approach to diagnostic classification of prostate cancer using pathology–radiology fusion10.1002/jmri.27599 ↗
17. Superpixel-based deep convolutional neural networks and active contour model for automatic prostate segmentation on 3D MRI scans10.1007/s11517-020-02199-5 ↗
18. Capsule GAN for prostate MRI super-resolution10.1007/s11042-021-11697-z ↗
19. A transfer learning approach for malignant prostate lesion detection on multiparametric MRI10.1177/1533033819858363 ↗
20. Prostate cancer detection using residual networks10.1007/s11548-019-01967-5 ↗
21. A novel deep learning-based technique for detecting prostate cancer in MRI images10.1007/s11042-023-15793-0 ↗
22. Deep learning in prostate cancer diagnosis and Gleason grading in histopathology images: An extensive study10.1016/j.imu.2021.100582 ↗
23. Rabilloud, N., Allaume, P., Acosta, O., De Crevoisier, R., Bourgade, R., Loussouarn, D., Rioux-Leclercq, N., Khene, Z.-E., Mathieu, R., and Bensalah, K. (2023). Deep learning methodologies applied to digital pathology in prostate cancer: A systematic review. Diagnostics, 13.10.3390/diagnostics13162676 ↗
24. A deep learning-based approach for the detection and localization of prostate cancer in T2 magnetic resonance images10.1007/s10278-018-0160-1 ↗
25. Viswanath, S.E., Chirra, P.V., Yim, M.C., Rofsky, N.M., Purysko, A.S., Rosen, M.A., Bloch, B.N., and Madabhushi, A. (2019). Comparing radiomic classifiers and classifier ensembles for detection of peripheral zone prostate tumors on T2-weighted MRI: A multi-site study. BMC Med. Imaging, 19.10.1186/s12880-019-0308-6 ↗
26. Computer-aided classification of prostate cancer grade groups from MRI images using texture features and stacked sparse autoencoder10.1016/j.compmedimag.2018.08.006 ↗
27. Computer-aided diagnosis of prostate cancer using a deep convolutional neural network from multiparametric MRI10.1002/jmri.26047 ↗
28. PI-RADSAI: Introducing a new human-in-the-loop AI model for prostate cancer diagnosis based on MRI10.1038/s41416-022-02137-2 ↗
29. Bygari, R., Rithesh, K., Ambesange, S., and Koolagudi, S.G. (2023). Prostate Cancer Grading Using Multistage Deep Neural Networks. Machine Learning, Image Processing, Network Security and Data Sciences: Select Proceedings of 3rd International Conference on MIND 2021, Springer Nature.10.1007/978-981-19-5868-7_21 ↗
30. Zhu, L., Shi, H., Wei, H., Wang, C., Shi, S., Zhang, F., Yan, R., Liu, Y., He, T., and Wang, L. (2023). An accurate prediction of the origin for bone metastatic cancer using deep learning on digital pathological images. EBioMedicine, 87.10.1016/j.ebiom.2022.104426 ↗
31. Talaat, F.M., El-Sappagh, S., Alnowaiser, K., and Hassan, E. (2024). Improved prostate cancer diagnosis using a modified ResNet50-based deep learning architecture. BMC Med. Inform. Decis. Mak., 24.10.1186/s12911-024-02419-0 ↗
32. Automated prostate cancer grading and diagnosis system using deep learning-based Yolo object detection algorithm10.1016/j.eswa.2022.117148 ↗
33. Vision Transformers in medical computer vision—A contemplative retrospection10.1016/j.engappai.2023.106126 ↗
34. Maxvit: Multi-axis vision transformer