Interpretable-by-design Deep Survival Analysis for Disease Progression Modeling

Julius Gervelmeyer,Sarah Mueller,Kerol Djoumessi,David Merle,Simon J Clark,Lisa Koch,Philipp Berens
DOI: https://doi.org/10.1101/2024.07.11.24310270
2024-07-11
Abstract:In the elderly, degenerative diseases often develop differently over time for individual patients. For optimal treatment, physicians and patients would like to know how much time is left for them until symptoms reach a certain stage. However, compared to simple disease detection tasks, disease progression modeling has received much less attention. In addition, most existing models are black-box models which provide little insight into the mechanisms driving the prediction. Here, we introduce an interpretable-by-design survival model to predict the progression of age-related macular degeneration (AMD) from fundus images. Our model not only achieves state-of-the-art prediction performance compared to black-box models but also provides a sparse map of local evidence of AMD progression for individual patients. Our evidence map faithfully reflects the decision-making process of the model in contrast to widely used post-hoc saliency methods. Furthermore, we show that the identified regions mostly align with established clinical AMD progression markers. We believe that our method may help to inform treatment decisions and may lead to better insights into imaging biomarkers indicative of disease progression. The project's code is available at github.com/berenslab/interpretable-deep-survival-analysis.
What problem does this paper attempt to address?
The paper aims to address the issue of predicting the progression of age-related macular degeneration (AMD) in elderly patients. Specifically, the researchers propose a "design as explanation" deep survival analysis model to predict AMD progression from fundus images. Compared to traditional black-box models, this model not only achieves state-of-the-art predictive performance but also provides sparse local evidence maps to intuitively display local evidence of AMD progression for individual patients. These evidence maps faithfully reflect the model's decision-making process, and the identified regions mostly align with established clinical markers of AMD progression. The researchers believe that this approach can help guide treatment decisions and may lead to new insights into imaging biomarkers, thereby improving the understanding of disease progression. In short, the main goal of the paper is to develop an inherently interpretable deep learning model for predicting AMD progression and to enhance the trust of doctors and patients by providing clear explanations.