Dorie 2019 - Simulation Design and Knobs

Summary

Data sets were simulated from 58 real covariates (4,802 children, Collaborative Perinatal Project) with randomly generated assignment mechanisms and response surfaces controlled by six “knobs”: degree of nonlinearity, percentage treated, overlap, alignment, treatment effect heterogeneity and effect magnitude. Of 216 possible knob combinations the organisers kept 77 non-trivial settings, each replicated 100 times for the black-box competition (7,700 data sets), with 20 settings used once each for the DIY competition.

Overview

The DGP is defined as a model over DGPs: because the functions are generated randomly, the parameters controlling the framework define a distribution over data-generating processes, and a specific value of those parameters is a “knob” setting.

Main Content

Real-data calibration (Sec. 4.2)

  • Source: Collaborative Perinatal Project (Niswander and Gordon 1972), a longitudinal study of pregnant women and their children 1959 to 1974; public data on over 55,000 pregnancies with over 6,500 variables.
  • Hypothetical study: a twins study of the impact of birth weight on a child’s IQ; confounders chosen as a researcher might choose.
  • Complete cases: 4,802 observations and 58 covariates: 3 categorical, 5 binary, 27 count, 23 continuous.

Model structure (Sec. 4.3)

ACIC data-generating factorisation

The joint distribution of potential outcomes and treatment given covariates factors as

which encodes ignorability because . Both pieces are generalised additive functions: covariates pass through transformations (polynomial terms, indicators, step functions) that are added or multiplied, e.g. , and the sum may pass through a link such as .

Generation details (A.1). Coefficients are drawn from Student- distributions when unbounded and beta-prime when strictly positive. Covariates are scaled to about ; sub-function locations and scales are generated to map to the same region. Terms are rescaled so propensity scores lie almost exclusively in to and response variables have mean and expected SD . The treated and control response surfaces are adjusted to meet a generated average treatment effect; the noise and treatment effect are generated from heavy-tailed distributions.

The six knobs

Knobs and their implementation

KnobMeaningImplementation (Table 2 columns and A.1 bullets)Oracle/diagnostic metric (with reported range)
1. Degree of nonlinearityNonlinearity in response and assignmentTreatment model : linear adds predictors as linear terms with random coefficients; polynomial may add quadratic or cubic terms for continuous predictors; step adds jumps and kinks . Response model ; exponential contains the polynomial condition and adds one term . Up to three-way interactions and third-order polynomials of on observed untransformed : range to , quartiles
2. Percentage treatedBaseline fraction treatedLow (about 35% expected) or high (about 65%)Low: 95% of sims in ; high: (range differs due to overlap)
3. OverlapControls dissimilar to all treatedpenalize vs full: when not full, a large negative penalty term in forces propensity near in a corner of covariate space (B, C from marginal quantiles)Mean Mahalanobis distance to nearest opposite-arm neighbour in ground-truth covariate space: quartiles (overlap knob = 1) and (knob = 0)
4. AlignmentOverlap in terms between assignment and response (true confounders)Marginal probability that an assignment term is copied to the response: low about 25%, high about 75%, or noneabsolute correlation between and , from about to , median
5. Treatment effect heterogeneityNon-parallel response surfacesnone: treatment is a single additive term; low: treatment interacts with about three response terms; high: about sixSD of normalised by SD of outcome: to , quartiles ; exactly for 200 realizations with knob 0
6. Magnitude of effectSize of SATTNo explicit knob; varies implicitly with other knobs; participants told it variesMedian SATT across 7,700 realizations SD, IQR to

Imbalance. Not a knob, but arises from the assignment mechanism. Oracle imbalance metric: Euclidean norm of the distance between control and treated covariate means in ground-truth space, quartiles . Lack of overlap implies lack of balance, but imbalance can exist with perfect overlap.

Why alignment matters. Only covariates in both the assignment mechanism and the response surface cause bias. A covariate in only one affects efficiency, not bias. Functional form matters: if age enters assignment linearly but the response linearly and squared, including linear age removes bias and the squared term is not needed; if the squared term is in both, conditioning on it is required. Following Kern et al. (2016), alignment reflects the dimension of the confounder space. Practical implications: more potential bias for a given covariate set; difficulty of finding true confounders when they are few relative to the covariates; methods privileging predictors of only treatment or only response are disadvantaged.

Scenario counts

All knob combinations yield 216 scenarios; the organisers focused on the most interesting 77 by removing trivial or redundant cases such as simple linear models or unconfounded treatment and response. Each of the 77 settings has 100 independent replications, giving 7,700 realizations. For the DIY competition, 20 settings (marked in Table 2) were used, one data set each.

Table 2: the 77 simulation settings (Appendix A.1)

Columns as printed. “DIY #” gives the DIY data-set number for settings 1-20.

