Next Article in Journal
Boerhaave Syndrome—Narrative Review
Previous Article in Journal
Relationship Between Serum Apelin-13 and Apelin Receptor Levels and the Severity of Disease in Patients Diagnosed with Obstructive Sleep Apnea Syndrome
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Using Machine Learning to Create Prognostic Systems for Primary Prostate Cancer

1
F. Edward Hébert School of Medicine, Uniformed Services University of the Health Sciences (USUHS), Bethesda, MD 20814, USA
2
The Henry M. Jackson Foundation for the Advancement of Military Medicine Inc., Bethesda, MD 20817, USA
3
Commissioned Corps of the U.S. Public Health Service (USPHS), Rockville, MD 20852, USA
4
Center for Prostate Disease Research (CPDR), Murtha Cancer Center Research Program, Department of Surgery, Uniformed Services University of the Health Sciences, Bethesda, MD 20817, USA
5
Genitourinary Malignancies Branch, Center for Cancer Research, National Cancer Institute, Bethesda, MD 20814, USA
*
Authors to whom correspondence should be addressed.
Diagnostics 2025, 15(19), 2462; https://doi.org/10.3390/diagnostics15192462
Submission received: 24 July 2025 / Revised: 15 September 2025 / Accepted: 24 September 2025 / Published: 26 September 2025
(This article belongs to the Special Issue AI and Big Data in Medical Diagnostics)

Abstract

Background: Cancer staging, guided by anatomical and clinicopathologic factors, is essential for determining treatment strategies and patient prognosis. The current gold standard for prostate cancer is the American Joint Committee on Cancer (AJCC) Tumor, Lymph Node, and Metastasis (TNM) Staging System 9th Version (2024). This system incorporates five prognostic variables: tumor (T), spread to lymph nodes (N), metastasis (M), prostate-specific antigen (PSA) levels (P), and Grade Group/Gleason score (G). While effective, further refinement of prognostic systems may improve prediction of patient outcomes and support more individualized treatment. Methods: We applied the Ensemble Algorithm for Clustering Cancer Data (EACCD), an unsupervised machine learning approach. EACCD involves three steps: calculating initial dissimilarities, performing ensemble learning, and conducting hierarchical clustering. We first developed an EACCD model using the five AJCC variables (T, N, M, P, G). The model was then expanded to include two additional factors, age (A) and race (R). Prostate cancer patient data were obtained from the Surveillance, Epidemiology, and End Results (SEER) program from the National Cancer Institute. Results: The EACCD algorithm effectively stratified patients into distinct prognostic groups, each with well-separated survival curves. The five-variable model achieved a concordance index (C-index) of 0.8293 (95% CI: 0.8245–0.8341), while the seven-variable model, including age and race, improved performance to 0.8504 (95% CI: 0.8461–0.8547). Both outperformed the AJCC TNM system, which had a C-index of 0.7676 (95% CI: 0.7622–0.7731). Conclusions: EACCD provides a refined prognostic framework for primary localized prostate cancer, demonstrating superior accuracy over the AJCC staging system. With further validation in independent cohorts, EACCD could enhance risk stratification and support precision oncology.

1. Introduction

Prostate cancer is the most prevalent cancer among American men, with an estimated 299,010 new diagnoses and 35,250 deaths in 2024 [1]. Men of African ancestry are disproportionately affected, with higher rates of both diagnosis and mortality compared to men of European ancestry [1]. The widespread use of prostate-specific antigen (PSA) testing and advances in treatment has helped to reduce prostate cancer mortality by half since its peak in 1993 [2,3,4]. However, the potential for false positives has led to debates over PSA testing [5]. Meanwhile, the US Preventive Services Task Force (USPSTF) recommendations against screening in certain age groups have contributed to an increase in both localized and distant-stage diseases [6,7]. Accurately staging the disease is crucial for effective treatment and improved survival, as the 5-year survival rate drops significantly from 100% for localized cancer to 33% for distant stage patients [1,8].
The current standard for prostate cancer staging is the American Joint Committee on Cancer (AJCC) Tumor, Lymph Node, and Metastasis (TNM) Staging System [9,10]. This system relies on five key clinical–pathologic factors: the tumor size and extent (T), spread to lymph nodes (N), presence of metastasis (M), PSA levels (P), and Grade Group (G) [9,11]. The AJCC TNM system classifies prostate cancer into four stages [9]. Stage I is a non-palpable tumor that is undetectable by digital rectal exam (DRE) or imaging. In Stage II, the tumor is larger and detectable within the prostate by DRE or imaging. Stage III tumors have spread to nearby tissues such as seminal vesicles, while Stage IV indicates the tumor has metastasized to distant parts of the body, such as lymph nodes or bones. Localized prostate cancer (stages I, II, and IIIA) and regional prostate cancer metastasized to lymph nodes (stages IIIB and IVA) have an almost 99% five year survival rate, and distant prostate cancer (stage IVB) and unknown or unstaged prostate cancer have a five year relative survival rate of 34% and 92%, respectively [8]. Though staging is important to guide treatment modalities, the AJCC TNM System has significant limitations. It omits potentially important variables and has shown limited accuracy in predicting progression and recurrence. As demonstrated by the current AJCC standards, there is illogical staging of localized prostate cancer as stage IIIA along with the other localized prostate cancer stages I and II. The stage IIIA discrepancy is further highlighted as stage IIIA is treated as localized prostate cancer in contrast to stage IIIB, as regional prostate cancer. Other studies demonstrate the difficulty in distinguishing survival outcomes between stage III and stage IV patients [12]. Attempts to revise or improve the AJCC staging have been limited. For example, Sun et al. [12] proposed a modified staging system based on a nomogram that integrates T, N, and M stages, primary and secondary Gleason pattern scores, and PSA levels. However, nomograms rely on the proportional hazards assumption, which is often violated in real-world datasets, limiting their prognostic accuracy. In addition, there are no standardized guidelines for translating nomogram scores into discrete risk groups, reducing their practical utility.
To overcome these limitations, we applied the Ensemble Algorithm for Clustering Cancer Data (EACCD) [13,14] to develop a more accurate prognostic system. This machine learning algorithm has been successfully used to cluster survival data to identify highly accurate prognostic groups from survival data for other cancers, including breast [15,16], colorectal [17], and lung cancer [18], as well as in ovarian [19] and endometrial cancers [20], Hodgkin and Non-Hodgkin lymphoma [21], and melanomas [22]. By using EACCD on population-level SEER data, we aim to enhance prostate cancer staging and provide a more accurate tool for predicting patient outcomes. We first constructed models using the five standard AJCC variables and then extended them by incorporating age and race, two well-established prognostic factors not currently included in the AJCC system. We subsequently evaluated the performance of the EACCD models using the validation dataset.
This paper is organized as follows: Section 2 describes the methods, Section 3 presents the results, and Section 4 provides the conclusions.

