• Home
  • Search
  • Learning Subgroups with Maximum Treatment Effects Without Causal Heuristics
  • https://doi.org/10.1609/aaai.v40i32.39976Copy DOI Icon

Learning Subgroups with Maximum Treatment Effects Without Causal Heuristics

  • Mar 14, 2026
  • Lincen Yang +3 more
Show More
  • Abstract
  • Literature Map
  • Similar Papers
Abstract

Discovering subgroups with the maximum average treatment effect is crucial for targeted decision making in domains such as precision medicine, public policy, and education. While most prior work is formulated in the potential‑outcome framework, the corresponding structural causal model (SCM) for this task has been largely overlooked. In practice, two approaches dominate. The first estimates pointwise conditional treatment effects and then fits a tree on those estimates, effectively turning subgroup estimation into the harder problem of accurate pointwise estimation. The second constructs decision trees or rule sets with ad‑hoc 'causal' heuristics, typically without rigorous justification for why a given heuristic may be used or whether such heuristics are necessary at all. We address these issues by studying the problem directly under the SCM framework. Under the assumption of a partition-based model, we show that optimal subgroup discovery reduces to recovering the data-generating models and hence a standard supervised learning problem (regression or classification). This allows us to adopt any partition-based methods to learn the subgroup from data. We instantiate the approach with CART, arguably one of the most widely used tree-based method, to learn the subgroup with maximum treatment effect. Finally, on a large collection of synthetic and semi‑synthetic datasets, we compare our method against a wide range of baselines and find that our approach, which avoids such causal heuristics, more accurately identifies subgroups with maximum treatment effect.

Similar Papers
  • Research Article
  • Citations5

Evaluation of the effects of a dexmedetomidine-midazolam-ketamine combination administered intramuscularly to captive red-footed tortoises (Chelonoidis carbonaria).

  • Nov 01, 2021
  • American Journal of Veterinary Research
  • David Eshar +3
  • Conference Article
  • Citations2

A weight-based exacerbation dose-response analysis of mepolizumab in severe asthma with eosinophilic phenotype

  • Sep 01, 2016
  • Isabelle Pouliquen +3
  • Research Article
  • Citations16

Suicide and violence assessment in psychiatry

  • Nov 01, 1996
  • General Hospital Psychiatry
  • Douglas H Hughes
  • PDF
  • Research Article
  • Citations1

COMPARATIVE EVALUATION OF SURGICAL CORRECTION METHODS OF VESICO-URETERIC REFLUX IN CHILDREN

  • Apr 01, 2014
  • Current pediatrics
  • S P Yatsyk +5
  • Single Report
  • Citations33

The Impact of Group Diversity on Performance and Knowledge Spillover -- An Experiment in a College Classroom

  • May 01, 2006
  • National Bureau of Economic Research
  • Zeynep Hansen +2
  • Research Article
  • Citations2

Comparative effectiveness of different chemotherapy regimens of advanced-stage Hodgkin lymphoma in adults: a network meta-analysis

  • Nov 22, 2018
  • Cancer Management and Research
  • Tingting Zhang +10
  • Front Matter
  • Citations10

Malocclusion and treatment need in United States youths 12 to 17 years of age

  • Nov 01, 1977
  • American Journal of Orthodontics
  • J.A Salzmann
  • Research Article
  • Citations5

Bayesian identification of structural coefficients in causal models and the causal false-positive risk of confounders and colliders in linear Markovian models

  • Feb 27, 2022
  • BMC Medical Research Methodology
  • Riko Kelter
  • Research Article

Axiomatization of interventional probability distributions

  • Aug 21, 2024
  • Biometrika
  • Kayvan Sadeghi +1
  • Research Article
  • Citations16

Data-driven root cause analysis via causal discovery using time-to-event data

  • Mar 04, 2024
  • Computers & Industrial Engineering
  • Henry M Mbogu +1
  • PDF
  • Research Article

Causality and tractable probabilistic models

  • Jan 08, 2024
  • Frontiers in Computer Science
  • David Cruz +1
  • Supplementary Content

Causal AI in Cardiac Arrhythmia: From Pattern Recognition to Mechanistic Insight

  • Apr 01, 2026
  • Clinical Cardiology
  • Bara Abubaha +11
  • Conference Article
  • Citations1

Causal-aware Large Language Models: Enhancing Decision-Making Through Learning, Adapting and Acting

  • Sep 01, 2025
  • Wei Chen +7
  • Conference Article
  • Citations6

Causal-aware Large Language Models: Enhancing Decision-Making Through Learning, Adapting and Acting

  • Aug 01, 2024
  • Wei Chen +7
  • Research Article
  • Citations6

Synthesizing decision rules from multiple information sources: a neighborhood granulation viewpoint

  • Jan 27, 2018
  • International Journal of Machine Learning and Cybernetics
  • Yaojin Lin +5
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.