Abstract
Patient stratification is a crucial task aimed at categorizing individuals with a specific disease into more homogeneous subgroups based on critical disease-related characteristics. This process enables personalized interventions, optimized care management, and tailored treatments. Patient stratification plays a significant role in drug development and clinical practice for many diseases. However, with the increasing availability of biomedical data, such as gene expression data, clinical records, and lifestyle/environmental factors, the analysis of this vast and multimodal data becomes highly challenging. Machine learning offers methods that can help address the challenges of transforming this extensive and diverse data into usable decision-support tools. Deep learning methods, in particular, have shown impressive results in tasks such as risk stratification and treatment response prediction. However, their impact on data-driven medicine remains limited due to their ‘black-box’ nature and their inability to provide human-interpretable outputs. In this study, we propose applying topological data analysis to enhance the interpretability of deep learning patient stratification models. Specifically, we suggest using the Mapper algorithm to visualize the latent space learned by the models through the lens of its predictions. We apply the Mapper algorithm to various architectures of recently developed deep patient stratification models and demonstrate how it helps reveal relationships among different patient subgroups. Furthermore, we adapt the Normalized Mutual Information measure to identify the Mapper’s parameters that yield the most optimal graph-based representation of the latent space. This approach aims to enrich the power of deep learning with interpretable results in the field of patient stratification.
Original language | English |
---|---|
Title of host publication | International Conference on e-Health and Bioengineering (EHB 2023 ): proceedings |
Publisher | Springer Cham |
Pages | 563–574 |
ISBN (Electronic) | 9783031625022 |
ISBN (Print) | 9783031625015 |
DOIs | |
Publication status | Published - 30 Aug 2024 |
Event | 11th International Conference on e-Health and Bioengineering - Romania, Bucharest, Romania Duration: 09 Nov 2023 → 10 Nov 2023 http://www.ehbconference.ro/Home.aspx |
Publication series
Name | IFMBE Proceedings |
---|---|
Publisher | Springer, Cham |
ISSN (Print) | 1680-0737 |
ISSN (Electronic) | 1433-9277 |
Conference
Conference | 11th International Conference on e-Health and Bioengineering |
---|---|
Abbreviated title | EHB 2023 |
Country/Territory | Romania |
City | Bucharest |
Period | 09/11/2023 → 10/11/2023 |
Internet address |
Keywords
- interpreting
- deep patient stratification models
- topological data analysis