Development and internal validation of an XGBoost-based prognostic prediction model and risk stratification system for overall survival in pediatric neuroblastoma: a SEER-based study
Original Article

Development and internal validation of an XGBoost-based prognostic prediction model and risk stratification system for overall survival in pediatric neuroblastoma: a SEER-based study

Huan Li1#, Jianfeng Luo2#, Jun Yang1

1Department of General Surgery, Wuhan Children’s Hospital (Wuhan Maternal and Child Healthcare Hospital), Tongji Medical College, Huazhong University of Science & Technology, Wuhan, China; 2Graduate School of Peking Union Medical College, Chinese Academy of Medical Sciences, Beijing, China

Contributions: (I) Conception and design: H Li, J Luo; (II) Administrative support: J Yang; (III) Provision of study materials or patients: H Li, J Luo; (IV) Collection and assembly of data: H Li, J Luo; (V) Data analysis and interpretation: H Li, J Luo; (VI) Manuscript writing: All authors; (VII) Final approval of manuscript: All authors.

#These authors contributed equally to this work.

Correspondence to: Jun Yang, MD. Department of General Surgery, Wuhan Children’s Hospital (Wuhan Maternal and Child Healthcare Hospital), Tongji Medical College, Huazhong University of Science & Technology, 100 Hongkong Road, Wuhan 430016, China. Email: 1823759661@qq.com.

Background: Neuroblastoma (NB) is a highly heterogeneous pediatric malignancy with markedly variable clinical outcomes. Although established risk classification systems provide important prognostic information, individualized survival prediction remains challenging. This study aimed to develop and internally validate machine learning-based prognostic models for overall survival (OS) in pediatric NB patients and to establish a risk stratification system based on the best-performing model.

Methods: Data from 2,427 pediatric patients diagnosed with NB between 2000 and 2021 were extracted from the Surveillance, Epidemiology, and End Results (SEER) database. Eligible patients were randomly assigned to a training cohort (80%) and an internal validation cohort (20%). Least absolute shrinkage and selection operator (LASSO) regression identified six prognostic factors, including age, primary tumor site, tumor size, distant metastasis, SEER stage, and chemotherapy status. Six machine learning algorithms, including extreme gradient boosting (XGBoost), logistic regression, support vector machine, random forest, k-nearest neighbors, and decision tree, were developed to predict 1-, 3-, and 5-year OS. Model performance was evaluated using the area under the receiver operating characteristic curve (AUC), concordance index (C-index), and Brier score.

Results: Among the six models evaluated, XGBoost demonstrated the best predictive performance in both the training and internal validation cohorts. In the validation cohort, the AUCs for predicting 1-, 3-, and 5-year OS were 0.767, 0.796, and 0.807, respectively. The C-index of the XGBoost model was 0.784 [95% confidence interval (CI): 0.746–0.821]. Metastatic status was identified as the most influential predictor. Based on XGBoost-derived risk scores, patients were successfully stratified into low-, intermediate-, and high-risk groups with significantly different survival outcomes (all P<0.05).

Conclusions: An XGBoost-based prognostic model demonstrated favorable performance for individualized survival prediction in pediatric NB and enabled effective risk stratification using routinely available clinical variables. This model may complement existing prognostic assessment approaches; however, further external validation is required before clinical implementation.

Keywords: Neuroblastoma (NB); extreme gradient boosting (XGBoost); prognostic prediction; machine learning; risk stratification


Submitted Apr 02, 2026. Accepted for publication Jun 11, 2026. Published online Jun 25, 2026.

doi: 10.21037/tp-2026-0337


Highlight box

Key findings

• Six readily available clinical variables (metastatic status, Surveillance, Epidemiology, and End Results (SEER) stage, age, chemotherapy, tumor size, and primary tumor site) were identified as key prognostic factors for overall survival in pediatric neuroblastoma (NB).

• Among six machine-learning algorithms, the extreme gradient boosting (XGBoost) model demonstrated the best predictive performance and enabled effective risk stratification into low-, intermediate-, and high-risk groups.

What is known and what is new?

• NB is a highly heterogeneous pediatric malignancy, and accurate prognostic assessment remains essential for risk-adapted management. Existing prognostic systems, including the International Neuroblastoma Risk Group (INRG) classification, rely heavily on biological and molecular markers but may not always provide individualized survival estimates.

• This study developed and internally validated an XGBoost-based prognostic prediction model using a large population-based cohort from the SEER database and established an accompanying risk stratification system based on routinely available clinical variables.

What is the implication, and what should change now?

• The proposed XGBoost model may serve as a complementary prognostic tool by providing individualized survival probability estimates in addition to existing risk-classification systems.

• Risk stratification based on routinely available clinical variables may assist clinicians in prognostic assessment, follow-up planning, and patient counseling, particularly when comprehensive molecular profiling is unavailable.

• Further external validation and integration of molecular biomarkers are needed before routine clinical implementation.


Introduction