2. Methods

2.1. Clinical–Pathologic Datasets

Prostate cancer data as identified by the International Classification of Disease for Oncology (Third Edition) [ICD-O-3] histology code (C61.9) were extracted from de-identified data on patient demographics from the National Institutes of Health (NIH) National Cancer Institute (NCI) Surveillance, Epidemiology, and End Results Program (SEER) database. A total of 254,390 out of the total 354,781 prostate patients diagnosed between 2010 and 2015 were retrieved from 17 datasets from the SEER 17 Registries Database 2023 version [23]. SEER cases before 2010 were excluded to reflect the same revision criteria as AJCC’s TNM Staging Editions used to diagnose prostate cancers on or after 1 January 2020, with the adoption of the PSA and Grade Group. Cases after 2015 were also excluded because 2015 was the last year that allowed a follow up year of at least five years. For the study data, survival time measured in months were included in the prostate-specific survival analysis.
Following the AJCC staging system, this study included five primary variables: Tumor (T), Lymph Node (N), Metastasis (M), PSA (P), and Grade Group (G). Additionally, two other variables—Age (A) and Race (R)—were included due to their established prognostic influence [24,25,26,27]. A detailed description of these seven variables is provided in Table 1. The original AJCC staging system is outlined in Table 2, compiled from information in the literature [9,10].
Other variables used in clinical prognosis such as androgen deprivation therapy, neoadjuvant androgen deprivation therapy, and genetic predisposition to prostate cancer from family lineage were not collected in the SEER database. Therefore, they could not be included in the EACCD calculations.
Cases diagnosed between 2010 and 2014 were used to create the training dataset, while cases diagnosed in 2015 served as the validation set. The inclusion and exclusion criteria outlined in Figure 1 and Figure 2 were applied to filter the training and validation datasets, respectively. The final training and validation datasets contain known values corresponding to the prognostic variables of interest. Any missing or unknown values were not imputed, and therefore excluded. Only patients with known Survival Time and Cause of Death were included because these variables are essential for reliable survival analysis; without them, it is not possible to calculate time-to-event outcomes or to distinguish prostate cancer–specific mortality. In the datasets, these exclusions represented a small fraction of the overall cohort (e.g., 464 patients, 1.6% of the total), and their removal ensured that survival endpoints were well-defined for all included cases. Any combination during the data processing step that had less than 25 patients was also excluded due to the low statistical power associated with lower representation. In this context, a “combination” refers to a subset of patients corresponding to specific levels of the selected factors, such as T4N1M1, which represents patients with T = T4, N = N1, and M = M1. A threshold of 25 was used to optimize the robustness of the statistical techniques, though any reasonable cutoff could be applied. Importantly, after applying these exclusion principles, the demographic and clinical–pathologic distribution of the final dataset remained consistent with the broader SEER prostate cancer population, suggesting no systematic bias was introduced. Thus, the final dataset structure was robust, reproducible, and appropriate for developing and validating prognostic models. The sample characteristics distribution of the variables in the training and validation datasets are outlined in Table 3 and Table 4, respectively. The EACCD algorithm was utilized on the training data to develop prognostic models, which were subsequently validated using the validation dataset.

2.2. EACCD

The Ensemble Algorithm for Clustering Cancer Data (EACCD) is a machine learning technique introduced by Chen et al. [13,16], designed to partition cancer survival data based on combinations of factors. Given a collection of combinations { C 1 , C 2 , . . . , C n } and nonnegative weights w 1 , w 2 , . . . , w n   with k = 1 n w k = 1 , the EACCD consists of the following three main steps (see Section Experiments in [14]):
  • Define the initial dissimilarity d i s 0 ( C i , C j ) for any pair C i and C j .
  • For each k with 1   k n , apply the two-phase Partitioning Around Medoids and the initial dissimilarities in Step 1 to partition combinations into k clusters. δ k ( i , j ) is defined as 1 if C i and C j are not assigned into the same cluster, otherwise δ k ( i , j ) was defined as 0. Then the learned dissimilarity is defined as follows: d i s ( C i , C j ) = k = 1 n w k δ k ( i , j ) .
  • Perform hierarchical clustering to cluster the combinations by using d i s ( C i , C j ) .
Each of these steps can employ different approaches. In the present study, the initial dissimilarity between C i and C j is defined as the survival difference between them, estimated using the effect size derived from the Gehan–Wilcoxon test statistic [14,28]. The ensemble learning process utilizes weights w k = 1 / n for k = 1,2 , . . . , n . For hierarchical clustering, the minimax linkage method [29] is selected.
The EACCD algorithm generates a dendrogram showcasing all the possible combinations in terms of survival. Dendrograms enable researchers and physicians to understand the rationale behind the grouping of patients. Combinations can be partitioned by horizontally cutting the dendrogram to the specific level of dissimilarity. Harrell’s C-index (also known as the concordance index) [30] is then computed from the available (prognostic) groups of combinations to highlight the statistical predictive accuracy. Generally, the curve for the C-index plotted against the number of groups increases for a relatively small number of groups before plateauing as more groups are generated. The optimal n* of prognostic groups is obtained from the “knee” point of the C-index curve.
We conclude this subsection with the following note: When we focus on patient features and survival times (numerical data), EACCD serves as an outcome-guided clustering method, that is, a form of supervised clustering where the supervision is provided by the outcome of interest. In contrast, when we focus on the survival curves of combinations (curves treated as geometric objects rather than raw numbers), EACCD functions as a fully unsupervised learning method, clustering these unlabeled curves based solely on their shapes.

2.3. Prognostic Systems