#Trt modelTrt %OverlapResponse modelTrt/Rsp alignmentHeterogeneityDIY #
1linearlowpenalizelinearhighhigh10
2polynomiallowpenalizeexponentialhighnone1
3linearlowpenalizelinearhighnone9
4polynomiallowfullexponentialhighhigh4
5linearlowpenalizeexponentialhighhigh15
6polynomiallowpenalizelinearhighhigh2
7polynomiallowpenalizeexponentialhighhigh5
8polynomiallowpenalizeexponentialnonehigh13
9steplowpenalizestephighhigh8
10linearlowpenalizeexponentiallowhigh14
11polynomiallowpenalizelinearlowhigh19
12polynomiallowpenalizeexponentiallowhigh12
13linearhighpenalizeexponentialhighhigh18
14polynomialhighpenalizelinearhighhigh20
15polynomialhighpenalizeexponentialhighhigh6
16polynomialhighpenalizeexponentialnonehigh17
17stephighpenalizestephighhigh7
18linearhighpenalizeexponentiallowhigh3
19polynomialhighpenalizelinearlowhigh16
20polynomialhighpenalizeexponentiallowhigh11
21polynomiallowpenalizesteplowlow
22polynomiallowpenalizesteplowhigh
23polynomiallowpenalizestephighlow
24polynomiallowpenalizestephighhigh
25polynomiallowpenalizeexponentiallowlow
26polynomiallowpenalizeexponentialhighlow
27polynomiallowfullsteplowlow
28polynomiallowfullsteplowhigh
29polynomiallowfullstephighlow
30polynomiallowfullstephighhigh
31polynomiallowfullexponentiallowlow
32polynomiallowfullexponentiallowhigh
33polynomiallowfullexponentialhighlow
34polynomialhighpenalizesteplowlow
35polynomialhighpenalizesteplowhigh
36polynomialhighpenalizestephighlow
37polynomialhighpenalizestephighhigh
38polynomialhighpenalizeexponentiallowlow
39polynomialhighpenalizeexponentialhighlow
40polynomialhighfullsteplowlow
41polynomialhighfullsteplowhigh
42polynomialhighfullstephighlow
43polynomialhighfullstephighhigh
44polynomialhighfullexponentiallowlow
45polynomialhighfullexponentiallowhigh
46polynomialhighfullexponentialhighlow
47polynomialhighfullexponentialhighhigh
48steplowpenalizesteplowlow
49steplowpenalizesteplowhigh
50steplowpenalizestephighlow
51steplowpenalizeexponentiallowlow
52steplowpenalizeexponentiallowhigh
53steplowpenalizeexponentialhighlow
54steplowpenalizeexponentialhighhigh
55steplowfullsteplowlow
56steplowfullsteplowhigh
57steplowfullstephighlow
58steplowfullstephighhigh
59steplowfullexponentiallowlow
60steplowfullexponentiallowhigh
61steplowfullexponentialhighlow
62steplowfullexponentialhighhigh
63stephighpenalizesteplowlow
64stephighpenalizesteplowhigh
65stephighpenalizestephighlow
66stephighpenalizeexponentiallowlow
67stephighpenalizeexponentiallowhigh
68stephighpenalizeexponentialhighlow
69stephighpenalizeexponentialhighhigh
70stephighfullsteplowlow
71stephighfullsteplowhigh
72stephighfullstephighlow
73stephighfullstephighhigh
74stephighfullexponentiallowlow
75stephighfullexponentiallowhigh
76stephighfullexponentialhighlow
77stephighfullexponentialhighhigh

Reading the table. Settings 1-20 (the DIY sets) mix linear, polynomial and step models with mostly penalize overlap and high heterogeneity; only setting 4 in the DIY block uses full overlap. Settings 21-77 are a factorial-style exploration with polynomial and step treatment models. Counts from the table: treatment model polynomial 39, step 32, linear 6; low-treated 40, high-treated 37; overlap penalized 45, full 32; response exponential 39, step 32, linear 6; alignment high 39, low 36, none 2; heterogeneity high 43, low 32, none 2.

Examples

Alignment knob in miniature

Setup. Assignment uses covariates ; the response copies each assignment term with probability (low) or (high). High alignment. Most assignment terms also drive , so most are true confounders; omitting or mis-modelling them biases the estimate heavily. Low alignment. Only about one in four is a confounder; flexible methods that can identify which terms matter gain relative to those that adjust for everything or select on the assignment model alone. Interpretation. In the paper, poor alignment plus response nonlinearity were “almost completely predictive of poor performance” (Dorie 2019 - Predicting Performance and Variance Decomposition). A runnable toy is in Dorie 2019 - Toy ACIC-Style Simulation in Python.

Connections

See Also