cs.LGMay 16, 2026

Decision-Aware Proximal Bridge Learning for Optimal Treatment Selection

Authors: Tomàs GarrigaAlejandro AlmodóvarAxel BrandoGerard SanzEduard Serrahima de CambraJuan Parras

Organizations: 1Novartis · 2Barcelona Supercomputing Center · 3Universidad Politécnica de Madrid

Abstract

Individualized treatment selection with continuous actions requires accurate causal response estimation in decision-relevant regions, rather than uniformly over the entire action space. Estimating a global causal response surface and then choosing the treatment that maximizes it can therefore be suboptimal, since standard estimation objectives allocate modeling effort according to the observed treatment distribution rather than the regions that determine the optimal decision. While decision-aware approaches have been studied in unconfounded settings, this problem remains underexplored in proximal causal inference, where proxy variables and bridge functions enable identification under suitable assumptions even in the presence of hidden confounding. Despite recent progress, proximal methods have primarily focused on treatment-effect and potential-outcome estimation rather than treatment selection and optimal decision-making. To bridge this gap, we introduce a policy-targeted weighted bridge loss that emphasizes decision-relevant treatment regions while retaining global stabilization. We prove a regret bound showing that the proposed weighted bridge loss controls treatment-selection regret through a weighted ill-posedness constant. We instantiate the framework in decision-aware variants of several proximal bridge solvers, yielding practical algorithms that alternate between weighted bridge estimation, response-surface projection, policy update, and weight refinement. Empirically, we find that decision-aware weighting reduces regret across several bridge solvers, suggesting improved treatment selection in proximal settings.

Explore similar work

CardsList
  1. Doubly Robust Proxy Causal Learning with Neural Mean Embeddings

    May 10, 2026Bariscan Bozkurt, Alexandre Galashov, Dimitri Meunier +3EstimatorsCausal Reasoning

  2. Learning Who to Treat When Treatment is Missing

    Jul 15, 2026Johnna Sundberg, Rayid Ghani, Eli Ben-Michael +1Heterogeneous Treatment EffectsMissing Data