Conservative Continuous-Time Treatment Optimization

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Schneider, Nora, Manten, Georg, Kilbertus, Niki
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914402643476480
author Schneider, Nora
Manten, Georg
Kilbertus, Niki
author_facet Schneider, Nora
Manten, Georg
Kilbertus, Niki
contents We develop a conservative continuous-time stochastic control framework for treatment optimization from irregularly sampled patient trajectories. The unknown patient dynamics are modeled as a controlled stochastic differential equation with treatment as a continuous-time control. Naive model-based optimization can exploit model errors and propose out-of-support controls, so optimizing the estimated dynamics may not optimize the true dynamics. To limit extrapolation, we add a consistent signature-based MMD regularizer on path space that penalizes treatment plans whose induced trajectory distribution deviates from observed trajectories. The resulting objective minimizes a computable upper bound on the true cost. Experiments on benchmark datasets show improved robustness and performance compared to non-conservative baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2603_16789
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Conservative Continuous-Time Treatment Optimization
Schneider, Nora
Manten, Georg
Kilbertus, Niki
Machine Learning
Quantitative Methods
We develop a conservative continuous-time stochastic control framework for treatment optimization from irregularly sampled patient trajectories. The unknown patient dynamics are modeled as a controlled stochastic differential equation with treatment as a continuous-time control. Naive model-based optimization can exploit model errors and propose out-of-support controls, so optimizing the estimated dynamics may not optimize the true dynamics. To limit extrapolation, we add a consistent signature-based MMD regularizer on path space that penalizes treatment plans whose induced trajectory distribution deviates from observed trajectories. The resulting objective minimizes a computable upper bound on the true cost. Experiments on benchmark datasets show improved robustness and performance compared to non-conservative baselines.
title Conservative Continuous-Time Treatment Optimization
topic Machine Learning
Quantitative Methods
url https://arxiv.org/abs/2603.16789