Mortality Prediction in Imbalanced Pediatric Oncology Data: A Cost-Sensitive Hybrid Resampling Framework Using Recurrent Deep Learning
Background: Severe class imbalance hampers mortality prediction in pediatric oncology, where clinical-improvement discharges greatly outnumber deaths and bias standard models toward the majority class. Objectives: We aimed to develop and evaluate a Cost-Sensitive Hybrid Resampling Framework (CSHRF) across recurrent deep-learning architectures to mitigate this bias and improve discharge-outcome classification. Methods: A retrospective cross-sectional study was conducted using electronic health record data from a National Institute of Health in Mexico City during 2023. We analyzed 1607 hospital-discharge records, comprising 1554 clinical improvements and 53 deaths. The CSHRF combines cost-sensitive learning, which amplified the mortality class by a factor of 2.5, with a fixed-seed hybrid resampling protocol embedded in a regularized funnel-shaped architecture (N → N/2 → N/4) trained with gradient clipping (clipnorm = 1.0). Simple Recurrent Neural Network (SimpleRNN), Long Short-Term Memory (LSTM), and Gated Recurrent Unit (GRU) models were compared with a naïve recurrent baseline and standard tabular classifiers (Weighted Logistic Regression, Balanced Random Forest, and XGBoost) using stratified cross-validation. Performance was evaluated using the area under the receiver operating characteristic curve (AUC), G-mean, sensitivity, specificity, precision, F1-score, mean absolute error (MAE), and root mean squared error (RMSE). Results: The naïve baseline showed limited mortality detection (sensitivity 0.150 ± 0.147). The proposed framework substantially improved minority-class detection, with CSHRF-GRU achieving sensitivity of 0.740 ± 0.145, specificity of 0.700 ± 0.076, G-mean of 0.714 ± 0.045, and AUC of 0.761 ± 0.075. Ablation analysis demonstrated that cost-sensitive weighting contributed most substantially to mortality detection, whereas hybrid resampling alone produced more modest gains. Among tabular baselines, Weighted Logistic Regression achieved competitive performance, with sensitivity of 0.684 ± 0.130 and G-mean of 0.767 ± 0.073. Precision remained low (0.079 ± 0.015) and probabilistic calibration errors were substantial (MAE 0.448; RMSE 0.488), reflecting the trade-off required to prioritize mortality detection under severe imbalance. Conclusions: The CSHRF improved discharge-outcome classification in a highly imbalanced pediatric oncology dataset and provides a methodological framework for data-constrained hospital environments. Given its high false-positive rate and probability overestimation, the approach should be interpreted as a discharge-related outcome classification model rather than a standalone clinical decision-support system, and future work should incorporate richer clinical and disease-specific variables to improve predictive performance and calibration.
Authors
- Sandra Hernández‐Corral (ORCID: https://orcid.org/0000-0003-4292-2380)
- Rosa Paola Figuerola-Escoto (ORCID: https://orcid.org/0000-0002-1349-4623)
- Leonor Rivera-Rivera (ORCID: https://orcid.org/0000-0002-8501-0173)
- David Luna (ORCID: https://orcid.org/0000-0003-0427-3789)
- Ernesto Roldán-Valadéz (ORCID: https://orcid.org/0000-0002-7116-5289)
- José Fernando Mora Romo (ORCID: https://orcid.org/0000-0002-6201-4622)
- Hugo Sandoval (ORCID: https://orcid.org/0000-0002-9622-1558)
- Ahideé Leyva-López (ORCID: https://orcid.org/0000-0002-8413-0771)
- Filiberto Toledano‐Toledano (ORCID: https://orcid.org/0000-0002-7126-5351)
- Joel Martínez-Salazar (ORCID: https://orcid.org/0009-0007-8753-4257)
Institutions
- Hospital Infantil de México Federico Gómez (MX)
- Instituto Nacional de Rehabilitación (MX)
- Instituto Nacional de Salud Pública (MX)
- Centro Regional de Investigaciones Multidisciplinarias (MX)
- Universidad Autónoma de Zacatecas "Francisco García Salinas" (MX)
- Instituto Politécnico Nacional (MX)
- Universidad Autónoma del Estado de Morelos (MX)
- Universidad Nacional Autónoma de México (MX)
Publication Details
- Journal
- Medical Sciences
- Published
- 2026-10-27
- DOI
- https://doi.org/10.3390/medsci14070713
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00