Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach
Abstract Ureteropelvic junction obstruction (UPJO) is a common pediatric condition often treated with pyeloplasty. Despite the surgical intervention, postoperative urinary tract infections (UTIs) occur in over 30% of cases within six months, adversely affecting recovery and increasing both clinical...
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Nature Portfolio
2025-01-01
|
| Series: | Scientific Reports |
| Subjects: | |
| Online Access: | https://doi.org/10.1038/s41598-024-82282-1 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832585909712715776 |
|---|---|
| author | Hongyang Wang Junpeng Ding Shuochen Wang Long Li Jinqiu Song Dongsheng Bai |
| author_facet | Hongyang Wang Junpeng Ding Shuochen Wang Long Li Jinqiu Song Dongsheng Bai |
| author_sort | Hongyang Wang |
| collection | DOAJ |
| description | Abstract Ureteropelvic junction obstruction (UPJO) is a common pediatric condition often treated with pyeloplasty. Despite the surgical intervention, postoperative urinary tract infections (UTIs) occur in over 30% of cases within six months, adversely affecting recovery and increasing both clinical and economic burdens. Current prediction methods for postoperative UTIs rely on empirical judgment and limited clinical parameters, underscoring the need for a robust, multifactorial predictive model. We retrospectively analyzed data from 764 pediatric patients who underwent unilateral pyeloplasty at the Children’s Hospital affiliated with the Capital Institute of Pediatrics between January 2012 and January 2023. A total of 25 clinical features were extracted, including patient demographics, medical history, surgical details, and various postoperative indicators. Feature engineering was initially performed, followed by a comparative analysis of five machine learning algorithms (Logistic Regression, SVM, Random Forest, XGBoost, and LightGBM) and the deep learning TabNet model. This comparison highlighted the respective strengths and limitations of traditional machine learning versus deep learning approaches. Building on these findings, we developed an ensemble learning model, meta-learner, that effectively integrates both methodologies, and utilized SHAP(Shapley Additive Explanation, SHAP) to complete the visualization of the integrated black-box model. Among the 764 pediatric pyeloplasty cases analyzed, 265 (34.7%) developed postoperative UTIs, predominantly within the first three months. Early UTIs significantly increased the likelihood of re-obstruction (P < 0.01), underscoring the critical impact of infection on surgical outcomes. In evaluating the performance of six algorithms, TabNet outperformed traditional models, with the order from lowest to highest as follows: Logistic Regression, SVM, Random Forest, XGBoost, LightGBM, and TabNet. Feature engineering markedly improved the predictive accuracy of traditional models, as evidenced by the enhanced performance of LightGBM (Accuracy: 0.71, AUC: 0.78 post-engineering). The proposed ensemble approach, combining LightGBM and TabNet with a Logistic Regression meta-learner, achieved superior predictive accuracy (Accuracy: 0.80, AUC: 0.80) while reducing dependence on feature engineering. SHAP analysis further revealed eGFR and ALB as significant predictors of UTIs post-pyeloplasty, providing new clinical insights into risk factors. In summary, we have introduced the first ensemble prediction model, incorporating both machine learning and deep learning (meta-learner), to predict urinary tract infections following pediatric pyeloplasty. This ensemble approach mitigates the dependency of machine learning models on feature engineering while addressing the issue of overfitting in deep learning-based models like TabNet, particularly in the context of small medical datasets. By improving prediction accuracy, this model supports proactive interventions, reduces postoperative infections and re-obstruction rates, enhances pyeloplasty outcomes, and alleviates health and economic burdens. Level of evidence IV Case series with no comparison group. |
| format | Article |
| id | doaj-art-518a60a2de384a8d988c314cf81166c3 |
| institution | Kabale University |
| issn | 2045-2322 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Nature Portfolio |
| record_format | Article |
| series | Scientific Reports |
| spelling | doaj-art-518a60a2de384a8d988c314cf81166c32025-01-26T12:24:00ZengNature PortfolioScientific Reports2045-23222025-01-0115112110.1038/s41598-024-82282-1Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approachHongyang Wang0Junpeng Ding1Shuochen Wang2Long Li3Jinqiu Song4Dongsheng Bai5Department of Urology, Capital Institute of PediatricsSchool of Computer Science, Beijing University of Posts and TelecommunicationsSchool of Mathematics Sciences, Capital Normal UniversityDepartment of Urology, Capital Institute of PediatricsDepartment of Urology, Capital Institute of PediatricsDepartment of Urology, Capital Institute of PediatricsAbstract Ureteropelvic junction obstruction (UPJO) is a common pediatric condition often treated with pyeloplasty. Despite the surgical intervention, postoperative urinary tract infections (UTIs) occur in over 30% of cases within six months, adversely affecting recovery and increasing both clinical and economic burdens. Current prediction methods for postoperative UTIs rely on empirical judgment and limited clinical parameters, underscoring the need for a robust, multifactorial predictive model. We retrospectively analyzed data from 764 pediatric patients who underwent unilateral pyeloplasty at the Children’s Hospital affiliated with the Capital Institute of Pediatrics between January 2012 and January 2023. A total of 25 clinical features were extracted, including patient demographics, medical history, surgical details, and various postoperative indicators. Feature engineering was initially performed, followed by a comparative analysis of five machine learning algorithms (Logistic Regression, SVM, Random Forest, XGBoost, and LightGBM) and the deep learning TabNet model. This comparison highlighted the respective strengths and limitations of traditional machine learning versus deep learning approaches. Building on these findings, we developed an ensemble learning model, meta-learner, that effectively integrates both methodologies, and utilized SHAP(Shapley Additive Explanation, SHAP) to complete the visualization of the integrated black-box model. Among the 764 pediatric pyeloplasty cases analyzed, 265 (34.7%) developed postoperative UTIs, predominantly within the first three months. Early UTIs significantly increased the likelihood of re-obstruction (P < 0.01), underscoring the critical impact of infection on surgical outcomes. In evaluating the performance of six algorithms, TabNet outperformed traditional models, with the order from lowest to highest as follows: Logistic Regression, SVM, Random Forest, XGBoost, LightGBM, and TabNet. Feature engineering markedly improved the predictive accuracy of traditional models, as evidenced by the enhanced performance of LightGBM (Accuracy: 0.71, AUC: 0.78 post-engineering). The proposed ensemble approach, combining LightGBM and TabNet with a Logistic Regression meta-learner, achieved superior predictive accuracy (Accuracy: 0.80, AUC: 0.80) while reducing dependence on feature engineering. SHAP analysis further revealed eGFR and ALB as significant predictors of UTIs post-pyeloplasty, providing new clinical insights into risk factors. In summary, we have introduced the first ensemble prediction model, incorporating both machine learning and deep learning (meta-learner), to predict urinary tract infections following pediatric pyeloplasty. This ensemble approach mitigates the dependency of machine learning models on feature engineering while addressing the issue of overfitting in deep learning-based models like TabNet, particularly in the context of small medical datasets. By improving prediction accuracy, this model supports proactive interventions, reduces postoperative infections and re-obstruction rates, enhances pyeloplasty outcomes, and alleviates health and economic burdens. Level of evidence IV Case series with no comparison group.https://doi.org/10.1038/s41598-024-82282-1Urinary tract infectionUnilateral pyeloplastyMachine learningDeep learningTabNetEnsemble model |
| spellingShingle | Hongyang Wang Junpeng Ding Shuochen Wang Long Li Jinqiu Song Dongsheng Bai Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach Scientific Reports Urinary tract infection Unilateral pyeloplasty Machine learning Deep learning TabNet Ensemble model |
| title | Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach |
| title_full | Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach |
| title_fullStr | Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach |
| title_full_unstemmed | Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach |
| title_short | Enhancing predictive accuracy for urinary tract infections post-pediatric pyeloplasty with explainable AI: an ensemble TabNet approach |
| title_sort | enhancing predictive accuracy for urinary tract infections post pediatric pyeloplasty with explainable ai an ensemble tabnet approach |
| topic | Urinary tract infection Unilateral pyeloplasty Machine learning Deep learning TabNet Ensemble model |
| url | https://doi.org/10.1038/s41598-024-82282-1 |
| work_keys_str_mv | AT hongyangwang enhancingpredictiveaccuracyforurinarytractinfectionspostpediatricpyeloplastywithexplainableaianensembletabnetapproach AT junpengding enhancingpredictiveaccuracyforurinarytractinfectionspostpediatricpyeloplastywithexplainableaianensembletabnetapproach AT shuochenwang enhancingpredictiveaccuracyforurinarytractinfectionspostpediatricpyeloplastywithexplainableaianensembletabnetapproach AT longli enhancingpredictiveaccuracyforurinarytractinfectionspostpediatricpyeloplastywithexplainableaianensembletabnetapproach AT jinqiusong enhancingpredictiveaccuracyforurinarytractinfectionspostpediatricpyeloplastywithexplainableaianensembletabnetapproach AT dongshengbai enhancingpredictiveaccuracyforurinarytractinfectionspostpediatricpyeloplastywithexplainableaianensembletabnetapproach |