The proposed EACCD was trained on SEER 2010–2014 data to obtain n* prognostic groups. Survival curves for these groups were then plotted with Kaplan–Meier estimates to visually examine the survival differences among the different prognostic groups. The final prognostic system, named the EACCD prognostic system, consists of the dendrogram, group assignment, C-index, and survival curves.
As more variables are included, the number of possible combinations of values of variables increases, making it more difficult to visually examine the dendrogram. To address this, the dendrogram and corresponding group assignments can be summarized in tables (see the Results section below). In the EACCD system, risk stratification can be evaluated by visually examining the survival curves of the prognostic groups, while the accuracy of survival prediction is measured using the C-index. Overall, the EACCD system has been shown to provide effective patient risk stratification and high predictive accuracy.
The use of the EACCD system is straightforward. First, a patient’s values for the relevant variables are determined. Next, the patient is assigned to a prognostic group based on these values. Finally, the system’s survival curve associated with that group is used to help doctors and the patient understand the prognosis and guide treatment decisions.

2.4. Software

All statistical analyses as well as their respective visualizations were conducted in R (Version 4.2.2) using the following libraries: survival (Version 3.4-0), cluster (Version 2.1.4), protoclust (Version 1.6.4), factoextra (Version 1.0.7), and compareC (Version 1.3.2).

3. Results

In this section, we begin by presenting the AJCC 9th edition staging system and the EACCD system, both of which are based on five key variables: tumor, lymph nodes, metastasis, PSA, and Grade Group. Then an extended version of the EACCD system, which includes two additional variables race and age, was introduced. Finally, the validation results were shown to demonstrate reproducibility. For simplicity, the various AJCC prognostic stages are referred to as stages, while the corresponding EACCD outputs are referred to as groups.

3.1. Five Variable AJCC System

The original stages from the AJCC TNM System 9th Version consisting of four principal stages (I, II, III, and IV) from the five variables of Tumor (T), Metastasis (M), Lymph Nodes (N), PSA(P), and Grade Group (G) were included for comparisons to the groupings produced by the EACCD algorithm. Depending on the anatomic extent of the disease, PSA levels, and Grade Group, the AJCC TNM Prostate Staging System can be further divided into nine prognostic substages (I, IIA, IIB, IIC, IIIA, IIIB, IIIC, IVA, and IVB) as described in Table 2. Survival curves provide visual comparisons of the substages over time within the data. By applying the AJCC 9th Version Staging System to the training dataset, the survival curves shown in Figure 3 were generated. These curves reveal weak separations between stages. For instance, the curve for Stage I closely resembles that of Stage IIA, and the curve for Stage IIC is close to that of Stage IIIA. The C-index for the AJCC 9th Version was calculated to be 0.7676 (95% CI: 0.7622–0.7731).

3.2. Five Variable EACCD System

Using the same training data and the same five variables—T, N, M, P, and G—as in the AJCC system, the EACCD algorithm clustered 227 unique combinations of patients into 10 distinct groups in Figure 4. To aid readability, Supplementary Table S1 lists the specific combinations for each of the 10 groups in Figure 4. In addition, Supplementary Table S2 shows the number of events (deaths) in each of the 10 groups. The survival curves for these 10 groups were plotted in Figure 5. Note that the separation between any two adjacent curves is statistically significant (log-rank test p < 0.05). Comparing Figure 5 with Figure 3 shows that EACCD distinguishes and stratifies prostate cancer patients much more effectively than the AJCC system. Furthermore, the C-index for EACCD on the training set is 0.8293 (95% CI: 0.8245–0.8341) as demonstrated in Figure 6, which is higher than the AJCC C-index of 0.7676 (95% CI: 0.7622–0.7731).

3.3. Seven Variable EACCD System

The inclusion of two additional variables, age and race, in the EACCD algorithm results in the dendrogram shown in Figure 7. To improve readability, Supplementary Table S3 lists the specific combinations corresponding to each of the 13 groups in Figure 7. Supplementary Table S4 further provides the number of events (deaths) within each group. The survival curves for the 13 resulting groups are shown in Figure 8. These curves are clearly separated (log-rank test p < 0.05 for any two adjacent curve comparisons), similar to those of the EACCD five-variable system in Figure 5. The C-index for this extended model is 0.8504 (95% CI: 0.8461–0.8547) as shown in Figure 9, which is higher than the AJCC C-index of 0.7676 (95% CI: 0.7622–0.7731) and the C-index of 0.8293 (95% CI: 0.8245–0.8341) from the EACCD five-variable model.

3.4. Validation

In Section 3.2 and Section 3.3, we developed the EACCD systems using five and seven variables with the training data. Both the five- and seven-variable EACCD systems outperformed the AJCC 9th Edition, providing clearer separation of survival groups and higher predictive accuracy. In this subsection, we present the results of evaluating these systems using the validation dataset. The validation process was as follows: we first applied the assignment rules of the systems to the validation dataset to group (“stage”) the patients. Next, we computed the C-indices and plotted survival curves based on these groups. Finally, we compared the C-indices and survival curves from the validation dataset with those from the training dataset in Section 3.2 and Section 3.3. Consistency of C-indices and survival curves between training and validation data indicates the reliability of the EACCD systems.
Table 5 compares the C-indices between the training and validation datasets in the five variable system and seven variable system. The five-variable EACCD system has a C-index of 0.8293 (95% CI: 0.8245–0.8341) in the training data and 0.8437 (95% CI: 0.8308–0.8566) in the validation data, while the seven-variable EACCD system has a C-index of 0.8504 (95% CI: 0.8461–0.8547) in the training data and 0.8585 (95% CI: 0.8468–0.8703) in the validation data. This indicates that the survival prediction performance of each EACCD system is consistent across both the training and validation datasets.
Figure 10 shows the survival curves of the groups in the validation dataset. For the five-variable system, 10 survival curves corresponding to 10 groups are obtained. The separation between adjacent curves is significant (log-rank test p < 0.05) except for group 5 vs. group 4, group 7 vs. group 6, and group 8 vs. group 7. In these cases, the hazard ratios still point in the expected direction, providing objective evidence of group separation. Thus, applying the EACCD TNM system to the validation data yields 10 groups with clearly separated survival curves. Similarly, applying the EACCD TNMA system to the validation data results in 13 groups with separated survival curves. Here, the separation is significant for all adjacent pairs except group 10 vs. group 9, group 11 vs. group 10, and group 12 vs. group 11. Again, the hazard ratios for these comparisons remain consistent with the intended risk ordering. Together, these results indicate that the risk stratification performance of both systems is consistent across the training and validation datasets.
The above validation results suggest that EACCD performance is reproducible on future data. This further supports the robustness of the system for clinical application.

