Heatmaps for visual explainability of CNN-based predictions for multivariate time series with application to healthcare

Viton, Fabien; Elbattah, Mahmoud; Guerin, Jean Luc; Dequen, Gilles

Abstract

The need for explainable AI is becoming increasingly important for critical decision domains such as healthcare for example. In this context, this paper is concerned with explaining the predictions of Convolutional Neural Networks (CNNs) with particular focus on multivariate Time Series (TS) problems. The approach is based on heatmaps as a visual means to highlight the significant variables over the temporal sequence. Furthermore, a channel-wise CNN architecture is implemented to allow for considering the TS variables separately. The approach is applied to the problem of predicting the risk of in-hospital mortality. We use a dataset from the MIMIC-III database, which included about 13K ICU records. The experimental results demonstrate rational insights that conform with the common medical knowledge in this regard.

• Presentation Conference Type: Conference Paper (Published)
• Conference Name: 2020 IEEE International Conference on Healthcare Informatics, ICHI 2020
• Start Date: Nov 30, 2020
• End Date: Dec 3, 2020
• Acceptance Date: Apr 17, 2020
• Online Publication Date: Mar 12, 2021
• Publication Date: Nov 1, 2020
• Deposit Date: Apr 26, 2022
• Publisher: Institute of Electrical and Electronics Engineers (IEEE)
• Book Title: 2020 IEEE International Conference on Healthcare Informatics (ICHI)
• ISBN: 9781728153834
• DOI: https://doi.org/10.1109/ICHI48887.2020.9374393
• Public URL: https://uwe-repository.worktribe.com/output/9206328