https://orcid.org/0009-0006-6121-3730
Figures
Abstract
Introduction
Burn injuries are a major global health challenge. Predicting mortality in burn patients using data mining can enhance therapeutic decision-making, but the long-term reliability of such models is a critical concern for safe clinical deployment. This study aimed to develop a high-performing mortality prediction model and rigorously evaluate its temporal stability.
Methods
This retrospective cohort study utilized data from 651 patients, comprising 93 predictive features, from the burn registry of the Injuries and Burn Subspecialized Teaching Hospital, affiliated with Isfahan University of Medical Sciences. After data preprocessing, including robust imputation using a Generalized Linear Model (GLM), five tree-based models were developed. Model performance was evaluated using a 10-fold stratified cross-validation and a rigorous temporal validation, with models trained on an initial nine-month cohort (n = 435) and tested on a subsequent nine-month cohort (n = 216).
Results
The Gradient Boosted Trees (GBT) model demonstrated the best overall performance in cross-validation, achieving an accuracy of 93.1% (± 2.1%), an Area Under the ROC Curve (AUC) of 0.966 (± 0.010), and the lowest Brier Score of 0.060 (± 0.017), indicating superior discrimination and calibration. The most influential predictors were the Abbreviated Burn Severity Index (ABSI), Total Burn Surface Area (TBSA), and the percentage of third-degree burns. Crucially, the temporal validation revealed strong model stability, with the GBT model’s AUC only decreasing slightly to 0.948. However, a clinically significant drop in sensitivity was observed (from 78.1% to 68.3%).
Conclusion
Tree-based models, particularly GBT, are powerful and accurate tools for predicting burn mortality. While our model demonstrated strong temporal stability, the observed decrease in sensitivity highlights that even robust models are subject to performance shifts over time. This underscores the vital importance of implementing a governance framework, including continuous monitoring and periodic recalibration, to ensure the sustained safety and efficacy of predictive models in clinical practice.
Citation: Sabet Kouhanjani Y, Sattari M, Ehteshami A (2026) From data to decisions: Predicting inpatient burn mortality with advanced classification models. PLoS One 21(1): e0338564. https://doi.org/10.1371/journal.pone.0338564
Editor: Amirreza Khalaji, Emory University School of Medicine, UNITED STATES OF AMERICA
Received: February 14, 2025; Accepted: November 24, 2025; Published: January 2, 2026
Copyright: © 2026 Sabet Kouhanjani et al. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
Data Availability: This study was a secondary analysis of data extracted from the burn registry of the Injuries and Burn Subspecialized Teaching Hospital, affiliated with Isfahan University of Medical Sciences. Although the dataset was de-identified prior to analysis, it remains subject to national data protection laws and institutional privacy regulations. Due to restrictions imposed by the Ethics Committee and data owner, the dataset cannot be shared publicly. However, access to a minimal de-identified dataset may be granted to qualified researchers upon reasonable request. Data access requests should be submitted formally to the Isfahan University of Medical Sciences Ethics Committee at resrec@mui.ac.ir. The full RapidMiner processes used for model development and validation in this study are publicly available in the Zenodo repository under DOI: 10.5281/zenodo.15639850.
Funding: The author(s) received no specific funding for this work.
Competing interests: The authors have declared that no competing interests exist.
Introduction
Burn injuries are among the most common and severe medical conditions globally, leading not only to physical trauma but also to substantial psychological, social, and economic consequences [1]. According to the World Health Organization (WHO), burn injuries result in approximately 180,000 deaths annually, with the majority occurring in low- and middle-income countries [2]. In Iran, data from 2017 show that more than 35,000 patients were hospitalized due to burns, highlighting the significant national burden and emphasizing challenges in burn prevention and management within the healthcare system [3]. Burn injuries are also associated with long-term physical and psychological complications, high medical costs, and income loss [4]. Despite significant advances in burn care, the complex physiological responses to burns continue to pose major challenges to improving patient outcomes [5]. Moreover, burn-related complications can further hinder recovery. One such complication is gastrointestinal-origin abdominal pain, which has been identified as a prevalent yet often overlooked post-burn issue that requires dedicated clinical attention [6]. As healthcare systems increasingly adopt digital technologies, access to vast amounts of health-related data has opened new possibilities for optimizing patient care [7]. It is estimated that over a patient’s lifetime, approximately one million gigabytes of health data are generated, with this volume doubling every 2–5 years [8]. Analyzing such high-dimensional datasets exceeds the capabilities of traditional statistical methods and requires advanced techniques such as data mining [9]. Data mining, which incorporates statistical modeling, machine learning, and artificial intelligence, enables the extraction of hidden patterns and meaningful insights from large datasets [10,11]. In medicine, these tools have been applied to support prediction, diagnosis, and treatment decision-making [12].
Recent studies have demonstrated the value of data mining in burn care. For example, Ahmadi et al. (2021) used data mining to examine factors influencing the delay between burn injury and treatment initiation [13]. Yang et al. (2023) employed gradient boosting to predict inhalation injury with high diagnostic accuracy [14]. Aydin et al. (2023) applied machine learning to classify esophageal burns in children following caustic ingestion [15]. Yazici et al. (2023) evaluated various machine learning algorithms for predicting burn-related mortality [16]. In Iran, Ehteshami et al. (2024) analyzed data from the Injuries and Burn Subspecialized Teaching Hospital affiliated with Isfahan University of Medical Sciences using nine features to compare several algorithms for mortality prediction [17].
However, the clinical applicability of such models often faces two critical challenges. First, the generalizability of results from such models may be limited due to dataset-specific characteristics [18]. Second, their reported performance, may not reflect the model’s stability over time in a dynamic clinical environment, a challenge known as ‘concept drift’ [19]. To address these gaps, the present study aims to identify key predictors of burn mortality and to develop robust predictive models using multiple data mining techniques, including Decision Trees (DT), Random Forest (RF), Gradient Boosted Trees (GBT), Random Tree (RT), and Decision Stump (DS). Unlike traditional prognostic models such as the Abbreviated Burn Severity Index (ABSI) and Baux scores, which rely on a limited number of predefined clinical variables, our approach utilizes advanced tree-based machine learning algorithms capable of capturing complex, non-linear interactions across a broader set of input features [20,21]. In contrast to prior machine learning studies, this research evaluates a diverse range of classifiers using a unique and feature-rich dataset extracted from a specialized burn registry. While the study by Ehteshami et al. utilized a limited set of features from the same hospital [17], our research is based on a larger, more comprehensive dataset derived from the hospital’s burn registry, incorporating a wider range of variables and more recent patient records. While the first part of our approach addresses these challenges by enhancing potential accuracy and generalizability, a second and more critical objective of this study is to conduct a rigorous temporal validation. This step allows us to directly assess our model’s stability and evaluate its robustness against potential concept drift, thereby establishing greater confidence in its applicability for real-world clinical deployment.
Methods
Proposed methodology
For this research, Microsoft Excel 2021 was employed for initial data examination and part of the preprocessing tasks, while RapidMiner software (version 9.10.000) was used for advanced data preprocessing and modeling. The methodology applied was based on the CRISP-DM framework (Cross Industry Standard Process for Data Mining), which is recognized as one of the most robust methodologies for implementing and executing data mining projects [22]. The CRISP-DM framework includes six key phases: System understanding, Data Understanding, Data preprocessing, Modeling, Evaluation, and Deployment. This study covers the first five phases of this framework, from understanding the problem to the final evaluation of the models.
Dataset description
This retrospective cohort study utilized data from the burn registry of the Injuries and Burn Subspecialized Teaching Hospital, affiliated with Isfahan University of Medical Sciences, covering the period from 12/04/2022 to 08/10/2023. Each patient is represented by a single observation (row) in the dataset; no longitudinal data were used. Following an initial assessment, 11 records with a missing final outcome were excluded from the dataset. Thus, the final analytical dataset consisted of a cohort of 651 patients (144 deceased and 507 surviving), for whom 93 predictive features were used for model development, a complete list of which is provided in S1 Table. The dataset was retrospectively accessed for research purposes between 21/06/2024 and 21/09/2024.
Data preprocessing
Prior to applying classification techniques, the following preprocessing steps were performed:
Initial feature engineering and selection.
The initial dataset consisted of 720 features. To create a relevant and manageable feature set for mortality prediction at the time of admission, an initial selection was performed. Features not recorded during the emergency phase or unrelated to the study’s objective were excluded, resulting in 124 features. Subsequently, 24 redundant or similar features were consolidated, and 6 composite features were expanded into 38 new binary variables using a one-hot encoding approach. After removing 16 features with highly imbalanced classes (where over 95% of values belonged to a single class, for instance, a specific comorbidity present in less than 5% of patients), 121 candidate features remained. Subsequently, a further data reduction step was performed to handle highly incomplete features. From the 121 candidate features, 28 were removed because they had more than 85% missing values. This threshold was chosen as a pragmatic balance to remove only the most sparsely populated features, which could introduce noise, while retaining a comprehensive set of potentially valuable clinical variables. This process resulted in a final set of 93 features used for model development.
Handling missing values.
Missing data remained a challenge in the final 93-feature set. An analysis revealed that 79 of the 93 features (approximately 85%) contained at least some missing values; however, the degree of missingness was generally low, with the majority of these affected features having less than 20% missing data. The specific percentage of missing values for each individual feature is detailed in S1 Table.
To handle this, model-based imputation was employed using the ‘Impute Missing Values’ operator within RapidMiner. A Generalized Linear Model (GLM) was used as the learner within this operator. This advanced imputation method was chosen over simpler techniques (e.g., mean imputation) because it is capable of capturing complex relationships between variables, and our sensitivity analysis confirmed it yielded a slight improvement in overall model performance. The GLM was configured with its family and solver parameters set to ‘AUTO’, allowing the software to select the optimal distribution and optimization algorithm for each feature. Key parameters for the GLM also included the use of regularization and data standardization to ensure robust model fitting.
Crucially, to prevent data leakage from the outcome into the feature set, the outcome variable (label) was excluded from the imputation process. This ensures that information about the final patient outcome did not influence the imputation of predictor variables. Furthermore, to prevent overly optimistic performance estimates, the entire imputation process was nested within the model training and evaluation procedure. This ensures that for each evaluation run, the imputation model was built using only information from the training data for that specific run.
Model development and evaluation
The predictive models were developed using the final set of 93 features as independent variables to predict the binary ‘Final outcome’ variable (deceased vs. survived) as the target.
Model Development and Hyperparameter Tuning.
The study focused on tree-based ensemble models due to their strong performance on tabular data, their inherent ability to capture complex non-linear relationships without extensive feature engineering, and their relatively high interpretability compared to ‘black-box’ models. While other model families like logistic regression are valuable, the objective here was a deep evaluation of this specific class of powerful algorithms rather than a broad, shallow comparison across multiple families.
Five tree-based algorithms were implemented in RapidMiner (version 9.10.000): Gradient Boosted Trees (GBT), Decision Tree (DT), Random Forest (RF), Decision Stump (DS), and Random Tree (RT). To select optimal hyperparameters and simultaneously obtain an unbiased estimate of each model’s final performance, a nested validation procedure was implemented. The outer loop utilized the ‘Optimize Parameters (Grid)’ operator to systematically search for the optimal combination of hyperparameters for each algorithm. The parameter grids and search ranges are detailed in S2 Table. To robustly evaluate the performance of each hyperparameter combination within this search, a complete 10-fold cross-validation was executed as the inner loop of the process. This inner validation was configured to use shuffled sampling to partition the data. The performance metric used to guide the optimization process and select the best hyperparameter set was overall Accuracy. The final performance metrics reported in this study represent the performance of the models configured with these optimal hyperparameters, as evaluated by this robust nested procedure. The optimal hyperparameters selected through this process for the primary models are presented in Table 1.
Model performance evaluation.
The performance of the finalized models was rigorously assessed using a 10-fold cross-validation process with stratified sampling. Stratified sampling was chosen to ensure that the proportion of outcome classes (deceased vs. survived) was maintained across all folds, thereby providing a more stable and reliable estimate of performance. For all classification algorithms, the default decision threshold of 0.5 was applied to convert predicted probabilities into binary class labels (Deceased vs. Survived).
Model performance was quantified using several standard metrics: Accuracy, Precision (Positive Predictive Value), Sensitivity (Recall), Specificity, and F1-Score. Additionally, the Area Under the ROC Curve (AUC) was used to assess the overall discriminative ability. To evaluate model calibration —a measure of how well the predicted probabilities match actual outcomes—the Brier Score was also calculated. A lower Brier Score indicates better calibration and more reliable probability estimates, a crucial feature for clinical risk stratification tools.
To determine if the observed differences in performance metrics between models were statistically significant, paired t-tests were conducted on the metric scores obtained from the 10-fold cross-validation. A p-value of less than 0.05 was considered statistically significant. This study adheres to the Transparent Reporting of a multivariable prediction model for Individual Prognosis or Diagnosis (TRIPOD) statement; the completed checklist is provided in S1 Checklist.
Assessment of feature importance.
To identify the most influential predictors from the final feature set, a two-stage feature importance analysis was conducted. First, for an initial exploratory overview, the importance of each feature was assessed using several standard weighting methods within RapidMiner (e.g., Gini Index, Information Gain); the full results of this exploratory analysis are available in the Supporting Information (S3 Table).
Second, for the definitive analysis, model-based feature importance scores were extracted from the final trained models. A detailed visualization and interpretation of these scores, focusing on the best-performing model, is presented in the Results section.
Temporal validation and sensitivity analyses.
To assess model stability—defined here as the consistency of model performance on future data from the same clinical environment—and to investigate the potential for concept drift, a rigorous temporal validation was performed. The dataset was divided into two non-overlapping chronological cohorts based on each patient’s admission date, ensuring that each unique patient belonged to only one cohort and thus preventing data leakage. The training cohort consisted of all patients admitted during the first nine months of the study period (n = 435), while the temporal test cohort comprised all patients from the subsequent nine months (n = 216). It is important to note that this temporal validation assesses the model’s robustness to shifts in the patient population at a single center over time, and is distinct from external validation, which assesses a model’s generalizability across different institutions.
For this temporal analysis, the evaluation focused on the two primary ensemble models: Gradient Boosted Trees (GBT) and Random Forest (RF). Each model was configured using the optimal hyperparameters identified during the nested validation procedure, as presented in Table 1. The finalized models were trained on the entire training cohort, and their performance was subsequently evaluated once on the entire temporal test cohort.
Furthermore, to investigate the robustness of our findings to methodological choices, three sensitivity analyses were conducted on the top-performing models (GBT and RF). The first analysis compared the performance of our primary imputation method (GLM) against a traditional Mean/Fixed Value Imputation. The second analysis assessed the impact of variable format by comparing the performance of models built with Continuous variables against those using categorized variables. For this comparison, the categorical variables were generated by discretizing the original Continuous features based on standard medical thresholds and conventional classification criteria. A third sensitivity analysis was performed to evaluate the model’s reliance on the pre-engineered ABSI score by re-evaluating the models after this feature was excluded from the dataset.
Ethical considerations
The study protocol was reviewed and approved by the Ethics Committee of Isfahan University of Medical Sciences (Approval ID: IR.MUI.NUREMA.REC.1403.048), and all methods were performed in accordance with the relevant guidelines and regulations of the Declaration of Helsinki. This research involved a secondary analysis of fully de-identified registry data. The Ethics Committee of Isfahan University of Medical Sciences explicitly waived the need for individual informed consent due to the retrospective nature of the study and the use of anonymized data, which posed no risk to patient privacy. Therefore, no written or verbal consent was obtained from participants or their legal guardians.
Results
Study population
Of the 662 burn inpatients initially identified, 11 were excluded due to missing final outcome data. The final analysis was performed on 651 patients, of whom 144 (22.1%) died during hospitalization and 507 (77.9%) survived. The flow of patient selection for the study is illustrated in Fig 1. Detailed baseline demographic and clinical features of the study population are provided in S1 Table, while a summary of the feature categories and variable types used for model development is presented in Table 2.
The flowchart illustrates the process of including and excluding patients for the study. A total of 662 burn inpatients were initially identified from the registry, of whom 11 were excluded due to missing final outcome data, resulting in a final analytical cohort of 651 patients.
Main model performance
The performance metrics for the five finalized models, evaluated using 10-fold cross-validation, are presented in Table 3. The GBT model demonstrated the highest performance across most metrics, achieving an accuracy of 93.06% and the best Brier Score of 0.060. The RF model also showed strong performance, with a comparable accuracy of 92.91% and an AUC of 0.961. The receiver operating characteristic (ROC) curves for the primary models are shown in Fig 2, visually illustrating the superior discriminative ability of the GBT and RF models. The aggregated confusion matrices for each primary model, summarizing the cumulative predictions across all cross-validation folds, are presented in Supporting information (S4 Table).
The curves illustrate the diagnostic ability of the Gradient Boosted Trees (GBT), Random Forest (RF), Decision Tree (DT), Decision Stump (DS), and Random Tree (RT) models. The Area Under the Curve (AUC) for the GBT model (0.966) and the RF model (0.961) demonstrates superior discrimination compared to the other models.
A closer examination of the performance metrics for the top-performing GBT model reveals important clinical trade-offs. The model achieved a high specificity of 97.5%, indicating it correctly identified the vast majority of patients who would survive. However, its sensitivity (recall) was 78.1%, meaning it correctly identified approximately four out of five patients who would ultimately die. The Precision (PPV) of 89.7% suggests that when the model predicted a patient would die, that prediction was correct nearly 90% of the time. These class-specific metrics are crucial for understanding the model’s real-world utility beyond overall accuracy.
Statistical comparison of models
To assess whether the observed performance differences between models were statistically significant, paired t-tests were conducted on the metric scores from the 10-fold cross-validation. The analysis revealed no statistically significant difference between the two top-performing models, GBT and RF, across the key metrics of Accuracy (p = 0.896), AUC (p = 0.393), and Brier Score (p = 0.620). However, both GBT and RF demonstrated a statistically significant superiority over the other models (DT, DS, and RT) on most metrics, particularly AUC (p < 0.05 for all comparisons). The full pairwise comparison results for all metrics are provided in the Supporting information (S5–S7 Tables).
Feature importance analysis
As shown in Table 3, the GBT model demonstrated the highest overall performance. Therefore, a deeper analysis of its feature importance was conducted to understand the key drivers of its predictions. Notably, the Abbreviated Burn Severity Index (ABSI) Score, the percentage of third-degree burns, and Total Burn Surface Area (TBSA) were identified as the most critical predictors. The top 20 most influential features, as determined by the GBT model’s internal weights, are visualized in Fig 3.
The horizontal bar chart displays the relative importance of the top predictors. The Abbreviated Burn Severity Index (ABSI) Score, percentage of third-degree burns, and Total Burn Surface Area (TBSA) were identified as the most critical features driving the model’s predictions.
Temporal validation
To assess model stability, the top-performing models were subjected to a temporal validation pipeline, as illustrated in Fig 4. As shown in Table 4 and visualized in Fig 5, both the GBT and RF models demonstrated strong temporal stability, retaining high overall accuracy and discriminative ability. The GBT model’s AUC, for instance, saw only a minor drop from 0.966 to 0.948. The most notable change was a decrease in sensitivity, particularly for the GBT model (from 78.1% to 68.3%), while specificity remained excellent.
The dataset was split chronologically. To prevent data leakage, the final classification model was trained on the training cohort (first 9 months) using the fixed optimal hyperparameters identified during the development phase. For the temporal test cohort (subsequent 9 months), preprocessing steps—including independent GLM imputation to address concept drift—were performed, after which the frozen classifier was applied to predict outcomes without retraining.
The blue bars represent the mean performance of the Main Model (derived from 10-fold cross-validation), while green bars show the performance during the Temporal Validation phase. The chart highlights the model’s stability in terms of Accuracy, AUC, and Specificity, while illustrating the impact of temporal concept drift through a notable reduction in Sensitivity. (Numerical details with confidence intervals are provided in Table 4).
To investigate the potential cause of this shift in performance, the baseline characteristics of the patients in the training and temporal test cohorts were compared. The results of this statistical comparison are presented in Table 5. Notably, a statistically significant difference was observed in the mean percentage of third-degree burns between the two cohorts (p < 0.001), while other key characteristics such as age and TBSA remained stable.
Sensitivity analyses
The results of the three sensitivity analyses, presented in Table 6, confirmed the model’s general robustness to different methodological choices. First, a comparison of imputation methods showed that while the baseline GBT model using GLM imputation had slightly higher performance point estimates than the model using Mean Imputation, this difference was not statistically significant (p = 0.404 for AUC). Second, the analysis of variable format revealed a similar pattern, where using continuous variables did not provide a statistically significant advantage over their categorized counterparts (p = 0.638 for GBT AUC); this suggests that the discretization process may lead to some loss of predictive information, although the overall high performance is maintained. Finally, the analysis evaluating the impact of the ABSI score confirmed that its exclusion also did not result in a statistically significant drop in the GBT model’s performance (p = 0.53 for GBT AUC). These findings collectively reinforce the model’s robustness (see S5-S8 Tables for full pairwise statistical comparisons).
Discussion
In this study, we developed and validated several machine learning models to predict inpatient mortality in burn patients using a comprehensive registry dataset. Our primary analysis identified the GBT model as the most effective algorithm. While the GBT and RF models showed comparable performance in terms of accuracy, GBT’s superiority was evident in its higher AUC (0.966) and lower Brier Score (0.060). The lower Brier score, in particular, suggests that GBT provides more reliable and well-calibrated probability estimates, a crucial feature for clinical decision support tools intended for risk stratification [23]. Interestingly, the statistical comparison did not show a significant difference for the Brier Score between the top models (p = 0.620), suggesting they both reached a high level of calibration.
A key finding of our study was the notable temporal stability of the GBT model. Unlike many clinical prediction models that can suffer from significant performance degradation over time due to concept drift [19], our model retained high overall accuracy and discriminative ability when tested on a future, unseen patient cohort from a subsequent nine-month period. This robustness suggests that the model learned generalizable, underlying patterns from the initial training data rather than spurious, time-sensitive correlations.
However, the temporal validation also revealed a critical clinical trade-off. The primary impact of the minor concept drift present in the data was a notable decrease in sensitivity (from 78.1% to 68.3%). This implies that while the model remains highly reliable in identifying patients who will survive (stable high specificity), its ability to detect high-risk patients who will ultimately die may diminish over time. This finding was likely driven by the observed shift in the patient population, specifically the statistically significant decrease in the percentage of third-degree burns in the temporal test cohort (Table 5).
The high performance of our GBT model in cross-validation (AUC = 0.966) is consistent with, and in some cases exceeds, the performance reported in other studies that have applied machine learning to predict burn mortality. Our findings align with those of Yazıcı et al. [16], who also identified tree-based ensembles as powerful predictors in this domain. However, a primary contribution that distinguishes our study is the rigorous temporal validation. While many studies demonstrate high predictive accuracy, the assessment of model stability over time is often overlooked. By confirming our model’s robustness, our work extends beyond simply applying machine learning, providing crucial evidence for the potential long-term reliability of such models in a real-world clinical setting.
Our feature importance analysis (Fig 3) confirmed that the model’s predictions were driven by clinically established risk factors, with the Abbreviated Burn Severity Index (ABSI), percentage of third-degree burns, and Total Burn Surface Area (TBSA) ranking as the most influential predictors [16,20,21]. This alignment with existing clinical knowledge increases confidence in the model’s validity and suggests that it did not identify any entirely novel or unexpected predictors. Furthermore, the model’s robustness was confirmed by a series of sensitivity analyses: neither the choice of imputation method nor the variable format resulted in a statistically significant difference in performance, and the model’s predictive power remained strong even after the pre-engineered ABSI score was removed (see Table 6).
Clinical implications
A highly accurate and stable model like the GBT could be a valuable asset for early risk stratification in burn units. However, our findings on decreased sensitivity carry a critical warning for clinical implementation. A model that becomes less sensitive over time could lead to false reassurances and missed opportunities for early, aggressive intervention in patients who are incorrectly flagged as low-risk. This underscores that even for robust models, periodic performance monitoring and recalibration are essential components of a responsible clinical deployment strategy to ensure sustained patient safety.
Strengths and limitations
This study has several strengths, including the use of a comprehensive clinical registry, a rigorous nested validation procedure for hyperparameter tuning, and the inclusion of a temporal validation analysis. Nevertheless, several limitations should be acknowledged. First, as the data were collected from a single medical center, the external generalizability of our findings to other populations may be limited. However, this single-center design also provided a controlled environment to specifically test for temporal stability, which strengthens the internal validity of this key analysis. It is critical to note that our temporal validation is not a substitute for external validation. Second, the sample size (n = 651), while substantial for a single-center study, may still be insufficient for developing models that are robustly generalizable at a national or international level. Additionally, given the ratio of features to patient records, the risk of overfitting, while mitigated by cross-validation, cannot be entirely dismissed. Third, our analysis was intentionally focused on tree-based models to allow for a deep evaluation of this specific class of algorithms; consequently, other model families such as logistic regression or more complex approaches like deep neural networks were not explored. While our model-based feature importance provided robust insights, we acknowledge that more advanced interpretability techniques like SHAP analysis, which can offer deeper explanations of individual predictions, could not be implemented due to technical limitations of our analytical software. These technical constraints also influenced the temporal validation pipeline; specifically, the GLM imputation model could not be directly serialized for the test set. Consequently, imputation for the temporal cohort was performed independently, an approach we consider appropriate given the observed concept drift to respect the specific distributional characteristics of the new time period. Furthermore, regarding statistical evaluation, we acknowledge that cross-validation folds are not strictly independent; thus, we have emphasized the confidence intervals from the independent temporal validation set and relied on Brier Scores for calibration assessment in the absence of retained probability logs. Finally, as a retrospective study based on registry data, there is inherent potential for unmeasured confounding variables and data quality issues, although extensive preprocessing and robust imputation methods were employed to mitigate this.
Conclusion
This study demonstrated that tree-based machine learning algorithms, particularly the GBT model, are powerful tools for predicting inpatient mortality in burn patients with high accuracy. The model’s validity was reinforced by the identification of clinically established risk factors—such as the ABSI score, TBSA, and the presence of third-degree burns—as the most critical predictors.
Crucially, our temporal validation analysis revealed that the model exhibited strong temporal stability, maintaining its high performance on a future patient cohort. However, we also identified a clinically significant decrease in sensitivity over time, highlighting that even robust models are subject to performance shifts. This central finding underscores the vital importance of implementing a robust governance framework, including continuous monitoring and periodic recalibration, to ensure that any predictive model remains safe and effective throughout its lifecycle in a clinical setting.
Future research should focus on validating these models on larger, multi-center datasets to enhance generalizability. Furthermore, exploring more advanced techniques, such as deep learning, could provide pathways to creating even more robust and adaptive predictive tools for burn care.
Supporting information
S1 Table. Features used for model development.
Detailed descriptive statistics and characteristics of all variables used in the study.
https://doi.org/10.1371/journal.pone.0338564.s001
(DOCX)
S2 Table. Hyperparameter tuning search space and optimal values.
Details of the grid search and selected parameters for each model.
https://doi.org/10.1371/journal.pone.0338564.s002
(DOCX)
S3 Table. Exploratory analysis of feature importance using multiple weighting criteria.
Ranking of features using multiple weighting criteria.
https://doi.org/10.1371/journal.pone.0338564.s003
(DOCX)
S4 Table. Confusion matrix of the final models.
Aggregated confusion matrices for GBT, RF, DT, DS, and RT models.
https://doi.org/10.1371/journal.pone.0338564.s004
(DOCX)
S5 Table. Full pairwise T-test p-values for model accuracy.
Statistical comparison of accuracy between models.
https://doi.org/10.1371/journal.pone.0338564.s005
(DOCX)
S6 Table. Full pairwise T-test p-values for the AUC metric.
Statistical comparison of AUC between models.
https://doi.org/10.1371/journal.pone.0338564.s006
(DOCX)
S7 Table. Full pairwise T-test p-values for the Brier score.
Statistical comparison of Brier scores between models.
https://doi.org/10.1371/journal.pone.0338564.s007
(DOCX)
S8 Table. Statistical comparison (p-values) of model performance with and without the ABSI feature.
https://doi.org/10.1371/journal.pone.0338564.s008
(DOCX)
S1 Checklist. TRIPOD checklist.
Completed checklist demonstrating adherence to reporting guidelines for prediction model development and validation.
https://doi.org/10.1371/journal.pone.0338564.s009
(DOCX)
Acknowledgments
The authors would like to thank Isfahan University of Medical Sciences and all contributors to the burn inpatients registry project at the Injuries and Burn Subspecialized Teaching Hospital for their valuable support and collaboration.
References
- 1. Peck MD. Epidemiology of burns throughout the world. Part I: distribution and risk factors. Burns. 2011;37(7):1087–100. pmid:21802856
- 2.
WHO. Burns; 2023. Available from: https://www.who.int/news-room/fact-sheets/detail/burns
- 3. Rezaee R, Alimohamadzadeh K, Hossini S-M. Epiemiologic features and hospitalization cost of burn injuries in iran based on national burn registry; a cross-sectional study. Arch Acad Emerg Med. 2019;7(1):e65. pmid:31875219
- 4. Edelman LS. Social and economic factors associated with the risk of burn injury. Burns. 2007;33(8):958–65. pmid:17869003
- 5. Jeschke MG, van Baar ME, Choudhry MA, Chung KK, Gibran NS, Logsetty S. Burn injury. Nat Rev Dis Primers. 2020;6(1):11. pmid:32054846
- 6. Soltany A, Hraib M, Apelian S, Mallouhy A, Kounsselie E, Diab S. Postburn abdominal pain of gastrointestinal origin: a scoping review. Plast Reconstr Surg Glob Open. 2024;12(11):e6300. pmid:39539412
- 7. Raghupathi W, Raghupathi V. Big data analytics in healthcare: promise and potential. Health Inf Sci Syst. 2014;2:3. pmid:25825667
- 8.
TMD. Millions of gigabytes of health-related data; 2015. Available from: https://www.todaysmedicaldevelopments.com/article/tmd0515-ibm-health-care-cloud/
- 9. Kanevsky J, Corban J, Gaster R, Kanevsky A, Lin S, Gilardino M. Big data and machine learning in plastic surgery: a new frontier in surgical innovation. Plast Reconstr Surg. 2016;137(5):890e–7e. pmid:27119951
- 10. Patil BM, Joshi RC, Toshniwal D, Biradar S. A new approach: role of data mining in prediction of survival of burn patients. J Med Syst. 2011;35(6):1531–42. pmid:20703764
- 11. Han J, Kamber M, Pei J. Data mining: concepts and techniques. 3rd ed. Waltham: Morgan Kaufmann; 2011.
- 12. Wu W-T, Li Y-J, Feng A-Z, Li L, Huang T, Xu A-D, et al. Data mining in clinical big data: the frequently used databases, steps, and methodological models. Mil Med Res. 2021;8(1):44. pmid:34380547
- 13. Ahmadi-Jouybari T, Najafi-Ghobadi S, Karami-Matin R, Najafian-Ghobadi S, Najafi-Ghobadi K. Investigating factors affecting the interval between a burn and the start of treatment using data mining methods and logistic regression. BMC Med Res Methodol. 2021;21(1):71. pmid:33853547
- 14. Yang S-Y, Huang C-J, Yen C-I, Kao Y-C, Hsiao Y-C, Yang J-Y, et al. Machine learning approach for predicting inhalation injury in patients with burns. Burns. 2023;49(7):1592–601. pmid:37055284
- 15. Aydın E, Khanmammadova N, Aslanyürek B, Urgancı N, Usta M, Parlak A, et al. A simple machine learning approach for preoperative diagnosis of esophageal burns after caustic substance ingestion in children. Pediatr Surg Int. 2023;40(1):20. pmid:38092997
- 16. Yazıcı H, Ugurlu O, Aygul Y, Yildirim M, Uçar AD. The effect of well-known burn-related features on machine learning algorithms in burn patients’ mortality prediction. Ulus Travma Acil Cerrahi Derg. 2023;29(10):1130–7. pmid:37791433
- 17. Ehteshami A, Karimi S, Shahmoradi Z, Sattari M. Predicting the survival or death of burn patients using data mining techniques. IRCMJ. 2024;26(1).
- 18. Nabovati E, Azizi A, Abbasi E, Vakili-Arki H, Zarei J, Razavi A. Using data mining to predict outcome in burn patients: a comparison between several algorithms. HIM. 2014:789–99. Available from: https://him.mui.ac.ir/article_11239.html
- 19. Schinkel M, Boerman AW, Paranjape K, Wiersinga WJ, Nanayakkara PWB. Detecting changes in the performance of a clinical machine learning tool over time. EBioMedicine. 2023;97:104823. pmid:37793210
- 20. Forster NA, Zingg M, Haile SR, Künzi W, Giovanoli P, Guggenheim M. 30 years later--does the ABSI need revision? Burns. 2011;37(6):958–63. pmid:21493008
- 21. Heng JS, Clancy O, Atkins J, Leon-Villapalos J, Williams AJ, Keays R, et al. Revised Baux Score and updated Charlson comorbidity index are independently associated with mortality in burns intensive care patients. Burns. 2015;41(7):1420–7. pmid:26187055
- 22. Mazaheri S, Ashoori M, Bechari Z. A model to predict heart disease treatment using data mining. Payavard Salamat. 2017;11(3):287–96. Available from: http://payavard.tums.ac.ir/article-1-6285-en.html
- 23. Steyerberg EW. Clinical prediction models: a practical approach to development, validation, and updating. Springer International Publishing; 2019.