Towards Better Interpretability of Sepsis Prediction by Deep Neural Networks with Variable-wise Attribution Maps - Archive ouverte HAL
Pré-Publication, Document De Travail (Preprint/Prepublication) Année : 2024

Towards Better Interpretability of Sepsis Prediction by Deep Neural Networks with Variable-wise Attribution Maps

Résumé

Because of the multi-symptomatic nature of sepsis, its prediction is challenging as it requires considering subtle changes in multiple monitored variables across time. Recent works based on deep neural networks improved the prediction performance, but still suffer from poor interpretability. Critical for healthcare applications, we propose to improve this aspect by separating time variables in the convolution layers of a sepsis prediction network. We reveal the improvement in interpretability capacity with the use of gradient-based attributions on high-level intermediate features and through a metric correlating the variable attribution with the prediction for perturbed pathologic samples. With 171,945 patients from the MIMIC-IV database, we demonstrate that our method not only maintains classification performances at similar network parameters count, but also substantially improves the faithfulness of per-variable attributions.
Fichier principal
Vignette du fichier
ISBI_Pierre_Elliot-2.pdf (217.96 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04776927 , version 1 (12-11-2024)
hal-04776927 , version 2 (02-12-2024)

Identifiants

  • HAL Id : hal-04776927 , version 2

Citer

P.-E Thiboud, V Wargnier-Dauchelle, M Lefort, N Duchateau, M Sdika. Towards Better Interpretability of Sepsis Prediction by Deep Neural Networks with Variable-wise Attribution Maps. 2024. ⟨hal-04776927v2⟩
10 Consultations
8 Téléchargements

Partager

More