Differential Subgroup Discovery: Characterizing Where Two Populations Differ, and Why
Not on the map. Why?
Abstract (arXiv)
We study the problem of understanding where two populations differ within a feature space, which we formalize in the concept of a differential subgroup: a subset of individuals from both populations who, despite sharing similar characteristics, exhibit exceptional differences in a target outcome. Differential subgroups reveal the regions of the feature space where population-level gaps are most pronounced and can help practitioners identify the covariate combinations that are structurally responsible for these differences, e.g.~in clinical analysis, model diagnostics, or treatment-effect studies. We introduce a general optimization objective for discovering differential subgroups and establish conditions under which the resulting subgroups admit a causal interpretation of population differences. We propose DiffSub, a gradient-based approach that discovers interpretable differential subgroups in tabular data. Across synthetic benchmarks, medical case studies, model-error analyses, and treatment-effect settings, DiffSub identifies informative subgroups that reveal where population differences arise and why.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 46 references that are in the map. First posted on arXiv 30 Apr 2026.
Builds on
11 of 46 references.
References · 11
- 2019A Survey on Bias and Fairness in Machine Learningcited by 1,911
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
- 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 408
- 2017Preventing Fairness Gerrymandering: Auditing and Learning for Subgroup Fairnesscited by 381
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 254
- 2018Multiaccuracy: Black-Box Post-Processing for Fairness in Classificationcited by 221
- 2020CausalML: Python Package for Causal Machine Learningcited by 74
- 2022On Learning Fairness and Accuracy on Multiple Subgroupscited by 31
- 2016Model-based Recursive Partitioning for Subgroup Analysescited by 21
- 2022Feature Selection for Discovering Distributional Treatment Effect Modifierscited by 7
- 2024Learning Exceptional Subgroups by End-to-End Maximizing KL-divergencecited by 5
Built on by
Not cited yet.