Uncertainty-Aware Optimal Treatment Selection for Clinical Time Series

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Schwarz, Thomas, Casolo, Cecilia, Kilbertus, Niki
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912069376278528
author Schwarz, Thomas
Casolo, Cecilia
Kilbertus, Niki
author_facet Schwarz, Thomas
Casolo, Cecilia
Kilbertus, Niki
contents In personalized medicine, the ability to predict and optimize treatment outcomes across various time frames is essential. Additionally, the ability to select cost-effective treatments within specific budget constraints is critical. Despite recent advancements in estimating counterfactual trajectories, a direct link to optimal treatment selection based on these estimates is missing. This paper introduces a novel method integrating counterfactual estimation techniques and uncertainty quantification to recommend personalized treatment plans adhering to predefined cost constraints. Our approach is distinctive in its handling of continuous treatment variables and its incorporation of uncertainty quantification to improve prediction reliability. We validate our method using two simulated datasets, one focused on the cardiovascular system and the other on COVID-19. Our findings indicate that our method has robust performance across different counterfactual estimation baselines, showing that introducing uncertainty quantification in these settings helps the current baselines in finding more reliable and accurate treatment selection. The robustness of our method across various settings highlights its potential for broad applicability in personalized healthcare solutions.
format Preprint
id arxiv_https___arxiv_org_abs_2410_08816
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Uncertainty-Aware Optimal Treatment Selection for Clinical Time Series
Schwarz, Thomas
Casolo, Cecilia
Kilbertus, Niki
Machine Learning
In personalized medicine, the ability to predict and optimize treatment outcomes across various time frames is essential. Additionally, the ability to select cost-effective treatments within specific budget constraints is critical. Despite recent advancements in estimating counterfactual trajectories, a direct link to optimal treatment selection based on these estimates is missing. This paper introduces a novel method integrating counterfactual estimation techniques and uncertainty quantification to recommend personalized treatment plans adhering to predefined cost constraints. Our approach is distinctive in its handling of continuous treatment variables and its incorporation of uncertainty quantification to improve prediction reliability. We validate our method using two simulated datasets, one focused on the cardiovascular system and the other on COVID-19. Our findings indicate that our method has robust performance across different counterfactual estimation baselines, showing that introducing uncertainty quantification in these settings helps the current baselines in finding more reliable and accurate treatment selection. The robustness of our method across various settings highlights its potential for broad applicability in personalized healthcare solutions.
title Uncertainty-Aware Optimal Treatment Selection for Clinical Time Series
topic Machine Learning
url https://arxiv.org/abs/2410.08816