Bayesian X-Learner: Calibrated Posterior Inference for Heterogeneous Treatment Effects under Heavy-Tailed Outcomes

2026-04-30 · Eichi Uehara arxiv

Conditional Average Treatment Effect (CATE) estimation in practice demands three properties simultaneously: heterogeneous effects $τ(x)$, calibrated uncertainty over them, and robustness to the heavy tails that contaminate real outcome data. Meta-learners (Künzel et al., 2019) give (i); causal forests and BART give (i)-(ii) with Gaussian-tail assumptions; no widely used tool gives all three. We present Bayesian X-Learner, an X-Learner built on cross-fitted doubly robust pseudo-outcomes (Kennedy, 2020) with a full MCMC posterior over $τ(x)$ via a Welsch redescending pseudo-likelihood. On Hill's IHDP benchmark the default configuration attains mean $\sqrt{\varepsilon_{\mathrm{PEHE}}} = 0.56$ on 5 replications (lowest mean; differences from S-/T-/X-learners, full-config Causal BART, and a causal forest baseline are not significant at $α=0.05$, and rank ordering is unstable at 10 replications -- IHDP comparisons are competitive rather than dominant). On contaminated "whale" DGPs with up to 20-25% tail density, a one-flag extension (contamination_severity) that selects a Huber-$δ$ nuisance loss per Huber's minimax-$δ$ relation recovers RMSE $\approx 0.13$ with tight credible intervals (single-cross-fit 30-seed coverage 83% [Wilson 66%, 93%] at 20% density; modular-Bayes pooling with Bayesian-bootstrap nuisance draws restores nominal 95% coverage).

📄 PDF Abstract BibTeX arXiv:2604.27394

Code (0)

등록된 구현이 없습니다.

Calibrated Test-Time Guidance for Bayesian Inference

2026-02-25 · Daniel Geyfman, Felix Draxler, Jan Groeneveld, Hyunsoo Lee 외 arxiv

Test-time guidance is a widely used mechanism for steering pretrained diffusion models toward outcomes specified by a reward function. Existing approaches, however, focus on maximizing reward rather than sampling from th…

Image Reconstruction Bayesian Inference

Calibrated One Round Federated Learning with Bayesian Inference in the Predictive Space

2023-12-15 · Mohsin Hasan, Guojun Zhang, Kaiyang Guo, Xi Chen 외

Federated Learning (FL) involves training a model over a dataset distributed among clients, with the constraint that each client's dataset is localized and possibly heterogeneous. In FL, small and noisy datasets are comm…

Bayesian Inference Federated Learning

Uncertainty-Aware Prediction of Lung Tumor Growth from Sparse Longitudinal CT Data via Bayesian Physics-Informed Neural Networks

2026-05-13 · Lingfei Kong, Haoran Ma arxiv

This work studies lung tumor growth prediction from sparse and irregular longitudinal computed tomography (CT) observations with measurement variability. A Bayesian physics-informed neural network is developed by combini…

Bayesian Inference

Loss-calibrated expectation propagation for approximate Bayesian decision-making

2022-01-10 · Michael J. Morais, Jonathan W. Pillow

Approximate Bayesian inference methods provide a powerful suite of tools for finding approximations to intractable posterior distributions. However, machine learning applications typically involve selecting actions, whic…

Bayesian Inference Decision Making

Generalized Bayes for Causal Inference

2026-03-03 · Emil Javurek, Dennis Frauen, Yuxin Wang, Stefan Feuerriegel arxiv

Uncertainty quantification is central to many applications of causal machine learning, yet principled Bayesian inference for causal effects remains challenging. Standard Bayesian approaches typically require specifying a…

Bayesian Inference Causal Inference