4. Discussion

There appear to be very few studies in the literature focused on developing methods to improve prostate cancer staging. A recent study by Sun et al. [12] modified the AJCC staging system using nomogram scores derived from multivariate Cox survival modeling. Sun et al. used a filtered dataset of 16,803 patients and achieved a C-index of 0.789 (95% CI: 0.777–0.801) compared to the calculated C-index 0.762 (95% CI: 0.748–0.776) on their dataset using the gold standard AJCC model [12]. In contrast, our EACCD model was trained on a larger and more recent dataset of 161,212 patients and produced C-indices of 0.8293 (95% CI: 0.8245–0.8341) and 0.8504 (95% CI: 0.8461–0.8547) for the five- and seven-factor prognostic systems, respectively.
The strength of unsupervised clustering algorithms lies in their ability to autonomously identify novel subgroups within complex and high-dimensional datasets. Traditional supervised methods, while effective for specific tasks like predicting biochemical recurrence, as seen in studies by Tan et al. [31] and Ekşi et al. [32], are ultimately designed to reinforce existing knowledge. Semwal et al. trained multiple supervised models to predict pathological stages using an XGBoost model that outperformed existing clinical methods [33]. Martelin et al. also used a gradient boost model that included PSA and personal variables to improve the model’s accuracy [34]. Nayan et al. employed an XGB algorithm to predict 5-year survival, highlighting the need to consider race in ML models to address healthcare disparities [35]. Finally, Fonseca et al. developed an XGB-based tool to predict whether there was enough circulating tumor DNA for informative genotyping in metastatic cases, further demonstrating the potential of machine learning to optimize biomarker use and refine risk stratification [36].
Our EACCD approach, which utilizes unsupervised clustering at its core, allowed us to identify new patient subgroups that were not evident in the current staging system. This is a crucial step toward creating more precise and personalized prognostic systems. Unsupervised learning approaches, with their ability to uncover hidden structures and inherent data patterns, provide a powerful tool for deepening our understanding of prostate cancer heterogeneity. They present an improvement beyond traditional approaches that rely on clinical pathologic features such as Gleason score, which often fail to capture the full picture of disease heterogeneity [37]. These approaches have been applied on diverse data types, including genomic and proteomic profiles, to discover new molecular subtypes with distinct clinical outcomes [38]. Increasingly, more studies employ integrated multi-modal systems that combine data from multiple sources such as digital pathology and clinical records to create highly accurate patient specific models [39]. These unsupervised clustering approaches provide more accurate and reliable prognostic information that is independent of existing risk classifications, therefore aiding in identifying at-risk patients who might otherwise be miscategorized. They will also improve as more data becomes available, supporting their value as a risk stratification tool to optimize for follow-up and reduce over-treatment.
The development of the EACCD prognostic system marks a promising step toward advancing personalized medicine in prostate cancer. Moving beyond the traditional and subjective committee-based decision-making, our EACCD method provides an algorithmic and data-driven framework. The EACCD system can be continually updated with new prognostic factors as they are discovered, such as circulating tumor DNA (ctDNA) and other clinical variables like lactate dehydrogenase [36]. This adaptability ensures its long-term relevance and accuracy. The potential for developing a clinical decision support tool that leverages EACCD to provide real-time prognostic groups for patients could significantly enhance clinical decision-making, optimize treatment strategies, and ultimately improve health outcomes.

Supplementary Materials

The following supporting information can be downloaded at https://www.mdpi.com/article/10.3390/diagnostics15192462/s1, Table S1: Groups and their combinations in the five variable EACCD system. Table S2: Number of patients and events in each of the 10 groups in the five variable EACCD system. Table S3: Groups and their combinations in the seven variable EACCD system. Table S4: Number of patients and events in each of the 13 groups in the seven variable EACCD system.

Author Contributions

K.G., S.-H.T., and D.C. designed the study. K.G. modified the EACCD code for the data analysis. K.G. and A.G. collected, wrangled, and analyzed the data. K.G., S.-H.T., and D.C. visualized the data into tables and figures. K.G., A.G., A.E.A., A.J.W., S.-H.T. and D.C. prepared the manuscript. All authors have read and agreed to the published version of the manuscript.

Funding

This work was supported by funding (HU00012320028) from the Murtha Cancer Center Program (MCCRP) awarded to KG and DC, and funding (HU00012520001) from the MCCRP, Center for Prostate Disease Research (CPDR) to SHT.

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

The data presented in this study are available in the SEER database at https://seer.cancer.gov/data (accessed on 3 October 2023). A similarly structured source code written by Huan Wang used in this study is available in: https://github.com/hwang0113/Prognostic-System-for-Lymphoma (accessed on 4 June 2023). The source code was programmed and modified in R (Version 4.2.2) using the survival (Version 3.4-0), cluster (Version 2.1.4), protoclust (Version 1.6.4), factoextra (Version 1.0.7), and compareC (Version 1.3.2) libraries to generate the results of the EACCD studies in this manuscript.

Acknowledgments

All the authors would like to graciously acknowledge Huan Wang, who wrote the original EACCD code in the public domain that KG modified for data analysis.

Conflicts of Interest

The authors declare that there are no conflicts of interest.

Disclaimer

The contents of this publication are the sole responsibility of the author(s) and do not necessarily reflect the views, opinions or policies of the Uniformed Services University of the Health Sciences, the Department of Defense (DoD), or the Departments of the Army, Navy, and Air Force. Mention of trade names, commercial products, or organizations does not imply endorsement by the US Government.

