Explainable AI-Integrated Stacked Machine-Learning Model for Detection of Infectious Conditions Utilizing Vital Signs and Hematological Biomarkers
Abstract
1. Introduction
2. Literature Review
- Descriptive statistical analysis of the dataset was carried out to examine distributions and correlations among vital signs and biomarkers.
- A hybrid feature selection pipeline was implemented, combining correlation filtering, mutual information, and tree-based feature selection.
- Of great importance was XAI validation (SHAP, permutation sensitivity, ICE plots), used to identify the most significant predictors.
- Machine-learning models, including Random Forest, Logistic Regression, Decision Tree, KNN, Adaboost, Catboost, Light GBM, and XGBoost, along with ensemble techniques such as stacking, were employed to predict ICU admission risk.
- The performance of these classifiers was further compared with explainable AI techniques, including SHAP, LIME, and permutation-based sensitivity analysis, to ensure interpretability of predictions.
- A clinical perspective discussion was provided to highlight the most important biomarkers and vital signs, such as lactate, respiratory rate, and systolic and diastolic blood pressure, for the detection of infectious diseases and early ICU risk stratification.
- In this study, ICU admission is considered a clinically relevant indicator of infection-associated severity.
3. Materials and Methods
3.1. Dataset Description
3.2. Data Pre-Processing
4. Results
5. Discussion
6. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Acknowledgments
Conflicts of Interest
References
- Chaw, J.K.; Chaw, S.H.; Quah, C.H.; Sahrani, S.; Ang, M.C.; Zhao, Y.; Ting, T.T. A predictive analytics model using machine learning algorithms to estimate the risk of shock development among dengue patients. Healthc. Anal. 2024, 5, 100290. [Google Scholar] [CrossRef] [Scilit]
- Wong, F.; de la Fuente-Nunez, C.; Collins, J.J. Leveraging artificial intelligence in the fight against infectious diseases. Science 2023, 381, 164–170. [Google Scholar] [CrossRef] [Scilit]
- Al Meslamani, A.Z.; Sobrino, I.; de la Fuente, J. Machine learning in infectious diseases: Potential applications and limitations. Ann. Med. 2024, 56, 2362869. [Google Scholar] [CrossRef] [Scilit]
- Bilsen, M.P.; Treep, M.M.; Aantjes, M.J.; van Andel, E.; Stalenhoef, J.E.; van Nieuwkoop, C.; Leyten, E.M.; Delfos, N.M.; van Uhm, J.I.; Sijbom, M.; et al. Diagnostic accuracy of urine biomarkers for urinary tract infection in older women: A case-control study. Clin. Microbiol. Infect. 2024, 30, 216–222. [Google Scholar] [CrossRef] [Scilit]
- Mancuso, G.; Midiri, A.; Gerace, E.; Marra, M.; Zummo, S.; Biondo, C. Urinary tract infections: The current scenario and future prospects. Pathogens 2023, 12, 623. [Google Scholar] [CrossRef] [Scilit]
- Czajkowski, K.; Broś-Konopielko, M.; Teliga-Czajkowska, J. Urinary tract infection in women. Prz. Menopauzalny 2021, 20, 40–47. [Google Scholar] [CrossRef] [Scilit]
- Evans, L.; Rhodes, A.; Alhazzani, W.; Antonelli, M.; Coopersmith, C.M.; French, C.; Machado, F.R.; Mcintyre, L.; Ostermann, M.; Prescott, H.C.; et al. Surviving sepsis campaign: International guidelines for management of sepsis and septic shock. Intensive Care Med. 2021, 47, 1181–1247. [Google Scholar] [CrossRef] [Scilit]
- Van der Poll, T.; Shankar-Hari, M.; Wiersinga, W.J. The immunology of sepsis. Immunity 2021, 54, 2450–2464. [Google Scholar] [CrossRef] [Scilit]
- Schlapbach, L.J.; Watson, R.S.; Sorce, L.R.; Argent, A.C.; Menon, K.; Hall, M.W.; Akech, S.; Albers, D.J.; Alpern, E.R.; Balamuth, F.; et al. International consensus criteria for pediatric sepsis and septic shock. JAMA 2024, 331, 665–674. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Torres, J.S.S.; Tamayo-Giraldo, F.J.; Bejarano-Zuleta, A.; Nati-Castillo, H.A.; Quintero, D.A.; Ospina-Mejía, M.J.; Salazar-Santoliva, C.; Suárez-Sangucho, I.; Ortiz-Prado, E.; Izquierdo-Condoy, J.S. Sepsis and post-sepsis syndrome: A multisystem challenge requiring comprehensive care and management—A review. Front. Med. 2025, 12, 1560737. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Giamarellos-Bourboulis, E.J.; Aschenbrenner, A.C.; Bauer, M.; Bock, C.; Calandra, T.; Gat-Viks, I.; Kyriazopoulou, E.; Lupse, M.; Monneret, G.; Pickkers, P.; et al. The pathophysiology of sepsis and precision-medicine-based immunotherapy. Nat. Immunol. 2024, 25, 19–28. [Google Scholar] [CrossRef] [Scilit]
- Menon, K.; Schlapbach, L.J.; Akech, S.; Argent, A.; Biban, P.; Carrol, E.D.; Chiotos, K.; Chisti, M.J.; Evans, I.V.R.; Inwald, D.P.; et al. Criteria for pediatric sepsis—A systematic review and meta-analysis by the pediatric sepsis definition taskforce. Crit. Care Med. 2022, 50, 21–36. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Gilholm, P.; Raman, S.; Irwin, A.; Lister, P.; Harley, A.; Schlapbach, L.J.; Gibbons, K.S. Identification of distinct clinical profiles of sepsis risk in paediatric emergency department patients using Bayesian profile regression. BMJ Paediatr. Open 2025, 9, e003100. [Google Scholar] [CrossRef] [Scilit]
- Baker, R.E.; Mahmud, A.S.; Miller, I.F.; Rajeev, M.; Rasambainarivo, F.; Rice, B.L.; Takahashi, S.; Tatem, A.J.; Wagner, C.E.; Wang, L.-F.; et al. Infectious disease in an era of global change. Nat. Rev. Microbiol. 2022, 20, 193–205. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Ugwu, O.P.C.; Alum, E.U.; Ugwu, J.N.; Eze, V.H.U.; Ugwu, C.N.; Ogenyi, F.C.; Okon, M.B. Harnessing technology for infectious disease response in conflict zones: Challenges, innovations, and policy implications. Medicine 2024, 103, e38834. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Colak, M.; Kilinc, M.A.; Güven, R.; Kutlu, N.O. Procalcitonin and blood lactate level as predictive biomarkers in pediatric multiple trauma patients’ pediatric intensive care outcomes: A retrospective observational study. Medicine 2023, 102, e36289. [Google Scholar] [CrossRef] [Scilit]
- Wald, E.R.; Schmit, K.M.; Gusland, D.Y. A pediatric infectious disease perspective on COVID-19. Clin. Infect. Dis. 2021, 72, 1660–1666. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Alsabri, M.; Siddiq, A.; Aderinto, N.; Ishola, I.V.; Shahid, M.A.; Kaul, A.; Taha, N.A.; Ahmad, A.H.; Gamboa, L.L. Infectious Disease Management in Pediatric Emergency Departments in Low-and Middle-Income Countries: A Review of Diagnostic Tools, Treatment Protocols, and Preventive Measures. Glob. Pediatr. Health 2024, 11, 2333794X241304663. [Google Scholar] [CrossRef] [Scilit]
- Fauziah, N.; Aviani, J.K.; Agrianfanny, Y.N.; Fatimah, S.N. Intestinal parasitic infection and nutritional status in children under five years old: A systematic review. Trop. Med. Infect. Dis. 2022, 7, 371. [Google Scholar] [CrossRef] [Scilit]
- Clemente, C.; Fuentes Ferrer, M.E.; Ortega Heredia, D.; Julián Jiménez, A.; Martín-Sánchez, F.J.; González Del Castillo, J. Usefulness of combining inflammatory biomarkers and clinical scales in an emergency department to stratify risk in patients with infections. Emergencias 2024, 36, 9–16. [Google Scholar] [CrossRef] [Scilit]
- Brunker, L.B.; Boncyk, C.S.; Rengel, K.F.; Hughes, C.G. Elderly patients and management in intensive care units (ICU): Clinical challenges. Clin. Interv. Aging 2023, 18, 93–112. [Google Scholar] [CrossRef] [Scilit]
- Bodaghi, A.; Fattahi, N.; Ramazani, A. Biomarkers: Promising and valuable tools towards diagnosis, prognosis and treatment of Covid-19 and other diseases. Heliyon 2023, 9, e13323. [Google Scholar] [CrossRef] [Scilit]
- Scavello, F.; Brunetta, E.; Mapelli, S.N.; Nappi, E.; Martin, I.D.G.; Sironi, M.; Leone, R.; Solane, S.; Angelotti, G.; Supino, D.; et al. The long Pentraxin PTX3 serves as an early predictive biomarker of co-infections in COVID-19. eBioMedicine 2024, 105, 105213. [Google Scholar] [PubMed]
- Zaman, N.I.D.; Hau, Y.W.; Leong, M.C.; Al-Ashwal, R.H.A. A review on the significance of body temperature interpretation for early infectious disease diagnosis. Artif. Intell. Rev. 2023, 56, 15449–15494. [Google Scholar] [CrossRef] [Scilit]
- Awotunde, J.B.; Folorunso, S.O.; Bhoi, A.K.; Adebayo, P.O.; Ijaz, M.F. Disease diagnosis system for IoT-based wearable body sensors with machine learning algorithm. In Hybrid Artificial Intelligence and IoT in Healthcare; Springer: Singapore, 2021; pp. 201–222. [Google Scholar]
- Goergen, C.J.; Tweardy, M.J.; Steinhubl, S.R.; Wegerich, S.W.; Singh, K.; Mieloszyk, R.J.; Dunn, J. Detection and monitoring of viral infections via wearable devices and biometric data. Annu. Rev. Biomed. Eng. 2022, 24, 1–27. [Google Scholar] [CrossRef] [Scilit]
- Ahmad, A.; Imran, M.; Ahsan, H. Biomarkers as biomedical bioindicators: Approaches and techniques for the detection, analysis, and validation of novel biomarkers of diseases. Pharmaceutics 2023, 15, 1630. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Marques, L.; Costa, B.; Pereira, M.; Silva, A.; Santos, J.; Saldanha, L.; Silva, I.; Magalhães, P.; Schmidt, S.; Vale, N. Advancing precision medicine: A review of innovative in silico approaches for drug development, clinical pharmacology and personalized healthcare. Pharmaceutics 2024, 16, 332. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Tong, L.; Shi, W.; Isgut, M.; Zhong, Y.; Lais, P.; Gloster, L.; Sun, J.; Swain, A.; Giuste, F.; Wang, M.D. Integrating multi-omics data with EHR for precision medicine using advanced artificial intelligence. IEEE Rev. Biomed. Eng. 2023, 17, 80–97. [Google Scholar] [CrossRef] [Scilit]
- Upadhyaya, D.P.; Tarabichi, Y.; Prantzalos, K.; Ayub, S.; Kaelber, D.C.; Sahoo, S.S. Machine learning interpretability methods to characterize the importance of hematologic biomarkers in prognosticating patients with suspected infection. Comput. Biol. Med. 2024, 183, 109251. [Google Scholar] [CrossRef] [Scilit]
- Konjety, P.; Chakole, V.G.; Konjety, P., Jr.; Pathade, A.G. Beyond the Horizon: A Comprehensive Review of Contemporary Strategies in Sepsis Management Encompassing Predictors, Diagnostic Tools, and Therapeutic Advances. Cureus 2024, 16, e64249. [Google Scholar] [CrossRef] [Scilit]
- Zhou, W.; Wang, Y.; Gu, X.; Feng, Z.P.; Lee, K.; Peng, Y.; Barszczyk, A. Importance of general adiposity, visceral adiposity and vital signs in predicting blood biomarkers using machine learning. Int. J. Clin. Pract. 2021, 75, e13664. [Google Scholar] [CrossRef] [Scilit]
- Mayne, E.S.; George, J.A.; Louw, S. Assessing biomarkers in viral infection. In Application of Omic Techniques to Identify New Biomarkers and Drug Targets for COVID-19; Springer: Cham, Switzerland, 2023; pp. 159–173. [Google Scholar]
- Fazal, M. C-reactive protein a promising biomarker of COVID-19 severity. Korean J. Clin. Lab. Sci. 2021, 53, 201–207. [Google Scholar] [CrossRef] [Scilit]
- Jazayeri, A.; Yang, C.C.; Capan, M. Frequent temporal patterns of physiological and biological biomarkers and their evolution in sepsis. Artif. Intell. Med. 2023, 143, 102576. [Google Scholar] [CrossRef] [Scilit]
- Zhang, B.; Shi, H.; Wang, H. Machine learning and AI in cancer prognosis, prediction, and treatment selection: A critical approach. J. Multidiscip. Healthc. 2023, 16, 1779–1791. [Google Scholar] [CrossRef] [Scilit]
- Gu, Q.; Wei, J.; Yoon, C.H.; Yuan, K.; Jones, N.; Brent, A.; Llewelyn, M.; Peto, T.E.; Pouwels, K.B.; Eyre, D.W.; et al. Distinct patterns of vital sign and inflammatory marker responses in adults with suspected bloodstream infection. J. Infect. 2024, 88, 106156. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Uffen, J.W.; Oosterheert, J.J.; Schweitzer, V.A.; Thursky, K.; Kaasjager, H.A.H.; Ekkelenkamp, M.B. Interventions for rapid recognition and treatment of sepsis in the emergency department: A narrative review. Clin. Microbiol. Infect. 2021, 27, 192–203. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Noor, N.M.; Sousa, P.; Paul, S.; Roblin, X. Early diagnosis, early stratification, and early intervention to deliver precision medicine in IBD. Inflamm. Bowel Dis. 2022, 28, 1254–1264. [Google Scholar]
- Ansari Khoushabar, M.; Ghafariasl, P. Advanced Meta-Ensemble Machine Learning Models for Early and Accurate Sepsis Prediction to Improve Patient Outcomes. arXiv 2024, arXiv:2407.08107. [Google Scholar] [CrossRef] [Scilit]
- Gangula, R.; Thirupathi, L.; Parupati, R.; Sreeveda, K.; Gattoju, S. Ensemble machine learning based prediction of dengue disease with performance and accuracy elevation patterns. Mater. Today Proc. 2023, 80, 3458–3463. [Google Scholar] [CrossRef] [Scilit]
- Santangelo, O.E.; Gentile, V.; Pizzo, S.; Giordano, D.; Cedrone, F. Machine learning and prediction of infectious diseases: A systematic review. Mach. Learn. Knowl. Extr. 2023, 5, 175–198. [Google Scholar] [CrossRef] [Scilit]
- Xu, C.; Zhao, L.Y.; Ye, C.S.; Xu, K.C.; Xu, K.Y. The application of machine learning in clinical microbiology and infectious diseases. Front. Cell. Infect. Microbiol. 2025, 15, 1545646. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Lam, C.; Calvert, J.; Siefkas, A.; Barnes, G.; Pellegrini, E.; Green-Saxena, A.; Hoffman, J.; Mao, Q.; Das, R. Personalized stratification of hospitalization risk amidst COVID-19: A machine learning approach. Health Policy Technol. 2021, 10, 100554. [Google Scholar] [CrossRef] [Scilit]










