Classification of Prostate Cancer in 3D Magnetic Resonance Imaging Data based on Convolutional Neural Networks (2404.10548v1)
Abstract: Prostate cancer is a commonly diagnosed cancerous disease among men world-wide. Even with modern technology such as multi-parametric magnetic resonance tomography and guided biopsies, the process for diagnosing prostate cancer remains time consuming and requires highly trained professionals. In this paper, different convolutional neural networks (CNN) are evaluated on their abilities to reliably classify whether an MRI sequence contains malignant lesions. Implementations of a ResNet, a ConvNet and a ConvNeXt for 3D image data are trained and evaluated. The models are trained using different data augmentation techniques, learning rates, and optimizers. The data is taken from a private dataset, provided by Cantonal Hospital Aarau. The best result was achieved by a ResNet3D, yielding an average precision score of 0.4583 and AUC ROC score of 0.6214.
- P. Rawla “Epidemiology of Prostate Cancer” In World journal of oncology 10, 2019 DOI: https://doi.org/10.14740/wjon1191
- “The current role of prostate multiparametric magnetic resonance imaging” In Asian Journal of Urology 6.2, 2019, pp. 137–146 DOI: 10.1016/j.ajur.2018.12.001
- “Automated Classification of Significant Prostate Cancer on MRI: A Systematic Review on the Performance of Machine Learning Applications” In Cancers 12.6, 2020 DOI: 10.3390/cancers12061606
- “Deep Learning-Based Gleason Grading of Prostate Cancer From Histopathology Images - Role of Multiscale Decision Aggregation and Data Augmentation” In IEEE journal of biomedical and health informatics 24.5, 2020, pp. 1413–1426 DOI: 10.1109/JBHI.2019.2944643
- “Deep learning-assisted differentiation of pathologically proven atypical and typical hepatocellular carcinoma (HCC) versus non-HCC on contrast-enhanced MRI of the liver” In European radiology 31.7, 2019, pp. 4981–4990 DOI: 10.1007/s00330-020-07559-1
- “Machine Learning in Prostate MRI for Prostate Cancer: Current Status and Future Opportunities” In Diagnostics (Basel, Switzerland), 2022 DOI: 10.3390/diagnostics12020289
- “Grading of prostatic adenocarcinoma current state and prognostic implications” In Diagn Pathol 11.25, 2016 DOI: 10.1186/s13000-016-0478-2
- “Semi-automatic classification of prostate cancer on multi-parametric MR imaging using a multi-channel 3D convolutional neural network” In European Radiology, 2019 DOI: 10.1007/s00330-019-06417-z
- “Clinically significant prostate cancer detection and segmentation in low-risk patients using a convolutional neural network on multi-parametric MRI” In European radiology, 2020 DOI: 10.1007/s00330-020-07008-z
- “Computer-Aided Diagnosis - A transfer learning approach for classification of clinical significant prostate cancers from mpMRI scans” In Medical Imaging 2017, 2017 DOI: 10.1117/12.2279021
- “Models Genesis” In Computer Vision and Pattern Recognition arXiv, 2020 DOI: 10.48550/ARXIV.2004.07882