# Estimation of Optimal Individualized Treatment Rules Using a Covariate-Specific Treatment Effect Curve with High-dimensional Covariates

## TL;DR

This paper introduces a semi-parametric method to estimate covariate-specific treatment effects and optimal individualized treatment rules in high-dimensional settings, enabling personalized medicine with confidence quantification.

## Contribution

It proposes a flexible, high-dimensional covariate modeling approach with confidence bands for treatment effect estimation and personalized treatment decision-making.

## Key findings

- Accurately estimates covariate-specific treatment effects in high-dimensional data.
- Provides simultaneous confidence bands for treatment effect curves.
- Enables identification of patient subgroups benefiting from specific treatments.

## Abstract

With a large number of baseline covariates, we propose a new semi-parametric modeling strategy for heterogeneous treatment effect estimation and individualized treatment selection, which are two major goals in personalized medicine. We achieve the first goal through estimating a covariate-specific treatment effect (CSTE) curve modeled as an unknown function of a weighted linear combination of all baseline covariates. The weight or the coefficient for each covariate is estimated by fitting a sparse semi-parametric logistic single-index coefficient model. The CSTE curve is estimated by a spline-backfitted kernel procedure, which enables us to further construct a simultaneous confidence band (SCB) for the CSTE curve under a desired confidence level. Based on the SCB, we find the subgroups of patients that benefit from each treatment, so that we can make individualized treatment selection. The innovations of the proposed method are three-fold. First, the proposed method can quantify variability associated with the estimated optimal individualized treatment rule with high-dimensional covariates. Second, the proposed method is very flexible to depict both local and global associations between the treatment and baseline covariates in the presence of high-dimensional covariates, and thus it enjoys flexibility while achieving dimensionality reduction. Third, the SCB achieves the nominal confidence level asymptotically, and it provides a uniform inferential tool in making individualized treatment decisions.