| SI No. | Clinical Marker | Marker Description | Unit |
| 1 | Albumin | A protein made by the liver | g/dL |
| 2 | Base Excess | Amount of excess base in blood | mmol/L |
| 3 | Bicarbonate | Main buffer in blood | mmol/L |
| 4 | Bilirubin | Breakdown product of hemoglobin | mg/dL |
| 5 | Blast Cells | Immature blood cells | % or cells/µL |
| 6 | Calcium | Mineral essential for bone, nerve, and muscle function | mg/dL |
| 7 | Creatinine | Waste product from muscle metabolism | mg/dL |
| 8 | Gamma-Glutamyl-Transferase | Enzyme produced in liver and bile ducts | IU/L |
| 9 | Glucose | Main blood sugar | mg/dL |
| 10 | Hematocrit | The percentage of blood volume made up by red blood cells. | % |
| 11 | Hemoglobin | Oxygen-carrying molecule in red blood cells | g/dL |
| 12 | Lactate | By-product of anaerobic metabolism | mmol/L |
| 13 | Leukocytes | White blood cells of the immune system | 103/µL |
| 14 | Neutrophils | A subtype of white blood cells | % |
| 15 | C-Reactive Protein | Acute-phase protein produced by the liver | mg/L |
| 16 | Platelets | Small cells involved in clot formation | 103/µL |
| 17 | Potassium | Main intracellular electrolyte | mmol/L |
| 18 | Sodium | Major extracellular cation | mmol/L |
| 19 | Alanine aminotransferase | Liver enzyme | IU/L |
| 20 | Activated partial thromboplastin time | Time taken for blood to clot via the intrinsic pathway | seconds |
| 21 | Urea | End-product of protein metabolism | mg/dL |
| SI No. | Vital Signs | Vital sign description | Unit |
| 1 | Body Temperature | Indicates core heat of the body | °C |
| 2 | Systolic Blood Pressure | Pressure generated when the heart contracts and pumps blood into the arteries | mmHg |
| 3 | Diastolic Blood Pressure | The pressure in the arteries when the heart relaxes between beats | mmHg |
| 4 | Respiratory Rate | Number of breaths taken per minute | Breaths/Minute |
| 5 | Heart Rate | Number of heartbeats per minute | Beats/Minute |
| 6 | Oxygen Saturation | The percentage of hemoglobin in arterial blood that is carrying oxygen | % |
| Marker | Label | Number of Instances |
|---|---|---|
| Albumin Mean | Non-severe | 533 |
| Albumin Mean | Severe | 288 |
| Base Excess Arterial Mean | Non-severe | 533 |
| Base Excess Arterial Mean | Severe | 288 |
| Base Excess Venous Mean | Non-severe | 533 |
| Base Excess Venous Mean | Severe | 288 |
| Bicarbonate Arterial Mean | Non-severe | 533 |
| Bicarbonate Arterial Mean | Severe | 288 |
| Bicarbonate Venous Mean | Non-severe | 533 |
| Bicarbonate Venous Mean | Severe | 288 |
| Bilirubin Mean | Non-severe | 533 |
| Bilirubin Mean | Severe | 288 |
| Blast Mean | Non-severe | 533 |
| Blast Mean | Severe | 288 |
| Calcium Mean | Non-severe | 533 |
| Calcium Mean | Severe | 288 |
| Creatinine Mean | Non-severe | 533 |
| Creatinine Mean | Severe | 288 |
| Gamma-Glutamyl-Transferase Mean | Non-severe | 533 |
| Gamma-Glutamyl-Transferase Mean | Severe | 288 |
| Glucose Mean | Non-severe | 533 |
| Glucose Mean | Severe | 288 |
| Hematocrit Mean | Non-severe | 533 |
| Hematocrit Mean | Severe | 288 |
| Hemoglobin Mean | Non-severe | 533 |
| Hemoglobin Mean | Severe | 288 |
| Lactate Mean | Non-severe | 533 |
| Lactate Mean | Severe | 288 |
| Leukocytes Mean | Non-severe | 533 |
| Leukocytes Mean | Severe | 288 |
| Neutrophils Mean | Non-severe | 533 |
| Neutrophils Mean | Severe | 288 |
| C-Reactive Protein Mean | Non-severe | 533 |
| C-Reactive Protein Mean | Severe | 288 |
| Platelets Mean | Non-severe | 533 |
| Platelets Mean | Severe | 288 |
| Potassium Mean | Non-severe | 533 |
| Potassium Mean | Severe | 288 |
| Sodium Mean | Non-severe | 533 |
| Sodium Mean | Severe | 288 |
| Alanine Aminotransferase Mean | Non-severe | 533 |
| Alanine Aminotransferase Mean | Severe | 288 |
| Activated Partial Thromboplastin Time Mean | Non-severe | 533 |
| Activated Partial Thromboplastin Time Mean | Severe | 288 |
| Urea Mean | Non-severe | 533 |
| Urea Mean | Severe | 288 |
| Blood Pressure Diastolic Mean | Non-severe | 778 |
| Blood Pressure Diastolic Mean | Severe | 462 |
| Blood Pressure Systolic Mean | Non-severe | 778 |
| Blood Pressure Systolic Mean | Severe | 462 |
| Heart Rate Mean | Non-severe | 785 |
| Heart Rate Mean | Severe | 455 |
| Temperature Mean | Non-severe | 779 |
| Temperature Mean | Severe | 452 |
| Oxygen Saturation Mean | Non-severe | 788 |
| Oxygen Saturation Mean | Severe | 451 |
| Respiratory Rate Mean | Non-severe | 730 |
| Respiratory Rate Mean | Severe | 447 |
| Model | AUC | Precision | Recall |
|---|---|---|---|
| Logistic Regression | 0.84 | 0.80 | 0.78 |
| SVM | 0.86 | 0.82 | 0.79 |
| Decision Tree | 0.87 | 0.83 | 0.80 |
| KNN | 0.88 | 0.84 | 0.81 |
| AdaBoost | 0.91 | 0.86 | 0.83 |
| Light GBM | 0.94 | 0.89 | 0.86 |
| CatBoost | 0.95 | 0.90 | 0.87 |
| XGboost | 0.96 | 0.92 | 0.89 |
| Stacking (Ensemble) | 0.97 | 0.93 | 0.91 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Prabhu, S.; Bairy, G.M.; Sampathila, N.; Darshan, B.S.D. Explainable AI-Integrated Stacked Machine-Learning Model for Detection of Infectious Conditions Utilizing Vital Signs and Hematological Biomarkers. Information 2026, 17, 227. https://doi.org/10.3390/info17030227
Prabhu S, Bairy GM, Sampathila N, Darshan BSD. Explainable AI-Integrated Stacked Machine-Learning Model for Detection of Infectious Conditions Utilizing Vital Signs and Hematological Biomarkers. Information. 2026; 17(3):227. https://doi.org/10.3390/info17030227
Chicago/Turabian StylePrabhu, Savithri, Giliyar Muralidhar Bairy, Niranjana Sampathila, and BRP Siddarama Dhruva Darshan. 2026. "Explainable AI-Integrated Stacked Machine-Learning Model for Detection of Infectious Conditions Utilizing Vital Signs and Hematological Biomarkers" Information 17, no. 3: 227. https://doi.org/10.3390/info17030227
APA StylePrabhu, S., Bairy, G. M., Sampathila, N., & Darshan, B. S. D. (2026). Explainable AI-Integrated Stacked Machine-Learning Model for Detection of Infectious Conditions Utilizing Vital Signs and Hematological Biomarkers. Information, 17(3), 227. https://doi.org/10.3390/info17030227