References

  1. Siegel, R.L.; Giaquinto, A.N.; Jemal, A. Cancer statistics, 2024. CA Cancer J. Clin. 2024, 74, 12–49. [Google Scholar] [CrossRef] [PubMed]
  2. Etzioni, R.; Tsodikov, A.; Mariotto, A.; Szabo, A.; Falcon, S.; Wegelin, J.; DiTommaso, D.; Karnofski, K.; Gulati, R.; Penson, D.F.; et al. Quantifying the role of PSA screening in the US prostate cancer mortality decline. Cancer Causes Control 2008, 19, 175–181. [Google Scholar] [CrossRef] [PubMed]
  3. Tsodikov, A.; Gulati, R.; Etzioni, R. Reconciling the Effects of Screening on Prostate Cancer Mortality in the ERSPC and PLCO Trials. Ann. Intern. Med. 2018, 168, 608–609. [Google Scholar] [CrossRef] [PubMed]
  4. Matsukawa, A.; Yanagisawa, T.; Bekku, K.; Kardoust Parizi, M.; Laukhtina, E.; Klemm, J.; Chiujdea, S.; Mori, K.; Kimura, S.; Fazekas, T.; et al. Comparing the Performance of Digital Rectal Examination and Prostate-specific Antigen as a Screening Test for Prostate Cancer: A Systematic Review and Meta-analysis. Eur. Urol. Oncol. 2024, 7, 697–704. [Google Scholar] [CrossRef]
  5. Borregales, L.D.; DeMeo, G.; Gu, X.; Cheng, E.; Dudley, V.; Schaeffer, E.M.; Nagar, H.; Carlsson, S.; Vickers, A.; Hu, J.C. Grade Migration of Prostate Cancer in the United States During the Last Decade. J. Natl. Cancer Inst. 2022, 114, 1012–1019. [Google Scholar] [CrossRef]
  6. Jemal, A.; Fedewa, S.A.; Ma, J.; Siegel, R.; Lin, C.C.; Brawley, O.; Ward, E.M. Prostate Cancer Incidence and PSA Testing Patterns in Relation to USPSTF Screening Recommendations. JAMA 2015, 314, 2054–2061. [Google Scholar] [CrossRef]
  7. Moyer, V.A.; U.S. Preventive Services Task Force. Screening for prostate cancer: U.S. Preventive Services Task Force recommendation statement. Ann. Intern. Med. 2012, 157, 120–134. [Google Scholar] [CrossRef]
  8. Siegel, D.A.; O’Neil, M.E.; Richards, T.B.; Dowling, N.F.; Weir, H.K. Prostate Cancer Incidence and Survival, by Stage and Race/Ethnicity—United States, 2001–2017. MMWR Morb. Mortal. Wkly. Rep. 2020, 69, 1473–1480. [Google Scholar] [CrossRef]
  9. Amin, M.B.; Greene, F.L.; Edge, S.B.; Compton, C.C.; Gershenwald, J.E.; Brookland, R.K.; Meyer, L.; Gress, D.M.; Byrd, D.R.; Winchester, D.P. The Eighth Edition AJCC Cancer Staging Manual: Continuing to build a bridge from a population-based to a more “personalized” approach to cancer staging. CA Cancer J. Clin. 2017, 67, 93–99. [Google Scholar] [CrossRef]
  10. Edge, S.B.; Compton, C.C. The American Joint Committee on Cancer: The 7th Edition of the AJCC Cancer Staging Manual and the Future of TNM. Ann. Surg. Oncol. 2010, 17, 1471–1474. [Google Scholar] [CrossRef]
  11. Epstein, J.I.; Zelefsky, M.J.; Sjoberg, D.D.; Nelson, J.B.; Egevad, L.; Magi-Galluzzi, C.; Vickers, A.J.; Parwani, A.V.; Reuter, V.E.; Fine, S.W.; et al. A Contemporary Prostate Cancer Grading System: A Validated Alternative to the Gleason Score. Eur. Urol. 2016, 69, 428–435. [Google Scholar] [CrossRef] [PubMed]
  12. Sun, C.; Yang, D.; Zhu, J.; Zhou, Y.; Xiang, C.; Wu, S. Modified the 8th AJCC staging system for patients with advanced prostate cancer: A study based on SEER database. BMC Urol. 2022, 22, 185. [Google Scholar] [CrossRef] [PubMed]
  13. Chen, D.; Xing, K.; Henson, D.; Sheng, L.; Schwartz, A.M.; Cheng, X. Developing prognostic systems of cancer patients by ensemble clustering. J. Biomed. Biotechnol. 2009, 2009, 632786. [Google Scholar] [CrossRef] [PubMed]
  14. Wang, H.; Chen, D.; Pan, Q.; Hueman, M.T. Using Weighted Differences in Hazards as Effect Sizes for Survival Data. J. Stat. Theory Pract. 2022, 16, 12. [Google Scholar] [CrossRef]
  15. Hueman, M.T.; Wang, H.; Yang, C.Q.; Sheng, L.; Henson, D.E.; Schwartz, A.M.; Chen, D. Creating prognostic systems for cancer patients: A demonstration using breast cancer. Cancer Med. 2018, 7, 3611–3621. [Google Scholar] [CrossRef]
  16. Qi, R.; Wu, D.; Sheng, L.; Henson, D.; Schwartz, A.; Xu, E.; Xing, K.; Chen, D. On an ensemble algorithm for clustering cancer patient data. BMC Syst. Biol. 2013, 7 (Suppl. 4), S9. [Google Scholar] [CrossRef]
  17. Henson, D.E.; Hueman, M.T.; Chen, D.; Patel, J.A.; Wang, H.; Schwartz, A.M. The anatomy of the TNM for colon cancer. J. Gastrointest. Oncol. 2017, 8, 12–19. [Google Scholar] [CrossRef]
  18. Hueman, M.; Wang, H.; Liu, Z.; Henson, D.; Nguyen, C.; Park, D.; Sheng, L.; Chen, D. Expanding TNM for lung cancer through machine learning. Thorac. Cancer 2021, 12, 1423–1430. [Google Scholar] [CrossRef]
  19. Grimley, P.M.; Liu, Z.; Darcy, K.M.; Hueman, M.T.; Wang, H.; Sheng, L.; Henson, D.E.; Chen, D. A prognostic system for epithelial ovarian carcinomas using machine learning. Acta Obstet. Gynecol. Scand. 2021, 100, 1511–1519. [Google Scholar] [CrossRef]
  20. Praiss, A.M.; Huang, Y.; St Clair, C.M.; Tergas, A.I.; Melamed, A.; Khoury-Collado, F.; Hou, J.Y.; Hu, J.; Hur, C.; Hershman, D.L.; et al. Using machine learning to create prognostic systems for endometrial cancer. Gynecol. Oncol. 2020, 159, 744–750. [Google Scholar] [CrossRef]
  21. Wang, H.; Liu, Z.; Yang, J.; Sheng, L.; Chen, D. Using Machine Learning to Expand the Ann Arbor Staging System for Hodgkin and Non-Hodgkin Lymphoma. Bio. Med. Informatics 2023, 3, 514–525. [Google Scholar] [CrossRef]
  22. Yang, C.Q.; Wang, H.; Liu, Z.; Hueman, M.T.; Bhaskaran, A.; Henson, D.E.; Sheng, L.; Chen, D. Integrating additional factors into the TNM staging for cutaneous melanoma by machine learning. PLoS ONE 2021, 16, e0257949. [Google Scholar] [CrossRef] [PubMed]
  23. Surveillance, Epidemiology, and End Results (SEER) Program. SEER*Stat Database: Incidence—SEER Research Data, 17 Registries, November 2023 Submission (2000–2021). Available online: https://seer.cancer.gov/data-software/documentation/seerstat/nov2023/ (accessed on 6 February 2025).
  24. Dess, R.T.; Hartman, H.E.; Mahal, B.A.; Soni, P.D.; Jackson, W.C.; Cooperberg, M.R.; Amling, C.L.; Aronson, W.J.; Kane, C.J.; Terris, M.K.; et al. Association of Black Race with Prostate Cancer-Specific and Other-Cause Mortality. JAMA Oncol. 2019, 5, 975–983. [Google Scholar] [CrossRef] [PubMed]
  25. Banez, L.L.; Terris, M.K.; Aronson, W.J.; Presti, J.C., Jr.; Kane, C.J.; Amling, C.L.; Freedland, S.J. Race and time from diagnosis to radical prostatectomy: Does equal access mean equal timely access to the operating room?—Results from the SEARCH database. Cancer Epidemiol. Biomark. Prev. 2009, 18, 1208–1212. [Google Scholar] [CrossRef] [PubMed]
  26. Wenzel, M.; Deuker, M.; Stolzenbach, F.; Nocera, L.; Colla Ruvolo, C.; Tian, Z.; Shariat, S.F.; Saad, F.; Briganti, A.; Kluth, L.A.; et al. The effect of race/ethnicity on histological subtype distribution, stage at presentation and cancer specific survival in urethral cancer. Urol. Oncol. 2021, 39, 369.e9–369.e17. [Google Scholar] [CrossRef] [PubMed]
  27. Lee, J.; Song, J.; Jung, G.; Song, S.H.; Hong, S.K. Prognosis after radical prostatectomy in men older than 75 years: Long-term results from a single tertiary center. Prostate Int. 2024, 12, 15–19. [Google Scholar] [CrossRef]
  28. Huan, W.; Li, S.; Dechang, C. Perspective Chapter: Using Effect Sizes to Study the Survival Difference between Two Groups. In Recent Advances in Biostatistics; Kumar, B.S., Ed.; IntechOpen: Rijeka, Croatia, 2023; p. 6. [Google Scholar]
  29. Bien, J.; Tibshirani, R. Hierarchical Clustering with Prototypes via Minimax Linkage. J. Am. Stat. Assoc. 2011, 106, 1075–1084. [Google Scholar] [CrossRef]
  30. Harrell, F.E., Jr.; Lee, K.L.; Mark, D.B. Multivariable prognostic models: Issues in developing models, evaluating assumptions and adequacy, and measuring and reducing errors. Stat. Med. 1996, 15, 361–387. [Google Scholar] [CrossRef]
  31. Tan, Y.G.; Fang, A.H.S.; Lim, J.K.S.; Khalid, F.; Chen, K.; Ho, H.S.S.; Yuen, J.S.P.; Huang, H.H.; Tay, K.J. Incorporating artificial intelligence in urology: Supervised machine learning algorithms demonstrate comparative advantage over nomograms in predicting biochemical recurrence after prostatectomy. Prostate 2022, 82, 298–305. [Google Scholar] [CrossRef]
  32. Ekşi, M.; Evren, İ.; Akkaş, F.; Arıkan, Y.; Özdemir, O.; Özlü, D.N.; Ayten, A.; Sahin, S.; Tuğcu, V.; Taşçı, A.I. Machine learning algorithms can more efficiently predict biochemical recurrence after robot-assisted radical prostatectomy. Prostate 2021, 81, 913–920. [Google Scholar] [CrossRef]
  33. Semwal, H.; Ladbury, C.; Sabbagh, A.; Mohamad, O.; Tilki, D.; Amini, A.; Wong, J.; Li, Y.R.; Glaser, S.; Yuh, B.; et al. Machine learning and explainable artificial intelligence to predict pathologic stage in men with localized prostate cancer. Prostate 2025, 85, 3–12. [Google Scholar] [CrossRef] [PubMed]
  34. Martelin, N.; De Witt, B.; Chen, B.; Eschwège, P. Development and validation of an imageless machine-learning algorithm for the initial screening of prostate cancer. Prostate 2024, 84, 842–849. [Google Scholar] [CrossRef] [PubMed]
  35. Nayan, M.; Salari, K.; Bozzo, A.; Ganglberger, W.; Carvalho, F.; Feldman, A.S.; Trinh, Q.D. Predicting survival after radical prostatectomy: Variation of machine learning performance by race. Prostate 2021, 81, 1355–1364. [Google Scholar] [CrossRef]
  36. Fonseca, N.M.; Maurice-Dror, C.; Herberts, C.; Tu, W.; Fan, W.; Murtha, A.J.; Kollmannsberger, C.; Kwan, E.M.; Parekh, K.; Schönlau, E.; et al. Prediction of plasma ctDNA fraction and prognostic implications of liquid biopsy in advanced prostate cancer. Nat. Commun. 2024, 15, 1828. [Google Scholar] [CrossRef]
  37. Markert, E.K.; Mizuno, H.; Vazquez, A.; Levine, A.J. Molecular classification of prostate cancer using curated expression signatures. Proc. Natl. Acad. Sci. USA 2011, 108, 21276–21281. [Google Scholar] [CrossRef]
  38. Sun, R.; Jun, A.; Yu, H.; Wang, Y.; He, M.; Tan, L.; Cheng, H.; Zhang, J.; Wang, Y.; Sun, X.; et al. Proteomic landscape profiling of primary prostate cancer reveals a 16-protein panel for prognosis prediction. Cell Rep. Med. 2024, 5, 101679. [Google Scholar] [CrossRef]
  39. Hu, J.; Miao, Q.; Ren, J.; Su, H.; Zhang, X.; Bi, J.; Zhang, G. An online clustering algorithm predicting model for prostate cancer based on PHI-related variables and PI-RADS in different PSA populations. Cancer Cell Int. 2025, 25, 44. [Google Scholar] [CrossRef] [PubMed]
