2024/11/28 by Oriol Corcoll Andreu, Athanasios Vlontzos, Andreu, Oriol Corcoll +6 · 1 voice · 4 citations
Chemistry · Engineering · #Advanced Polymer Synthesis and Characterization #Innovative Microfluidic and Catalytic Techniques Innovation #cs.AI #cs.LG #stat.ML
paper · pdf · doi:10.48550/arxiv.2411.19245
openalex publication_date 2024/11/28 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28
Estimating causal effects is vital for decision making. In standard causal effect estimation, treatments are usually binary- or continuous-valued. However, in many important real-world settings, treatments can be structured, high-dimensional objects, such as text, video, or audio. This provides a challenge to traditional causal effect estimation. While leveraging the shared structure across different treatments can help generalize to unseen treatments at test time, we show in this paper that using such structure blindly can lead to biased causal effect estimation. We address this challenge by devising a novel contrastive approach to learn a representation of the high-dimensional treatments, and prove that it identifies underlying causal factors and discards non-causally relevant factors. We prove that this treatment representation leads to unbiased estimates of the causal effect, and empirically validate and benchmark our results on synthetic and real-world datasets.