Beyond the ATE: Interpretable Modelling of Treatment Effects over Dose and Time

• URL: http://arxiv.org/abs/2507.07271v2
• Date: Mon, 21 Jul 2025 19:43:17 GMT
• Title: Beyond the ATE: Interpretable Modelling of Treatment Effects over Dose and Time
• Authors: Julianna Piskorz, Krzysztof Kacprzyk, Harry Amad, Mihaela van der Schaar,
• Abstract summary: We propose a framework for modelling treatment effect trajectories as smooth surfaces over dose and time.<n>Our approach decouples the estimation of trajectory shape from the specification of clinically relevant properties.<n>We show that our method yields accurate, interpretable, and editable models of treatment dynamics.
• Score: 46.2482873419289
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: The Average Treatment Effect (ATE) is a foundational metric in causal inference, widely used to assess intervention efficacy in randomized controlled trials (RCTs). However, in many applications -- particularly in healthcare -- this static summary fails to capture the nuanced dynamics of treatment effects that vary with both dose and time. We propose a framework for modelling treatment effect trajectories as smooth surfaces over dose and time, enabling the extraction of clinically actionable insights such as onset time, peak effect, and duration of benefit. To ensure interpretability, robustness, and verifiability -- key requirements in high-stakes domains -- we adapt SemanticODE, a recent framework for interpretable trajectory modelling, to the causal setting where treatment effects are never directly observed. Our approach decouples the estimation of trajectory shape from the specification of clinically relevant properties (e.g., maxima, inflection points), supporting domain-informed priors, post-hoc editing, and transparent analysis. We show that our method yields accurate, interpretable, and editable models of treatment dynamics, facilitating both rigorous causal analysis and practical decision-making.

Related papers

• Learning Causally Predictable Outcomes from Psychiatric Longitudinal Data [6.09170287691728]
  Causal inference in longitudinal biomedical data remains a central challenge.<n>Our algorithm learns non-negative, clinically interpretable weights for outcome aggregation.<n>Our algorithm consistently outperforms state-of-the-art methods in recovering causal effects.
  arXiv  Detail & Related papers  (2025-06-19T21:56:30Z)
• TV-SurvCaus: Dynamic Representation Balancing for Causal Survival Analysis [0.0]
  TV-SurvCaus is a novel framework that extends representation balancing techniques to the time-varying treatment setting for survival analysis.<n>We provide theoretical guarantees through (1) a generalized bound for time-varying precision in estimation of heterogeneous effects, (2) variance control via sequential balancing weights, (3) consistency results for dynamic treatment regimes, and (5) a formal bound on the bias due to treatment-confounder feedback.
  arXiv  Detail & Related papers  (2025-05-03T11:04:52Z)
• Dynamic Regularized CBDT: Variance-Calibrated Causal Boosting for Interpretable Heterogeneous Treatment Effects [3.390322732093844]
  We propose Dynamic Regularized Causal Boosted Decision Trees (CBDT)<n>CBDT integrates variance regularization and average treatment effect calibration into the loss function of gradient boosted decision trees.<n>We show thatCBDT significantly improves estimation accuracy while maintaining reliable coverage of true treatment effects.
  arXiv  Detail & Related papers  (2025-04-18T15:02:06Z)
• Higher-Order Causal Message Passing for Experimentation with Complex Interference [6.092214762701847]
  We introduce a new class of estimators based on causal message-passing, specifically designed for settings with pervasive, unknown interference.<n>Our estimator draws on information from the sample mean and variance of unit outcomes and treatments over time, enabling efficient use of observed data.
  arXiv  Detail & Related papers  (2024-11-01T18:00:51Z)
• Doubly Robust Proximal Causal Learning for Continuous Treatments [56.05592840537398]
  We propose a kernel-based doubly robust causal learning estimator for continuous treatments. We show that its oracle form is a consistent approximation of the influence function. We then provide a comprehensive convergence analysis in terms of the mean square error.
  arXiv  Detail & Related papers  (2023-09-22T12:18:53Z)
• TCFimt: Temporal Counterfactual Forecasting from Individual Multiple Treatment Perspective [50.675845725806724]
  We propose a comprehensive framework of temporal counterfactual forecasting from an individual multiple treatment perspective (TCFimt) TCFimt constructs adversarial tasks in a seq2seq framework to alleviate selection and time-varying bias and designs a contrastive learning-based block to decouple a mixed treatment effect into separated main treatment effects and causal interactions. The proposed method shows satisfactory performance in predicting future outcomes with specific treatments and in choosing optimal treatment type and timing than state-of-the-art methods.
  arXiv  Detail & Related papers  (2022-12-17T15:01:05Z)
• Interpretable Deep Causal Learning for Moderation Effects [0.0]
  We address the problem of interpretability and targeted regularization in causal machine learning models. We propose a novel deep counterfactual learning architecture for estimating individual treatment effects.
  arXiv  Detail & Related papers  (2022-06-21T11:21:09Z)
• Benchmarking Heterogeneous Treatment Effect Models through the Lens of Interpretability [82.29775890542967]
  Estimating personalized effects of treatments is a complex, yet pervasive problem. Recent developments in the machine learning literature on heterogeneous treatment effect estimation gave rise to many sophisticated, but opaque, tools. We use post-hoc feature importance methods to identify features that influence the model's predictions.
  arXiv  Detail & Related papers  (2022-06-16T17:59:05Z)
• Disentangled Counterfactual Recurrent Networks for Treatment Effect Inference over Time [71.30985926640659]
  We introduce the Disentangled Counterfactual Recurrent Network (DCRN), a sequence-to-sequence architecture that estimates treatment outcomes over time. With an architecture that is completely inspired by the causal structure of treatment influence over time, we advance forecast accuracy and disease understanding. We demonstrate that DCRN outperforms current state-of-the-art methods in forecasting treatment responses, on both real and simulated data.
  arXiv  Detail & Related papers  (2021-12-07T16:40:28Z)
• SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Data [83.50281440043241]
  We study the problem of inferring heterogeneous treatment effects from time-to-event data. We propose a novel deep learning method for treatment-specific hazard estimation based on balancing representations.
  arXiv  Detail & Related papers  (2021-10-26T20:13:17Z)
• On the role of surrogates in the efficient estimation of treatment effects with limited outcome data [43.17788100119767]
  We study how incorporating data on units for which only surrogate outcomes not of primary interest are observed can increase the precision of ATE estimation. We develop robust ATE estimation and inference methods that realize these efficiency gains.
  arXiv  Detail & Related papers  (2020-03-27T13:31:49Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.