R.K.A. Karlsson
Please Note
23 records found
1
Robust Causal Inference with Multi-task Gaussian Processes
Enhancing Generalization and Calibration through Data-Aware Kernel and Prior Design
dividualized treatment effects by modeling potential outcomes as correlated functions. However,
they struggle under high-dimensionality and treatment imbalance, leading to overfitting and unre-
liable uncertainty estimates. This study examines two failure modes: poor generalization in high-
dimensional spaces and overconfident predictions in low-overlap regions. To address these, two
data-aware enhancements are proposed: an overlap-adaptive kernel that scales similarity based on
local treatment density, and a regularized prior that down-weights unstable features using marginal
treatment effect variance. Evaluations on synthetic data and the IHDP benchmark show improved
effect estimation, credible interval calibration, and robustness in challenging settings. These find-
ings highlight practical strategies for enhancing CMGPs in real-world causal inference tasks. ...
dividualized treatment effects by modeling potential outcomes as correlated functions. However,
they struggle under high-dimensionality and treatment imbalance, leading to overfitting and unre-
liable uncertainty estimates. This study examines two failure modes: poor generalization in high-
dimensional spaces and overconfident predictions in low-overlap regions. To address these, two
data-aware enhancements are proposed: an overlap-adaptive kernel that scales similarity based on
local treatment density, and a regularized prior that down-weights unstable features using marginal
treatment effect variance. Evaluations on synthetic data and the IHDP benchmark show improved
effect estimation, credible interval calibration, and robustness in challenging settings. These find-
ings highlight practical strategies for enhancing CMGPs in real-world causal inference tasks.
This paper systematically investigates how the depth and width of TARNet affect the CATE estimation in diverse simulated data environments. The research investigates two central questions: how TARNet's performance varies across data regimes (e.g., confounding strength, sample size), and how its optimal architecture changes in response to these conditions.
A comprehensive set of simulation-based experiments is conducted using the CATENets framework, isolating and varying factors such as sample size, feature dimensionality, confounding strength, and the presence of noise. The results demonstrate that deeper architectures generally yield better performance in complex or high-dimensional scenarios, whereas narrower networks are preferable in small-sample or high-noise settings due to their regularizing effect. Furthermore, the findings suggest that there is no universally optimal architecture. The best configuration depends on the specific characteristics of the data. The study concludes with practical recommendations for architecture selection based on the experiments conducted.
...
This paper systematically investigates how the depth and width of TARNet affect the CATE estimation in diverse simulated data environments. The research investigates two central questions: how TARNet's performance varies across data regimes (e.g., confounding strength, sample size), and how its optimal architecture changes in response to these conditions.
A comprehensive set of simulation-based experiments is conducted using the CATENets framework, isolating and varying factors such as sample size, feature dimensionality, confounding strength, and the presence of noise. The results demonstrate that deeper architectures generally yield better performance in complex or high-dimensional scenarios, whereas narrower networks are preferable in small-sample or high-noise settings due to their regularizing effect. Furthermore, the findings suggest that there is no universally optimal architecture. The best configuration depends on the specific characteristics of the data. The study concludes with practical recommendations for architecture selection based on the experiments conducted.
When Causal Forests Mislead
Evaluating the precision of Confidence Intervals
A primary discovery is the identification of a practical limit for reliable confidence interval coverage: When the sum of confounders and effect modifiers exceeds 4, coverage rates drop considerably below 80%, even for simple treatment effect functions. This limitation remains steady despite substantial increases in computational resources.
The examination of hyperparameters revealed that the most influential parameters are the maximum tree depth and the balance tolerance in splits, which demonstrate substantial changes in performance, both of which performed best at their maximums (unlimited and 0.5, respectively). Other key suggestions involve increasing the training data fraction per tree from 0.45 to 0.5, keeping the minimum impurity decrease threshold at 0.0, and utilizing at least ≈ 2400 trees to meet theoretical expectations.
In addition, this paper did not identify any noteworthy interaction between tree count and sample size. As a result, both of these characteristics can be optimized independently of each other.
These findings provide systematic guidelines for practitioners to assess when causal forest confidence intervals are reliable and how to optimize them, bridging the gap between theoretical guarantees and practical performance. ...
A primary discovery is the identification of a practical limit for reliable confidence interval coverage: When the sum of confounders and effect modifiers exceeds 4, coverage rates drop considerably below 80%, even for simple treatment effect functions. This limitation remains steady despite substantial increases in computational resources.
The examination of hyperparameters revealed that the most influential parameters are the maximum tree depth and the balance tolerance in splits, which demonstrate substantial changes in performance, both of which performed best at their maximums (unlimited and 0.5, respectively). Other key suggestions involve increasing the training data fraction per tree from 0.45 to 0.5, keeping the minimum impurity decrease threshold at 0.0, and utilizing at least ≈ 2400 trees to meet theoretical expectations.
In addition, this paper did not identify any noteworthy interaction between tree count and sample size. As a result, both of these characteristics can be optimized independently of each other.
These findings provide systematic guidelines for practitioners to assess when causal forest confidence intervals are reliable and how to optimize them, bridging the gap between theoretical guarantees and practical performance.
Analyzing the Impact of Depth and Leaf Size on CATE Estimation in Honest Causal Trees
A Study of Model Accuracy and Generalization Across Simulated and Real-World Data
The results reveal that optimal hyperparameter configurations are dependent on the data characteristics, such as dimensionality, noise levels, and the complexity of the true causal effects. Honest causal trees demonstrate a better performance in high-dimensional and noisy environments due to their effective variance control. Conversely, in simpler, low-noise settings or complex CATE structures, adaptive causal trees or baseline models frequently achieve better results by reducing bias. The study also highlights the challenges of using moderately sized datasets, where the sample splitting limitations can lead to higher estimation errors. This work provides thorough suggestions for hyperparameter selection, emphasizing the fact that tuning based on the underlying characteristics of the data is needed for achieving the best CATE estimates possible. ...
The results reveal that optimal hyperparameter configurations are dependent on the data characteristics, such as dimensionality, noise levels, and the complexity of the true causal effects. Honest causal trees demonstrate a better performance in high-dimensional and noisy environments due to their effective variance control. Conversely, in simpler, low-noise settings or complex CATE structures, adaptive causal trees or baseline models frequently achieve better results by reducing bias. The study also highlights the challenges of using moderately sized datasets, where the sample splitting limitations can lead to higher estimation errors. This work provides thorough suggestions for hyperparameter selection, emphasizing the fact that tuning based on the underlying characteristics of the data is needed for achieving the best CATE estimates possible.
informative feature representations, which are then used for effect estimation. However, such feature representations can introduce confounding bias in the results if they lose confounding information contained within the original features. In this work, we evaluate an existing metric for measuring confounding bias in representations and use this metric to study two representation learning methods in terms of their biases in settings with low overlap between treated and control populations. We show that the metric is a suitable measure for the amount of confounding bias in a representation and representation learning methods that minimise this bias lead to better average treatment effect estimation in our experiments. ...
informative feature representations, which are then used for effect estimation. However, such feature representations can introduce confounding bias in the results if they lose confounding information contained within the original features. In this work, we evaluate an existing metric for measuring confounding bias in representations and use this metric to study two representation learning methods in terms of their biases in settings with low overlap between treated and control populations. We show that the metric is a suitable measure for the amount of confounding bias in a representation and representation learning methods that minimise this bias lead to better average treatment effect estimation in our experiments.
Optimizing Mechanical Ventilation Support for Patients in Intensive Care Units
An Analysis of Deep Learning Methods for Personalizing Positive End-Expiratory Pressure Regime
Individualized treatment effect prediction for Mechanical Ventilation
Using Causal Multi-task Gaussian Process to estimate the individualized treatment effect of a low vs high PEEP regime on ICU patients
Using forest-based models to personalise ventilation treatment in the ICU
Optimising positive end-expiratory pressure assignment based on the MIMIC-IV dataset
Machine Learning for Personalized Respiratory Care
A DR-learner Approach to Positive End-Expiratory Pressure Effect Estimation
Personalizing Treatment for Intensive Care Unit Patients with Acute Respiratory Distress Syndrome
Comparing the S-, T-, and X-learner to Estimate the Conditional Average Treatment Effect for High versus Low Positive End-Expiratory Pressure in Mechanical Ventilation
Possibility of using overrule to evaluate overlap in causal inference
What is the performance of overrule in identifying overlap for different types of datasets?
Following the experiments, the method seems to perform best when the treatment and control groups share one region of overlap. In this case, logistic regression works best out of the classifiers that were tested. The overall performance drops when the two groups have multiple regions of overlap. For this, the random forest classifier performs best instead. Throughout all scenarios, the performance of the model drops with increasing dimensionality. Furthermore, having a small percentage of outliers only slightly affects the model. With more outliers, logistic regression is the only classifier further affected. ...
Following the experiments, the method seems to perform best when the treatment and control groups share one region of overlap. In this case, logistic regression works best out of the classifiers that were tested. The overall performance drops when the two groups have multiple regions of overlap. For this, the random forest classifier performs best instead. Throughout all scenarios, the performance of the model drops with increasing dimensionality. Furthermore, having a small percentage of outliers only slightly affects the model. With more outliers, logistic regression is the only classifier further affected.