CoxSE: Exploring the potential of self-explaining neural networks with Cox proportional hazards model for survival analysisShow others and affiliations
2026 (English)In: Knowledge-Based Systems, ISSN 0950-7051, E-ISSN 1872-7409, Vol. 333, p. 1-13, article id 114996Article in journal (Refereed) Published
Abstract [en]
The Cox Proportional Hazards (CPH) model has long been the preferred survival model for its explainability. However, to increase its predictive power beyond its linear log-risk, it was extended to utilize deep neural networks, sacrificing its explainability. In this work, we explore the potential of self-explaining neural networks (SENN) for survival analysis. We propose a new locally explainable Cox proportional hazards model, named CoxSE, by estimating a locally-linear log-hazard function using the SENN. We also propose a modification to the Neural additive (NAM) model, hybrid with SENN, named CoxSENAM, which enables the control of the stability and consistency of the generated explanations. Several experiments using synthetic and real datasets are presented, benchmarking CoxSE and CoxSENAM against a NAM-based model, a DeepSurv model explained with SHAP, and a linear CPH model. The results show that, unlike the NAM-based model, the SENN-based model can provide more stable and consistent explanations while maintaining the predictive power of the black-box model. The results also show that, due to their structural design, NAM-based models demonstrate better robustness to non-informative features. Among the models, the hybrid model exhibits the best robustness. Full implementation is available on GitHub. © 2025 The Authors
Place, publisher, year, edition, pages
Asterdam: Elsevier, 2026. Vol. 333, p. 1-13, article id 114996
Keywords [en]
Cox proportional hazards, Interpretability, Neural additive models, Self-explaining neural networks, Survival analysis, XAI
National Category
Probability Theory and Statistics
Identifiers
URN: urn:nbn:se:hh:diva-58108DOI: 10.1016/j.knosys.2025.114996ISI: 001637163000001Scopus ID: 2-s2.0-105024190200OAI: oai:DiVA.org:hh-58108DiVA, id: diva2:2036324
Funder
Knowledge FoundationVinnova2026-02-062026-02-062026-04-17Bibliographically approved