Search papers, labs, and topics across Lattice.
This paper addresses challenges in estimating Individual Treatment Effects (ITE) in multi-treatment settings by deriving a novel multi-treatment generalization bound and a theoretical estimator for the optimal balancing weight, thus avoiding heuristic hyperparameter tuning. They introduce a Treatment Aggregation strategy that ensures both accuracy and O(1) scalability, outperforming Pairwise and One-vs-All approaches, especially in high-dimensional treatment spaces. They further develop Multi-Treatment CausalEGM, a generative architecture that preserves the Wasserstein geodesic structure of the treatment manifold, demonstrating improved accuracy and efficiency on semi-synthetic and image datasets.
Forget hyperparameter tuning for multi-treatment effect estimation: this work provides a theoretical estimator for optimal balancing weights, achieving O(1) scalability without sacrificing accuracy.
Estimating Individual Treatment Effects (ITE) in multi-treatment scenarios faces two critical challenges: the Hyperparameter Selection Dilemma for balancing weights and the Curse of Dimensionality in computational scalability. This paper derives a novel multi-treatment generalization bound and proposes a theoretical estimator for the optimal balancing weight $\alpha$, eliminating expensive heuristic tuning. We investigate three balancing strategies: Pairwise, One-vs-All (OVA), and Treatment Aggregation. While OVA achieves superior precision in low-dimensional settings, our proposed Treatment Aggregation ensures both accuracy and O(1) scalability as the treatment space expands. Furthermore, we extend our framework to a generative architecture, Multi-Treatment CausalEGM, which preserves the Wasserstein geodesic structure of the treatment manifold. Experiments on semi-synthetic and image datasets demonstrate that our approach significantly outperforms traditional models in estimation accuracy and efficiency, particularly in large-scale intervention scenarios.