Neuroblastoma (NB) is a heterogeneous malignancy arising from the developing sympathetic nervous system and represents the most common extracranial solid tumor in children (1,2). Although NB accounts for approximately 6–10% of all pediatric cancers, it is responsible for nearly 15% of childhood cancer-related deaths owing to its highly variable biological behavior and propensity for metastatic spread (3-5). Clinical outcomes range from spontaneous regression in some low-risk patients to rapid disease progression and treatment failure in high-risk cases. Despite substantial advances in multimodal treatment strategies, including chemotherapy, surgical resection, radiotherapy, autologous stem cell transplantation, immunotherapy, and targeted therapies, the prognosis of high-risk NB remains unsatisfactory, with long-term survival rates still below 50% in many cohorts (6-8). Accurate risk assessment is therefore essential for treatment selection, prognostic counseling, and follow-up planning. Currently, prognostic evaluation of NB primarily relies on established risk stratification systems, including the International Neuroblastoma Staging System (INSS) and the International Neuroblastoma Risk Group Staging System (INRGSS) (9-11). These systems incorporate important clinical and biological factors and play a central role in therapeutic decision-making. However, they mainly classify patients into broad risk categories and may not provide individualized estimates of survival probability. Furthermore, interactions among demographic, tumor-related, and treatment-related variables are often complex and nonlinear, making individualized prognostic assessment challenging using conventional risk classification approaches alone.

Several prognostic models for NB have been reported in recent years, including nomogram-based models and regression-based survival prediction tools developed from institutional and population-based datasets (12,13). Although these models have demonstrated reasonable predictive performance, many were developed using conventional statistical methods, relied on relatively limited predictor sets, or lacked comprehensive evaluation across different modeling approaches. In addition, few studies have systematically compared multiple machine-learning algorithms for individualized survival prediction in large population-based cohorts of NB patients.

Machine learning has emerged as a promising approach for clinical prediction because of its ability to model complex nonlinear relationships and high-order interactions among predictors (14-17). Compared with traditional regression-based methods, machine-learning algorithms may offer improved flexibility when handling heterogeneous clinical data and have shown encouraging results in prognostic modeling across multiple malignancies (18-21). Among these algorithms, extreme gradient boosting (XGBoost) has attracted considerable attention due to its regularization mechanisms, computational efficiency, and strong predictive performance in oncology-related prediction tasks (18-21). Nevertheless, evidence regarding its utility for individualized survival prediction and risk stratification in NB remains limited.

