mlsanityData to 14 Oct 2026

Conservative Continuous-Time Treatment Optimization

arXiv 2603.16789 ↗First posted 17 Mar 2026cs.LG q-bio.QMcited by 1

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 20 of its 43 references that are in the map. First posted on arXiv 17 Mar 2026.

Builds on

20 of 43 references.

References · 20

  1. 2021COMBO: Conservative Offline Model-Based Policy Optimizationcited by 370
  2. 2018Algorithmic Framework for Model-based Deep Reinforcement Learning with Theoretical Guaranteescited by 140
  3. 2020A Measure-Theoretic Approach to Kernel Conditional Mean Embeddingscited by 86
  4. 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
  5. 2020The Signature Kernel is the solution of a Goursat PDEcited by 58
  6. 2021Integrating Expert ODEs into Neural ODEs: Pharmacology and Disease Progressioncited by 48
  7. 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
  8. 2023Non-adversarial training of Neural SDEs with signature kernel scorescited by 37
  9. 2023Bayesian Neural Controlled Differential Equations for Treatment Effect Estimationcited by 26
  10. 2022Predicting the impact of treatments over time with uncertainty aware neural differential equationscited by 23
  11. 2023Accounting For Informative Sampling When Learning to Forecast Treatment Outcomes Over Timecited by 15
  12. 2023A Neural RDE approach for continuous-time non-Markovian stochastic control problemscited by 14
8 more
  1. 2024Lecture notes on rough paths and applications to machine learningcited by 14
  2. 2023Uncertainty-driven Trajectory Truncation for Data Augmentation in Offline Reinforcement Learningcited by 13
  3. 2023Finding Counterfactually Optimal Action Sequences in Continuous State Spacescited by 13
  4. 2024ODE Discovery for Longitudinal Heterogeneous Treatment Effects Inferencecited by 12
  5. 2022Optimizing Pessimism in Dynamic Treatment Regimes: A Bayesian Learning Approachcited by 7
  6. 2024Stabilized Neural Prediction of Potential Outcomes in Continuous Timecited by 7
  7. 2022Causal identification for continuous-time stochastic processescited by 2
  8. 2022Causal Identification for Complex Functional Longitudinal Studiescited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.