Clinically Interpretable Survival Prediction in Primary Biliary Cholangitis with TreeSHAP and Gradient-Boosted Model

Read full text
Authors: Emmanuel Pio Pastore
Cited by: 0

Abstract

Survival prediction tools aid management of chronic liver diseases by informing decisions on monitoring and transplantation. In primary biliary cholangitis (PBC), classical risk scores discriminate well but are hard to interpret and treat variables in isolation, whereas clinicians think in terms of broader constructs such as cholestasis or coagulopathy. We introduce Surv-TCAV, a concept-based interpretability method for accelerated failure-time (AFT) survival models, and apply it to gradient-boosted trees on the Mayo Clinic PBC cohort (276 complete cases, 17 baseline variables). An XGBoost-AFT model was trained on 25 repeated 80/20 splits; performance was assessed using Harrell’s C-index and the integrated Brier score, with Cox and Weibull AFT models as transparent baselines. TreeSHAP provided feature-level attributions, while Surv-TCAV probed model sensitivity along clinician-defined directions (cholestasis, coagulopathy, low albumin, older age and clinical complications). The boosted model achieved high discrimination (C-index ≈ 0.835) but weaker calibration (IBS ≈ 0.36), reflecting the usual flexibility–calibration trade-off. Both feature- and concept-level explanations aligned with hepatology practice: bilirubin, alkaline phosphatase, copper, low albumin, prolonged prothrombin time, age and clinical complications all reduced predicted survival. These results show how boosted AFT models can be interpreted in familiar clinical terms. The PBC analysis is a methodological illustration; all code, figures and data are provided for reproducibility.

Fields of Study & Subjects

Explainable Artificial Intelligence (XAI) Liver Diseases and Immunity Artificial Intelligence in Healthcare

Publication details

Publication date
11 February 2026
Language
EN
Document type
Preprint