Machine Learning

Causal Rule Sets for Identifying Subgroups with Enhanced Treatment Effect

Tagged: ,

This topic contains 0 replies, has 1 voice, and was last updated by  arXiv 1 year, 9 months ago.


  • arXiv
    5 pts

    Causal Rule Sets for Identifying Subgroups with Enhanced Treatment Effect

    We introduce a novel generative model for interpretable subgroup analysis for causal inference applications, Causal Rule Sets (CRS). A CRS model uses a small set of short rules to capture a subgroup where the average treatment effect is elevated compared to the entire population. We present a Bayesian framework for learning a causal rule set. The Bayesian framework consists of a prior that favors simpler models and a Bayesian logistic regression that characterizes the relation between outcomes, attributes and subgroup membership. We find maximum a posteriori models using discrete Monte Carlo steps in the joint solution space of rules sets and parameters. We provide theoretically grounded heuristics and bounding strategies to improve search efficiency. Experiments show that the search algorithm can efficiently recover a true underlying subgroup and CRS shows consistently competitive performance compared to other state-of-the-art baseline methods.

    Causal Rule Sets for Identifying Subgroups with Enhanced Treatment Effect
    by Tong Wang, Cynthia Rudin
    https://arxiv.org/pdf/1710.05426v1.pdf

You must be logged in to reply to this topic.