Therefore, the present study aimed to develop and internally validate prognostic prediction models for overall survival (OS) in patients with NB using a large population-based cohort from the Surveillance, Epidemiology, and End Results (SEER) database. Six machine-learning algorithms, including XGBoost, logistic regression (LR), support vector machine (SVM), random forest (RF), k-nearest neighbor (KNN), and decision tree ID3, were compared to identify the optimal predictive model. In addition, we sought to establish an XGBoost-based risk stratification system to facilitate individualized prognostic assessment. The proposed model is intended as a complementary prognostic tool based on routinely available clinical variables and requires further external validation before clinical implementation. We present this article in accordance with the TRIPOD reporting checklist (available at https://tp.amegroups.com/article/view/10.21037/tp-2026-0337/rc).


Methods

Data collection and study population

This prognostic model development study was conducted using data extracted from the SEER database of the National Cancer Institute. The SEER program collects population-based cancer incidence, treatment, and survival data from multiple cancer registries and currently covers approximately 48% of the U.S. population. Data quality is ensured through standardized coding procedures, routine quality control audits, and systematic follow-up mechanisms. The study was conducted in accordance with the Declaration of Helsinki and its subsequent amendments.

SEER*Stat software (version 8.4.4) was used to identify patients diagnosed with NB (ICD-O-3 code: 9500/3) or ganglioneuroblastoma (ICD-O-3 code: 9490/3) between January 2000 and December 2021.

Eligible patients met the following criteria: (I) age ≤18 years; and (II) histopathological confirmation of NB or GNB. Exclusion criteria included: (I) missing tumor size or SEER stage information; (II) unknown surgical treatment status; and (III) survival time <1 month or unknown survival duration.

After applying the eligibility criteria, 2,427 patients were included in the final analysis (Figure 1).

Figure 1 Flowchart of study design and patient selection. AUC, area under the curve; ML, machine learning; ROC, receiving operating characteristic; SEER, Surveillance, Epidemiology, and End Results.

Predictor variables and outcome definition

Candidate predictors included demographic variables (age, sex, race), tumor-related characteristics (histological subtype, primary tumor site, tumor size, tumor grade, distant metastasis, SEER stage, and regional lymph node involvement), and treatment-related variables (surgery, regional lymph node surgery, chemotherapy, and radiotherapy).

The primary outcome was OS, defined as the time from diagnosis to death from any cause or last follow-up. Survival time was obtained from the SEER variable “Survival Months”. Patients alive at the last follow-up were treated as censored observations.

For prognostic prediction, 1-, 3-, and 5-year OS outcomes were evaluated.

Data preprocessing and missing data handling

Data preprocessing followed standard SEER procedures. Patients with missing information regarding tumor size, SEER stage, surgical treatment status, or survival duration were excluded from the analysis. Therefore, a complete-case analysis was performed.

Continuous variables were categorized using X-tile software (version 3.6.1) to determine optimal prognostic cut-off values. Age was categorized into <1, 1 to <2, and ≥2 years, while tumor size was categorized into ≤60, >60–95, and >95 mm.

Model development and internal validation

The study was designed as a prognostic model development study with internal validation. Patients were randomly assigned to a training cohort (80%, n=1,941) and an internal validation cohort (20%, n=486) using a fixed random seed to ensure reproducibility. Feature selection was performed using least absolute shrinkage and selection operator (LASSO)-Cox regression. The optimal penalty parameter (λ) was determined using five-fold cross-validation.

Six machine learning algorithms were implemented to develop predictive models for 1-, 3-, and 5-year OS in NB patients, including XGBoost, LR, SVM, RF, KNN, and ID3. XGBoost is an optimized version of gradient boosting decision trees (GBDT) that incorporates L1 and L2 regularization to prevent overfitting, employs a quantile approximation algorithm for accelerated feature discretization, and enhances efficiency through parallel computing and feature subsampling (22). LR is a generalized linear model widely used for binary classification. It maps linear combinations of input variables onto a probability space using the logistic function, estimating the likelihood of a given sample belonging to a particular class (23). SVM is a supervised learning algorithm that constructs a hyperplane to maximize the margin between different classes. It is particularly effective for small-sample, nonlinear, and high-dimensional data classification, demonstrating strong performance in medical data analysis (24,25). RF is an ensemble learning method that aggregates multiple decision trees through a voting mechanism, enhancing model generalization and stability. It is capable of handling high-dimensional input features and is less prone to overfitting. KNN is an instance-based algorithm that classifies samples by computing their similarity to training samples. It assigns a class label based on a majority vote from the k-nearest neighbors, making it suitable for multi-class classification and non-linear decision boundary problems (26). ID3 constructs decision trees using information gain as the splitting criterion. It recursively selects features with the highest information gain to partition data, producing an interpretable decision process.

Hyperparameter optimization was performed within the training cohort using five-fold cross-validation combined with grid search. Model performance was subsequently evaluated in the independent internal validation cohort. Hyperparameters including n_estimators [100, 200], max_depth [3, 5, 7], learning_rate [0.01, 0.1, 0.2], subsample [0.8, 1.0], and colsample_bytree [0.8, 1.0] were optimized using grid search. The optimal parameter combination identified during cross-validation was subsequently used to construct the final model.

Model performance assessment

Model discrimination was evaluated using time-dependent receiver operating characteristic (ROC) curves, area under the ROC curve (AUC), and concordance index (C-index). Prediction accuracy was assessed using the Brier score. Model performance was evaluated in both the training and internal validation cohorts. Higher AUC and C-index values indicate better discrimination, whereas lower Brier scores indicate improved prediction accuracy. Calibration analysis was performed to assess the agreement between predicted and observed 5-year OS probabilities in the internal validation cohort.

Risk stratification

To further evaluate the discriminatory power of the predictive model, a risk score was computed for each patient. The optimal cut-off value for the risk score was determined, enabling risk stratification into three prognostic groups. Kaplan-Meier survival analysis and log-rank tests were performed to assess differences among risk groups, validating prognostic distinctions between them.

Statistical analysis

Continuous variables (e.g., age, tumor size) were converted into categorical variables using X-tile software (version 3.6.1) to optimize their statistical efficiency (Figure 2). Specifically: age was categorized into three groups: <1, 1 to <2, and ≥2 years. Tumor size was stratified into ≤60, >60–95, and >95 mm. Categorical variables were presented as frequencies (percentages), and comparisons between groups were performed using the chi-square test. A P value <0.05 was considered statistically significant. All data analyses and machine learning modeling were conducted using Python (version 3.12.9).

Figure 2 The optimal cutoff values of tumor size identified by X-tile. (A) The frequency distribution map of age. (B) The Kaplan-Meier curves for the subgroups of age (<1, 1 to <2, ≥2 years) for overall survival. (C) The frequency distribution map of tumor size. (D) The Kaplan-Meier curves for the subgroups of tumor size (≤60, >60–95, >95 mm) for overall survival.

Results

Patient characteristics

A total of 2,427 patients diagnosed with NB or GNB were included in this study. Among them, 53.23% were male, and 76.1% were White. Regarding age distribution, 47.80% of patients were aged ≥2 years. The adrenal gland was the most common primary tumor site, accounting for 50.56% of cases. At the time of diagnosis, 46.97% of patients presented with distant metastasis. In terms of treatment, 80.84% of patients underwent surgery (either radical or local resection), 65.47% received chemotherapy, and 24.43% underwent radiotherapy. All cases were randomly divided into a training set (n=1,941) and a validation set (n=486) in an 8:2 ratio. There were no significant differences in baseline characteristics between the two groups (P>0.05). Table 1 summarizes the detailed characteristics of the patient cohorts.

Table 1

Demographics and clinical characteristics of patients with neuroblastoma in the training and testing groups

Characteristics Overall, N=2,427 Training cohort, N=1,941 Testing cohort, N=486 P value
Age (years) 0.23
   <1 773 (31.85) 610 (31.43) 163 (33.54)
   1 to <2 494 (20.35) 401 (20.66) 93 (19.14)
   ≥2 1,160 (47.80) 930 (47.91) 230 (47.33)
Sex 0.43
   Male 1,292 (53.23) 1,027 (52.91) 265 (54.53)
   Female 1,135 (46.77) 914 (47.09) 221 (45.47)
Race 0.22
   White 1,847 (76.10) 1,488 (76.66) 359 (73.87)
   Black 311 (12.810) 249 (12.83) 62 (12.76)
   Other 311 (12.810) 204 (10.51) 65 (13.37)
Histologic type 0.77
   Neuroblastoma 1,989 (81.95) 1,576 (81.20) 413 (84.98)
   Ganglioneuroblastoma 438 (18.05) 365 (18.80) 73 (15.02)
Primary site 0.21
   Adrenal gland 1,227 (50.56) 979 (50.44) 248 (51.03)
   Retroperitoneum 296 (12.2) 243 (12.52) 53 (10.91)
   Enterocoelia 160 (6.59) 125 (6.44) 35 (7.2)
   Above the diaphragm 404 (16.65) 322 (16.59) 82 (16.87)
   Other 340 (14.01) 272 (14.01) 68 (13.99)
Tumor size (mm) 0.97
   ≤60 1,117 (46.02) 885 (45.60) 232 (47.74)
   >60–95 678 (27.94) 543 (27.98) 135 (27.78)
   >95 632 (26.04) 513 (26.43) 119 (24.49)
Metastasis status 0.73
   Yes 1,140 (46.97) 915 (47.14) 225 (46.3)
   No 1,287 (53.03) 1026 (52.86) 261 (53.7)
Grade recode 0.26
   I 57 (2.35) 41 (2.11) 16 (3.29)
   II 12 (0.49) 7 (0.36) 5 (1.03)
   III 1,166 (48.04) 918 (47.30) 248 (51.03)
   IV 160 (6.59) 127 (6.54) 33 (6.79)
   Unknown 1,032 (42.52) 848 (43.69) 184 (37.86)
SEER stage 0.92
   Localized 660 (27.19) 528 (27.20) 132 (27.16)
   Regional 601 (24.76) 475 (24.47) 126 (25.93)
   Distant 1,166 (48.04) 938 (48.33) 228 (46.91)
Regional nodes positive 0.12
   Positive 694 (28.59) 561 (28.90) 133 (27.37)
   Negative 272 (11.21) 211 (10.87) 61 (12.55)
   Unknown 1,461 (60.2) 1,169 (60.23) 292 (60.08)
Surgical therapy 0.09
   Radical excision 1,372 (56.53) 1,081 (55.69) 291 (59.88)
   Local excision 590 (24.31) 473 (24.37) 117 (24.07)
   No 465 (19.16) 387 (19.94) 78 (16.05)
Scope of regional lymph node surgery 0.18
   None 1,317 (54.26) 1,052 (54.20) 265 (54.53)
   1 to 3 482 (19.86) 382 (19.68) 100 (20.58)
   ≥4 314 (12.94) 255 (13.14) 59 (12.14)
   Unknown 314 (12.94) 252 (12.98) 62 (12.76)
Chemotherapy 0.43
   Yes 1,589 (65.47) 1,269 (65.38) 320 (65.84)
   No/unknown 838 (34.53) 672 (34.62) 166 (34.16)
Radiotherapy >0.99
   Yes 593 (24.43) 471 (24.27) 122 (25.1)
   No 1834 (75.57) 1470 (75.73) 364 (74.9)

Data are presented as n (%). SEER, Surveillance, Epidemiology, and End Results.

LASSO regression and variable selection

Feature selection was performed using LASSO regression. Figure 3A illustrates the coefficient paths of predictor variables at different log(λ) values. As λ increased, the coefficients of non-essential variables were gradually shrunk to zero, indicating a reduced contribution to the model. Figure 3B shows that the partial likelihood deviance increased as λ decreased, suggesting a risk of overfitting. Based on the λ value corresponding to the minimum partial likelihood deviance, six key prognostic variables were ultimately selected: age, primary tumor site, tumor size, distant metastasis, SEER stage, and chemotherapy.

Figure 3 The results of the LASSO regression. (A) LASSO coefficient profiles; (B) cross-validation curve of LASSO regression. LASSO, least absolute shrinkage and selection operator.

Predictive models development and performance evaluation

The time-dependent AUC of the six machine learning models in the training and validation sets is presented in Figure 4, with detailed predictive performance metrics summarized in Table 2. During the training phase, the RF model achieved the highest AUC values at all time points and demonstrated a lower Brier score, indicating strong discrimination between events and non-events and relatively accurate probability estimation. Additionally, its C-Index [95% confidence interval (CI)] was 0.806 (0.787–0.825), further confirming its strong predictive ability. However, in the independent validation set, the XGBoost model exhibited superior stability and generalization capability, achieving AUCs of 0.767, 0.796, and 0.807 for 1-, 3-, and 5-year survival predictions, respectively. The C-Index was 0.784 (95% CI: 0.746–0.821), significantly outperforming the other models. This indicates that XGBoost not only fit the training data well but also maintained robust predictive performance on unseen data. To further evaluate the reliability of predicted survival probabilities, calibration analyses were performed for the XGBoost model. As shown in Figure 5, the calibration curves for 5-year overall survival prediction demonstrated good agreement between the predicted probabilities and observed outcomes in both the training cohort (Brier score =0.1056) and the internal validation cohort (Brier score =0.1172). The calibration plots indicated that the XGBoost model provided reasonably accurate probability estimates, supporting its potential utility for individualized prognostic assessment.

Figure 4 ROC curves and AUCs demonstrating the predictions of the six models: XGBoost, LR, SVM, RF, KNN, and ID3. (A,C,E,G,I,K) The training cohort. (B,D,F,H,J,L) The validation cohort. AUC, area under the curve; ID3, Iterative Dichotomiser 3; KNN, K-nearest neighbor; LR, logistic regression; RF, random forest; ROC, receiving operating characteristic; SVM, support vector machine; XGBoost, extreme gradient boosting.

Table 2

Performance of prognostic models built by machine learning algorithms in the training and test sets

Model AUC Brier score C-index (95% CI)
1-year 3-year 5-year 1-year 3-year 5-year
Training set
   XGBoost 0.782 0.804 0.824 0.069 0.096 0.106 0.793 (0.775–0.815)
   LR 0.777 0.802 0.821 0.078 0.100 0.108 0.790 (0.766–0.813)
   SVM 0.703 0.736 0.744 0.060 0.108 0.129 0.723 (0.696–0.748)
   RF 0.801 0.820 0.837 0.071 0.096 0.105 0.806 (0.787–0.825)
   KNN 0.782 0.790 0.806 0.072 0.101 0.112 0.778 (0.757–0.799)
   ID3 0.806 0.817 0.833 0.076 0.098 0.106 0.804 (0.784–0.825)
Test set
   XGBoost 0.767 0.796 0.807 0.072 0.107 0.117 0.784 (0.746–0.821)
   LR 0.757 0.779 0.800 0.079 0.106 0.112 0.774 (0.739–0.807)
   SVM 0.654 0.679 0.672 0.064 0.112 0.128 0.653 (0.607–0.697)
   RF 0.743 0.776 0.795 0.076 0.105 0.112 0.769 (0.735–0.802)
   KNN 0.681 0.722 0.739 0.079 0.114 0.125 0.718 (0.678–0.761)
   ID3 0.724 0.761 0.780 0.082 0.110 0.117 0.753 (0.717–0.788)

AUC, area under the curve; CI, confidence interval; ID3, Iterative Dichotomiser 3; KNN, K-nearest neighbor; LR, logistic regression; RF, random forest; SVM, support vector machine; XGBoost, extreme gradient boosting.

Figure 5 Calibration curve of the XGBoost model for predicting 5-year overall survival in the training cohort (A) and validation cohort (B). OS, overall survival; XGBoost, extreme gradient boosting.

Considering model complexity, interpretability, and practical applicability, although the RF model demonstrated a slight advantage in the training phase, XGBoost was ultimately recommended as the optimal predictive model in this study due to its better generalization ability and stability.

Feature importance of the XGBoost model

The feature importance analysis of the XGBoost model (Figure 6) revealed that “metastatic status” contributed the most to survival prediction, followed by “SEER stage”, “age”, “chemotherapy”, “tumor size”, and “primary tumor site”, in decreasing order of importance. This finding aligns with the clinical consensus that metastatic disease is an independent adverse prognostic factor.

Figure 6 The XGBoost model was used to calculate the importance of six features. SEER, Surveillance, Epidemiology, and End Results; XGBoost, extreme gradient boosting.

Risk stratification

The XGBoost model was further employed for risk stratification, categorizing patients into high-risk, intermediate-risk, and low-risk groups. Based on optimal survival differences, two risk score thresholds were identified at 0.32 and 0.16 (Figure 7A), which were used to classify patients in both the training and validation cohorts. Kaplan-Meier survival analysis demonstrated that 5-year survival rates in the high-risk group were significantly lower than those in the intermediate- and low-risk groups. The log-rank test confirmed that the differences among the risk groups were statistically significant (Figure 7B,7C).

Figure 7 The XGBoost model-based risk stratification. Risk score distribution (A); Kaplan-Meier analysis was performed on high-, medium- and low-risk groups within both the training (B) and testing (C) cohorts. XGBoost, extreme gradient boosting.

Discussion

In this study, we identified six readily available clinical variables associated with OS in patients with NB, including metastatic status, SEER stage, age, chemotherapy, tumor size, and primary tumor site. Based on these variables, six machine-learning algorithms were developed and compared for prognostic prediction. Among the evaluated models, XGBoost achieved the best overall predictive performance in both the development and internal validation cohorts. Feature importance analysis further identified metastatic status as the most influential predictor. Using the predicted risk scores generated by the XGBoost model, we established a risk stratification system that categorized patients into low-, intermediate-, and high-risk groups. These findings suggest that routinely available clinical variables may provide useful prognostic information and support individualized survival assessment in pediatric NB.

Over the past decade, improving prognostic assessment to refine clinical decision-making for cancer patients has been a major research focus (27). However, traditional methods for predicting survival outcomes based on single prognostic factors face challenges such as limited accuracy and the complexity of integrating multiple factors. Given the significant heterogeneity in NB prognosis and the inadequacy of current tumor staging systems in providing precise, personalized predictions, developing widely applicable individualized models is of great clinical significance. Although Zhuo et al. (12) developed a nomogram model and risk stratification system for predicting OS in NB patients, and Amano et al. (13) integrated urinary catecholamine metabolites (HVA and VMA) to construct a risk assessment system for NB diagnosis and pre-treatment evaluation, these models primarily rely on traditional statistical methods and suffer from difficulties in clinical feature collection and poor generalizability. In recent years, machine learning-based artificial intelligence models have gained increasing attention in clinical practice, emerging as valuable tools for developing predictive models and significantly enhancing disease outcome predictions. Recent studies have examined XGBoost’s application in predicting cancer survival and validated its robust predictive performance across various cancers, such as elderly intrahepatic cholangiocarcinoma and breast cancer with brain metastases (28). In this study, among the six machine learning models, XGBoost demonstrated the most outstanding prognostic prediction performance. Nevertheless, although the model achieved acceptable predictive performance, its results should be interpreted in the context of internal validation only, and further external validation is required before broader clinical implementation can be considered.

To systematically evaluate XGBoost’s feature importance and enhance model interpretability, we ranked the predictive power of individual features. The results showed that metastatic status was the most predictive feature, followed by SEER stage, age, chemotherapy, tumor size, and primary tumor site. These findings align with previous research. Metastatic status is widely recognized as an independent prognostic factor, significantly negatively correlated with OS and cancer-specific survival (CSS) (29). Clinical data suggest that adrenal-origin NBs exhibit a higher tendency for metastasis, with bone and liver metastases being the most common single-organ metastatic patterns (30,31). However, current treatment strategies provide limited improvements in 5-year survival rates for patients with distant metastasis (32). leading to a poorer prognosis for these patients. SEER staging is a classification method used in the SEER program to describe the extent and severity of cancer, categorizing it into localized, regional, and distant stages. It aids clinicians in predicting patient survival and disease progression. In our study, SEER stage also demonstrated high predictive utility for NB prognosis within the XGBoost model. Age is another critical prognostic factor in NB. For stage 3 and 4 MYCN non-amplified tumors, patients younger than 18 months exhibit better event-free survival (EFS) than those older than 18 months (33). Chemotherapy was also identified as a prognostic factor through LASSO regression, potentially due to treatment-related mortality events (12). Traditionally, prognostic factors in NB are combined to stratify patients’ risk levels and guide treatment decisions. However, major classification systems such as the Children’s Oncology Group (COG) and the International Neuroblastoma Risk Group (INRG) do not include tumor size as a prognostic factor. Nevertheless, recent studies suggest that tumor size plays a crucial role in NB prognosis (34). Additionally, previous studies have indicated that primary tumor site significantly influences prognosis. Adrenal-origin tumors are associated with lower survival rates, whereas thoracic-origin tumors show higher survival rates (31).

To further explore the clinical relevance of the proposed model, we developed an XGBoost-based risk stratification system that classified patients into distinct prognostic groups. The model successfully differentiated survival outcomes in both the development and internal validation cohorts. Importantly, unlike established classification systems such as the INRG system, which primarily categorizes patients into broad risk groups for treatment allocation, the present model was designed to estimate individualized survival probabilities using routinely available clinical variables. Therefore, the proposed model should be considered a complementary prognostic tool rather than a replacement for existing biologically driven risk-classification systems. If validated in independent cohorts, such individualized risk estimates may provide additional information for prognostic assessment, follow-up planning, and patient counseling.

Several limitations should be acknowledged. First, this was a retrospective registry-based study and is therefore subject to potential selection bias, residual confounding, and misclassification of recorded variables. Second, although the SEER database provides a large population-based cohort, several important biological and molecular prognostic factors were unavailable, including MYCN amplification status, chromosome 11q aberration, DNA ploidy, histopathological classification, and urinary catecholamine metabolites. The absence of these variables may have limited model performance and clinical applicability. Third, complete-case analysis was performed, which may have introduced additional selection bias. Fourth, model development and validation were conducted using a random split of the same SEER dataset; therefore, the present study represents internal validation rather than true external validation. Although calibration and discrimination analyses demonstrated acceptable predictive performance, the possibility of model optimism and overfitting cannot be completely excluded. Future studies incorporating molecular, genetic, and clinical variables and validating the model in independent multicenter cohorts are needed before routine clinical application can be recommended.


Conclusions

In summary, we developed and internally validated six machine-learning models for predicting OS in pediatric NB using a population-based SEER cohort. Among these models, XGBoost demonstrated the best predictive performance and was subsequently used to establish a risk stratification system. The proposed model may facilitate individualized prognostic assessment based on routinely available clinical variables. However, given the retrospective study design, the absence of important biological predictors, and the lack of external validation, further evaluation in independent cohorts is required before clinical implementation.


Acknowledgments

We thank the SEER database for its open data access.


Footnote

Reporting Checklist: The authors have completed the TRIPOD reporting checklist. Available at https://tp.amegroups.com/article/view/10.21037/tp-2026-0337/rc

Peer Review File: Available at https://tp.amegroups.com/article/view/10.21037/tp-2026-0337/prf

Funding: None.

Conflicts of Interest: All authors have completed the ICMJE uniform disclosure form (available at https://tp.amegroups.com/article/view/10.21037/tp-2026-0337/coif). The authors have no conflicts of interest to declare.

Ethical Statement: The authors are accountable for all aspects of the work in ensuring that questions related to the accuracy or integrity of any part of the work are appropriately investigated and resolved. The study was conducted in accordance with the Declaration of Helsinki and its subsequent amendments.

Open Access Statement: This is an Open Access article distributed in accordance with the Creative Commons Attribution-NonCommercial-NoDerivs 4.0 International License (CC BY-NC-ND 4.0), which permits the non-commercial replication and distribution of the article with the strict proviso that no changes or edits are made and the original work is properly cited (including links to both the formal publication through the relevant DOI and the license). See: https://creativecommons.org/licenses/by-nc-nd/4.0/.


References

  1. Maris JM. Recent advances in neuroblastoma. N Engl J Med 2010;362:2202-11. [Crossref] [PubMed]
  2. Mlakar V, Jurkovic Mlakar S, Lopez G, et al. 11q deletion in neuroblastoma: a review of biological and clinical implications. Mol Cancer 2017;16:114. [Crossref] [PubMed]
  3. Tonini GP, Capasso M. Genetic predisposition and chromosome instability in neuroblastoma. Cancer Metastasis Rev 2020;39:275-85. [Crossref] [PubMed]
  4. Gatta G, Botta L, Rossi S, et al. Childhood cancer survival in Europe 1999-2007: results of EUROCARE-5--a population-based study. Lancet Oncol 2014;15:35-47. [Crossref] [PubMed]
  5. Lu W, Wang Q, Xu C, et al. SUMOylation is essential for Sirt2 tumor-suppressor function in neuroblastoma. Neoplasia 2021;23:129-39. [Crossref] [PubMed]
  6. Croteau N, Nuchtern J, LaQuaglia MP. Management of Neuroblastoma in Pediatric Patients. Surg Oncol Clin N Am 2021;30:291-304. [Crossref] [PubMed]
  7. Chung C, Boterberg T, Lucas J, et al. Neuroblastoma. Pediatr Blood Cancer 2021;68:e28473. [Crossref] [PubMed]
  8. Park JR, Kreissman SG, London WB, et al. Effect of Tandem Autologous Stem Cell Transplant vs Single Transplant on Event-Free Survival in Patients With High-Risk Neuroblastoma: A Randomized Clinical Trial. JAMA 2019;322:746-55. [Crossref] [PubMed]
  9. Irwin MS, Park JR. Neuroblastoma: paradigm for precision medicine. Pediatr Clin North Am 2015;62:225-56. [Crossref] [PubMed]
  10. Brodeur GM, Pritchard J, Berthold F, et al. Revisions of the international criteria for neuroblastoma diagnosis, staging, and response to treatment. J Clin Oncol 1993;11:1466-77. [Crossref] [PubMed]
  11. Monclair T, Brodeur GM, Ambros PF, et al. The International Neuroblastoma Risk Group (INRG) staging system: an INRG Task Force report. J Clin Oncol 2009;27:298-303. [Crossref] [PubMed]
  12. Zhuo X, Xia L, Tang W, et al. A practical nomogram and risk stratification system for predicting survival outcomes in neuroblastoma patients: a SEER population-based study. J Cancer Res Clin Oncol 2023;149:12285-96. [Crossref] [PubMed]
  13. Amano H, Uchida H, Harada K, et al. Scoring system for diagnosis and pretreatment risk assessment of neuroblastoma using urinary biomarker combinations. Cancer Sci 2024;115:1634-45. [Crossref] [PubMed]
  14. Tran KA, Kondrashova O, Bradley A, et al. Deep learning in cancer diagnosis, prognosis and treatment selection. Genome Med 2021;13:152. [Crossref] [PubMed]
  15. Nguyen TT, Ho CT, Bui HTT, et al. Multidimensional Machine Learning for Assessing Parameters Associated With COVID-19 in Vietnam: Validation Study. JMIR Form Res 2023;7:e42895. [Crossref] [PubMed]
  16. Sajda P. Machine learning for detection and diagnosis of disease. Annu Rev Biomed Eng 2006;8:537-65. [Crossref] [PubMed]
  17. Senders JT, Staples P, Mehrtash A, et al. An Online Calculator for the Prediction of Survival in Glioblastoma Patients Using Classical Statistics and Machine Learning. Neurosurgery 2020;86:E184-92. [Crossref] [PubMed]
  18. Xu Q, Lu X. Development and validation of an XGBoost model to predict 5-year survival in elderly patients with intrahepatic cholangiocarcinoma after surgery: a SEER-based study. J Gastrointest Oncol 2022;13:3290-9. [Crossref] [PubMed]
  19. Li C, Liu M, Zhang Y, et al. Novel models by machine learning to predict prognosis of breast cancer brain metastases. J Transl Med 2023;21:404. [Crossref] [PubMed]
  20. Zhong X, Lin Y, Zhang W, et al. Predicting diagnosis and survival of bone metastasis in breast cancer using machine learning. Sci Rep 2023;13:18301. [Crossref] [PubMed]
  21. Kinoshita F, Takenaka T, Yamashita T, et al. Development of artificial intelligence prognostic model for surgically resected non-small cell lung cancer. Sci Rep 2023;13:15683. [Crossref] [PubMed]
  22. Takefuji Y. Beyond XGBoost and SHAP: Unveiling true feature importance. J Hazard Mater 2025;488:137382. [Crossref] [PubMed]
  23. Zuo W, Yang X. A machine learning model predicts stroke associated with blood cadmium level. Sci Rep 2024;14:14739. [Crossref] [PubMed]
  24. Sayeed MA, Rahman A, Rahman A, et al. On the interpretability of the SVM model for predicting infant mortality in Bangladesh. J Health Popul Nutr 2024;43:170. [Crossref] [PubMed]
  25. Fan R, Yu N, Li G, et al. Machine-learning model comprising five clinical indices and liver stiffness measurement can accurately identify MASLD-related liver fibrosis. Liver Int 2024;44:749-59. [Crossref] [PubMed]
  26. Shrivastava T, Singh V, Agrawal A. Autism spectrum disorder detection with kNN imputer and machine learning classifiers via questionnaire mode of screening. Health Inf Sci Syst 2024;12:18. [Crossref] [PubMed]
  27. Staffa SJ, Zurakowski D. Statistical Development and Validation of Clinical Prediction Models. Anesthesiology 2021;135:396-405. [Crossref] [PubMed]
  28. Dong B, Zhang H, Duan Y, et al. Development of a machine learning-based model to predict prognosis of alpha-fetoprotein-positive hepatocellular carcinoma. J Transl Med 2024;22:455. [Crossref] [PubMed]
  29. Liu S, Yin W, Lin Y, et al. Metastasis pattern and prognosis in children with neuroblastoma. World J Surg Oncol 2023;21:130. [Crossref] [PubMed]
  30. Oldridge DA, Truong B, Russ D, et al. Differences in Genomic Profiles and Outcomes Between Thoracic and Adrenal Neuroblastoma. J Natl Cancer Inst 2019;111:1192-201. [Crossref] [PubMed]
  31. Vo KT, Matthay KK, Neuhaus J, et al. Clinical, biologic, and prognostic differences on the basis of primary tumor site in neuroblastoma: a report from the international neuroblastoma risk group project. J Clin Oncol 2014;32:3169-76. [Crossref] [PubMed]
  32. Morgenstern DA, Bagatell R, Cohn SL, et al. The challenge of defining “ultra-high-risk” neuroblastoma. Pediatr Blood Cancer 2019;66:e27556. [Crossref] [PubMed]
  33. Sokol E, Desai AV, Applebaum MA, et al. Age, Diagnostic Category, Tumor Grade, and Mitosis-Karyorrhexis Index Are Independently Prognostic in Neuroblastoma: An INRG Project. J Clin Oncol 2020;38:1906-18. [Crossref] [PubMed]
  34. Wang JX, Cao ZY, Wang CX, et al. Prognostic impact of tumor size on patients with neuroblastoma in a SEER-based study. Cancer Med 2022;11:2779-89. [Crossref] [PubMed]
Cite this article as: Li H, Luo J, Yang J. Development and internal validation of an XGBoost-based prognostic prediction model and risk stratification system for overall survival in pediatric neuroblastoma: a SEER-based study. Transl Pediatr 2026;15(7):270. doi: 10.21037/tp-2026-0337

Download Citation