A causal fused lasso for interpretable heterogeneous treatment effects estimation
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2021
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866912526024835072 |
|---|---|
| author | Padilla, Oscar Hernan Madrid Chen, Yanzhen Padilla, Carlos Misael Madrid Ruiz, Gabriel |
| author_facet | Padilla, Oscar Hernan Madrid Chen, Yanzhen Padilla, Carlos Misael Madrid Ruiz, Gabriel |
| contents | We propose a novel method for estimating heterogeneous treatment effects based on the fused lasso. By first ordering samples based on the propensity or prognostic score, we match units from the treatment and control groups. We then run the fused lasso to obtain piecewise constant treatment effects with respect to the ordering defined by the score. Similar to the existing methods based on discretizing the score, our methods yield interpretable subgroup effects. However, existing methods fixed the subgroup a priori, but our causal fused lasso forms data-adaptive subgroups. We show that the estimator consistently estimates the treatment effects conditional on the score under very general conditions on the covariates and treatment. We demonstrate the performance of our procedure using extensive experiments that show that it can be interpretable and competitive with state-of-the-art methods. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2110_00901 |
| institution | arXiv |
| publishDate | 2021 |
| record_format | arxiv |
| spellingShingle | A causal fused lasso for interpretable heterogeneous treatment effects estimation Padilla, Oscar Hernan Madrid Chen, Yanzhen Padilla, Carlos Misael Madrid Ruiz, Gabriel Methodology We propose a novel method for estimating heterogeneous treatment effects based on the fused lasso. By first ordering samples based on the propensity or prognostic score, we match units from the treatment and control groups. We then run the fused lasso to obtain piecewise constant treatment effects with respect to the ordering defined by the score. Similar to the existing methods based on discretizing the score, our methods yield interpretable subgroup effects. However, existing methods fixed the subgroup a priori, but our causal fused lasso forms data-adaptive subgroups. We show that the estimator consistently estimates the treatment effects conditional on the score under very general conditions on the covariates and treatment. We demonstrate the performance of our procedure using extensive experiments that show that it can be interpretable and competitive with state-of-the-art methods. |
| title | A causal fused lasso for interpretable heterogeneous treatment effects estimation |
| topic | Methodology |
| url | https://arxiv.org/abs/2110.00901 |