Figure 1. Criteria for the training dataset. Out of the 354,781 prostate cancer patients diagnosed and recorded in 2010–2021 from the SEER database, the 254,390 cases diagnosed from 2010–2014 inclusive were wrangled to form the initial dataset. Out of the 254,390 cases from the years 2010–2014, there are 164,236 cases with known survival time and known cause of death that form 741 known combinations of levels of (T)umor, Lymph (N)ode, (M)etastasis, (P)SA, (G)rade Group, (A)ge, and (R)ace. The final training set consists of 227 combinations each containing at least 25 patients.
Figure 1. Criteria for the training dataset. Out of the 354,781 prostate cancer patients diagnosed and recorded in 2010–2021 from the SEER database, the 254,390 cases diagnosed from 2010–2014 inclusive were wrangled to form the initial dataset. Out of the 254,390 cases from the years 2010–2014, there are 164,236 cases with known survival time and known cause of death that form 741 known combinations of levels of (T)umor, Lymph (N)ode, (M)etastasis, (P)SA, (G)rade Group, (A)ge, and (R)ace. The final training set consists of 227 combinations each containing at least 25 patients.
Diagnostics 15 02462 g001
Figure 2. Criteria for the validation dataset. Out of the 48,624 prostate cancer patients diagnosed and recorded in 2015 from the NIH SEER database, there are 31,295 cases with known survival time and known cause of death that form 544 known combinations of levels of (T)umor, Lymph (N)ode, (M)etastasis, (P)SA, (G)rade Group, (A)ge, and (R)ace. The validation set consists of 40 combinations each containing at least 25 patients.
Figure 2. Criteria for the validation dataset. Out of the 48,624 prostate cancer patients diagnosed and recorded in 2015 from the NIH SEER database, there are 31,295 cases with known survival time and known cause of death that form 544 known combinations of levels of (T)umor, Lymph (N)ode, (M)etastasis, (P)SA, (G)rade Group, (A)ge, and (R)ace. The validation set consists of 40 combinations each containing at least 25 patients.
Diagnostics 15 02462 g002
Figure 3. Survival curves plot using the training data and the AJCC 9th version prostate cancer staging system.
Figure 3. Survival curves plot using the training data and the AJCC 9th version prostate cancer staging system.
Diagnostics 15 02462 g003
Figure 4. Dendrogram generated by the EACCD using five variables: T, N, M, P, and G.
Figure 4. Dendrogram generated by the EACCD using five variables: T, N, M, P, and G.
Diagnostics 15 02462 g004
Figure 5. Survival curves of the 10 groups as extrapolated from Figure 4.
Figure 5. Survival curves of the 10 groups as extrapolated from Figure 4.
Diagnostics 15 02462 g005
Figure 6. The C-index plot generated using five variables: T, N, M, P, and G. Here n* (10) groups correspond to a C-index of 0.8293.
Figure 6. The C-index plot generated using five variables: T, N, M, P, and G. Here n* (10) groups correspond to a C-index of 0.8293.
Diagnostics 15 02462 g006
Figure 7. Dendrogram generated by the EACCD using seven variables: T, N, M, P, G, A, and R.
Figure 7. Dendrogram generated by the EACCD using seven variables: T, N, M, P, G, A, and R.
Diagnostics 15 02462 g007
Figure 8. Survival Curves of the 13 groups as extrapolated from Figure 7.
Figure 8. Survival Curves of the 13 groups as extrapolated from Figure 7.
Diagnostics 15 02462 g008
Figure 9. The C-index plot generated using seven variables: T, N, M, P, G, A, and R. Here n* (13) groups correspond to a C-index of 0.8504.
Figure 9. The C-index plot generated using seven variables: T, N, M, P, G, A, and R. Here n* (13) groups correspond to a C-index of 0.8504.
Diagnostics 15 02462 g009
Figure 10. Survival curves for the validation dataset. (A) Five-variable system; (B) Seven-variable system.
Figure 10. Survival curves for the validation dataset. (A) Five-variable system; (B) Seven-variable system.
Diagnostics 15 02462 g010
Table 1. Seven prognostic variables as defined by their usage in EACCD and AJCC, with information compiled from the American College of Surgeons (ACS) and the American Joint Committee on Cancer (AJCC) [9,10].
Table 1. Seven prognostic variables as defined by their usage in EACCD and AJCC, with information compiled from the American College of Surgeons (ACS) and the American Joint Committee on Cancer (AJCC) [9,10].
VariablesLevelsDefinition/Description
Tumor stage (T)TxTumor was not evaluated
T0No evidence of the primary tumor
T1Tumor was not detected during Digital Rectal Exam (DRE) and not seen in imaging
T2Tumor is detected in DRE but locally confined in the prostate
T2a: Tumor is half or less than one side of the prostate lobe
T2b: Tumor is in more than ½ of the prostate lobe but not in the other lobe
T2c: Tumor is present in both lobes
T3Tumor extends outside of the prostate
T3a: Tumor extends outside of the prostate on one or both sides
T3b: Tumor has spread to the seminal vesicles
T4Tumor has spread to nearby tissues outside of the prostate and seminal vesicles.
Metastasis to nearby or regional lymph nodes (N)NxNearby lymph nodes are not evaluated
N0No cancer cells are found in the lymph nodes
N1Cancer cells are found in the lymph nodes
Distant metastasis (M)M0Cancer has not spread past the prostate
M1Cancer has spread past the prostate
M1a: Cancer has spread to the distant lymph nodes
M1b: Cancer has spread to bones
M1c: Cancer has spread to other organs and sites with or without bone disease
PSA
(ng/mL)
P1P < 10
P210 ≤ P < 20
P3P ≥ 20
Grade Group (G)
/Gleason Score (GS)
G1: GS ≤ 6Only individual discrete well-formed glands
G2: GS 7 (3 + 4)Predominantly well-formed glands with lesser components of poorly formed, fused cribriform glands
G3: GS 7 (4 + 3)Predominantly poorly formed, fused, cribriform glands with lesser components of well-formed glands
G4: GS 8Only poorly formed, fused, cribriform glands or predominantly well-formed glands with a lesser component lacking or predominately lacking glands with a lesser component of well-formed glands
G5: GS 9–10Lacks gland formation (or with necrosis) with or without poorly formed/fused/cribriform gland
Age (years)A0<70
A1≥70
RaceR1White, Caucasian, or European Ancestry
R2Black or African Ancestry
R3Other
Table 2. The AJCC TNM staging of prostate cancer. The AJCC 8th Edition was published in 2017, and the revision in the 2024 manual updated the naming of the 8th Edition to the 9th Version. The AJCC 9th Version (2024) has no changes to the prognostic prostate staging system compared with the 8th Edition. Data compiled from information in [9,10].
Table 2. The AJCC TNM staging of prostate cancer. The AJCC 8th Edition was published in 2017, and the revision in the 2024 manual updated the naming of the 8th Edition to the 9th Version. The AJCC 9th Version (2024) has no changes to the prognostic prostate staging system compared with the 8th Edition. Data compiled from information in [9,10].
AJCC TNM
Prostate Cancer Stages
Stage SubgroupsGrade GroupTumorLymph NodeMetastasisPSA
Stage I (Localized)
Cancer is small and only in the prostate
I1cT1a–c
cT2a
pT2
N0M0<10
Stage II (Localized)
Cancer is larger and may be in both prostate lobes but still confined in the prostate
IIA1cT1–ac
cT2a
pT2
N0M010–20
IIA1cT2b–cN0M010–20
IIB2T1–2N0M0<20
IIC3
4
T1–2N0M0<20
Stage III
(Locally Advanced)
Cancer has spread from the prostate to nearby lymph nodes or seminal vesicles
IIIA1–4T1–2N0M0≥20
IIIB1–4T3–4N0M0Any
IIIC5Any TN0M0Any
Stage IV
(Metastatic/Advanced)
Cancer has spread to other parts of the body such as bones, liver, or lungs.
IVAAnyAny TN1M0Any
IVBAnyAny TAny NM1Any
Table 3. Sample characteristics of variable distributions in the Training Dataset.
Table 3. Sample characteristics of variable distributions in the Training Dataset.
Variables in the Training Dataset (N = 161, 212)n
Tumor stage (T)
T180,384 (50%)
T2a12,897 (8.0%)
T2bc46,316 (29%)
T320,599 (13%)
T41016 (0.6%)
Lymph node metastasis (N)
N0157,398 (98%)
N13814 (2.4%)
Metastasis (M)
M0157,680 (98%)
M13532 (2.2%)
Prostate-Specific Antigen (P)
P1926 (0.6%)
P22311 (1.4%)
P3157,975 (98%)
Grade Group (G)
G170,901 (44%)
G242,718 (26%)
G319,395 (12%)
G415,823 (9.8%)
G512,375 (7.7%)
Age (A)
A0115,078 (71%)
A146,134 (29%)
Race (R)
R1127,971 (79%)
R224,701 (15%)
R38540 (5.3%)
Table 4. Sample characteristics of variable distributions in the Validation Dataset.
Table 4. Sample characteristics of variable distributions in the Validation Dataset.
Variables in the Validation Dataset (N = 29, 161)n
Tumor stage (T)
T115,360 (53%)
T2a1801 (6.2%)
T2bc7229 (25%)
T34614 (16%)
T4157 (0.5%)
Lymph node metastasis (N)
N028,293 (97%)
N1868 (3.0%)
Metastasis (M)
M028,636 (98%)
M1525 (1.8%)
Prostate-Specific Antigen (P)
P1106 (0.4%)
P2262 (0.9%)
P328,793 (99%)
Grade Group (G)
G110,443 (36%)
G28357 (29%)
G34253 (15%)
G43252 (11%)
G52856 (9.8%)
Age (A)
A020,725 (71%)
A18436 (29%)
Race (R)
R123,308 (80%)
R24470 (15%)
R31383 (4.7%)
Table 5. Comparisons of C-indices by number of variables, data type, and method.
Table 5. Comparisons of C-indices by number of variables, data type, and method.
Number of VariablesData TypeMethodC-Index95%CI
5TrainingAJCC0.76760.7622–0.7731
5TrainingEACCD0.82930.8245–0.8341
5ValidationEACCD0.84370.8308–0.8566
7TrainingEACCD0.85040.8461–0.8547
7ValidationEACCD0.85850.8468–0.8703
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Guan, K.; Guan, A.; Ahmed, A.E.; Waters, A.J.; Tan, S.-H.; Chen, D. Using Machine Learning to Create Prognostic Systems for Primary Prostate Cancer. Diagnostics 2025, 15, 2462. https://doi.org/10.3390/diagnostics15192462

AMA Style

Guan K, Guan A, Ahmed AE, Waters AJ, Tan S-H, Chen D. Using Machine Learning to Create Prognostic Systems for Primary Prostate Cancer. Diagnostics. 2025; 15(19):2462. https://doi.org/10.3390/diagnostics15192462

Chicago/Turabian Style

Guan, Kevin, Andy Guan, Anwar E. Ahmed, Andrew J. Waters, Shyh-Han Tan, and Dechang Chen. 2025. "Using Machine Learning to Create Prognostic Systems for Primary Prostate Cancer" Diagnostics 15, no. 19: 2462. https://doi.org/10.3390/diagnostics15192462

APA Style

Guan, K., Guan, A., Ahmed, A. E., Waters, A. J., Tan, S.-H., & Chen, D. (2025). Using Machine Learning to Create Prognostic Systems for Primary Prostate Cancer. Diagnostics, 15(19), 2462. https://doi.org/10.3390/diagnostics15192462

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop