acceptodds
Under review as a conference paper at ICLR 2027

Orthogonal Learning for Conditional Geodesic Treatment Effects

Abstract

Estimating conditional average treatment effects (CATEs) is a fundamental task in causal inference, and orthogonal learners reduce sensitivity to nuisance estimation. In many scientific applications, however, outcomes are naturally represented on Riemannian manifolds, where Euclidean differences do not define geometry-respecting treatment effects. Existing work develops Fr\'echet regression for manifold-valued outcomes and geodesic causal effects, but leaves open how to perform Neyman-orthogonal pointwise estimation and inference for covariate-dependent geodesic effects from observational data. To address this, we define the conditional geodesic treatment effect through treatment-specific conditional Fr\'echet means and develop the Geodesic Orthogonal learner (GO-learner) for its estimation. We derive a joint error expansion in which errors in estimating the two means are propagated through the logarithm map, propensity and conditional regression errors enter through products, and the resulting bounds yield convergence guarantees and asymptotically valid pointwise confidence sets. Experimental results demonstrate that (i) GO-learner has lower average MSE among the three evaluated estimators with sample size increasing; (ii) the predicted bias reduces when either nuisance class is correctly specified; and (iii) inference-specific bandwidths attain near-nominal pointwise coverage at the largest sample sizes.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.