AI-driven analysis by identifying risk factors of VL relapse in HIV co-infected patients.

Journal: Scientific reports
Published Date:

Abstract

Visceral Leishmaniasis (VL), also known as Kala-Azar, poses a significant global public health challenge and is a neglected disease, with relapses and treatment failures leading to increased morbidity and mortality. This study introduces an explainable machine learning approach to predict VL relapse and identify critical risk factors, thereby aiding patient monitoring and treatment strategies. Leveraging data from a follow-up study of 571 patients, the survival machine learning models are applied, including Random Survival Forest (RSF), Survival Support Vector Machine (SSVM), and eXtreme Gradient Boosting (XGBoost), for relapse prediction. The results demonstrated that RSF, with a C-index of 0.85, outperformed the conventional Cox Proportional Hazard (CPH) model (C-index 0.8), offering improved prediction capabilities by capturing non-linear relationships and variable interactions. To address the lack of transparency (in terms of feature importance) in Machine Learning (ML) models, the SHapley Additive exPlanation (SHAP) method is employed, which enhances model interpretability (feature importance) through visual insights. SHAP dependence plots allowed the healthcare professionals to evaluate which factors encourage the occurrence of the relapse. A statistically significant relationship between HIV co-infection (HR=3.92, 95% CI=2.03-7.58) and VL relapse was identified through -2 log-likelihood ratio and chi-square tests. These results indicate the promise of explainable artificial intelligence (XAI) for making clinical decisions and remedying recurrences in VL.

Authors

  • Abhishek Kumar
    Manipal Academy of Higher Education (MAHE), Manipal, India.
  • Sanchita Mondal
    School of Medical Science and Technology, IIT Kharagpur, Kharagpur, West Bengal, 721302, India.
  • Debnarayan Khatua
    Department of Mathematics and Statistics, Vignan's Foundation for Science, Technology and Research, Andhra Pradesh, 522213, India.
  • Debashree Guha
    School of Medical Science and Technology, IIT Kharagpur, Kharagpur, West Bengal, 721302, India.
  • Budhaditya Mukherjee
    School of Medical Science and Technology, IIT Kharagpur, Kharagpur, West Bengal, 721302, India.
  • Arista Lahiri
    Dr B C Roy Multi Speciality Medical Research Centre, IIT Kharagpur, Kharagpur, West Bengal, 721302, India.
  • Dilip K Prasad
    Department of Computer Science, UiT The Arctic University of Norway, Tromsø, Norway.
  • Arif Ahmed Sekh
    Department of Computer Science, UiT The Arctic University of Norway, Tromsø, 9017, Norway. arif.a.sekh@uit.no.