GeoACE: Geometry-Diverse Anchor-Correction Ensembles for Heterogeneous Treatment Effect Estimation

Ali Haghpanah Jahromi, Mohammad Taheri, Zohreh Azimifar

Predicting heterogeneous causal effects supports informed decision-making, yet suitable inductive biases vary with treatment overlap, covariate imbalance, prognostic structure, and sample size. The Geometry-Diverse Anchor--Correction Expert Ensemble (GeoACE) addresses this variation through five complementary experts whose predictions are combined using weights estimated from validation data. Compared with ensemble designs that derive diversity primarily from sample or feature subsets, reweighting, or feature transformations, its central contribution is the coordinated design of training objectives and anchor geometries that target different prognostic and overlap structures. Following validation, each expert is refitted on the combined training and validation data, while retaining its distinctive structure. The shared architecture couples a structured outcome anchor with a learned neural correction. ACE serves as the reference estimator, while OW-ACE emphasizes well-supported covariate regions through overlap-weighted factual training. G$Φ$-ACE constructs a global outcome-guided geometry from pooled prognostic associations, whereas A$Φ$-ACE incorporates arm-specific outcome…