An effective framework for estimating individualized treatment rules

Neural Information Processing Systems 

Estimating individualized treatment rules (ITRs) is fundamental in causal inference, particularly for precision medicine applications. Traditional ITR estimation methods rely on inverse probability weighting (IPW) to address confounding factors and L_{1} -penalization for simplicity and interpretability. However, IPW can introduce statistical bias without precise propensity score modeling, while L_1 -penalization makes the objective non-smooth, leading to computational bias and requiring subgradient methods. In this paper, we propose a unified ITR estimation framework formulated as a constrained, weighted, and smooth convex optimization problem. The optimal ITR can be robustly and effectively computed by projected gradient descent.