Strobl 2008 - Simulation Results for Conditional Importance

Summary

Over 500 replications of the 12-predictor design, marginal permutation importance does not reproduce the true coefficient pattern. It strongly overstates the correlated predictors –, most of all at small mtry. Conditional importance, with = all variables correlated with at , does much better. The correlated () move much closer to the uncorrelated (), while fall. The text calls the result “almost equal”, but the digitized Figure 3 medians are still about 1.5–2× higher. Correlated variables with small or zero coefficients are still overstated, but much less. Conditional importance also has lower variance at every mtry. Variability rises with mtry.

Overview

This note records §3 and Figure 3. The design is in ^ex-sim-design. The two scores compared are the marginal permutation importance (^def-vi-forest) and the conditional one (^alg-cpi).

Main Content

Figure 3 (“Permutation importance”): median permutation importance over 500 replications for the marginal scheme (dashed) and the conditional scheme (solid, per the caption; in the published image both lines are in fact dashed and the schemes are distinguished by colour, red squares vs blue circles; see Strobl 2008 - Digitized Figure Data for Selection Rates and Importance), with inter-quartile ranges, for each of the 12 variables. There is one panel for each of mtry = 1, 3, 8, each with its own y-scale (0-30, 0-60, 0-100). The caption notes that the ordering of variables on the plot is arbitrary.

Settings. 500 iterations. Conditioning set: all variables with empirical correlation with . In this design that means – condition on each other, and – have, in expectation, no conditioning variables.

Finding 1: marginal importance misrepresents the DGP (§3, para. 2)

Under the ordinary permutation scheme, “the importance scores of the correlated predictor variables are highly overestimated.” The coefficient pattern is not recovered. The effect is largest for small mtry, because correlated variables are more likely to reach top positions in a tree when their correlated competitors are not available as split candidates.

Finding 2: conditional importance tracks the true pattern better (§3, para. 3)

  1. (correlated, ) “show an almost equal level of importance” as (uncorrelated, ). Fig. 3 readings are 16/20/31 vs 9/13/16 at mtry = 1/3/8, so closer than under marginal permutation but not equal.
  2. () and (), which are correlated, decrease.
  3. For small and zero coefficients, correlated variables still score above their uncorrelated counterparts ( vs , vs ). The overestimation remains but is “to a much lesser extent”. The proposed reason is that uncorrelated variables are chosen less often and lower in trees (Fig. 1), so they have less chance to earn a high importance.
  4. The remaining preference for correlated variables depends on mtry and is strongest at small mtry.

Finding 3: variance and the mtry trade-off (§3, para. 3)

  • The variability of the importance increases with mtry, while prediction accuracy is expected to be higher at smaller mtry.
  • Within each mtry level, conditional importance has lower variability than unconditional importance.

Finding 4: two different tasks (§3, para. 4)

  • Picking influential variables out of a correlated set (the Archer & Kimes / Nicodemus & Shugart task). Comparing – with , conditional importance shows the same pattern as marginal importance with notably smaller variation, which may improve identification.
  • Comparing correlated with uncorrelated predictors. Conditional importance is “much better suited” than the original.
  • For piecewise-constant response functions, which recursive partitioning fits more easily, the benefit of conditioning is “even stronger than presented here”.

Summary table: text statements vs figure readings

Column “§3 text” gives only what the text says. Cells marked (inferred) are not stated in §3 and come from §2.4 logic or the design. The numbers are medians digitized from Figure 3 for mtry = 1 / 3 / 8 (^ex-fig3-values).

VariableCorrelated?§3 text: marginal§3 text: conditionalFig. 3 marginal medianFig. 3 conditional median
5yes”highly overestimated""almost equal level” to 29 / 50 / 7016 / 20 / 31–33
2yes”highly overestimated”decreases; still overestimated vs uncorrelated27 / 35 / 1714 / 12 / 8
0yes”highly overestimated”decreases; still overestimated vs uncorrelated26 / 27 / 713 / 8 / 3
−5no(inferred: not discussed)reference level for ≈9 / 13 / 169 / 13 / 16
−2no(inferred: not discussed separately)the “uncorrelated” comparator for small coefficients≈1.6 / 1.2 / 0.41.6 / 1.2 / 0.4
–0no(inferred: not discussed separately)the “uncorrelated” comparator for zero coefficients≈0≈0

The mtry grid. The panels of Figure 3 (and Figure 1) are mtry = 1, 3 and 8. The text never gives these values.

A discrepancy between text and figure. The digitized conditional medians for vs are 16 vs 9, 20 vs 13 and 31 vs 16. Conditioning shrinks the correlated/uncorrelated ratio from about 3.2–4.4× to about 1.5–2×, but the two are not “almost equal” as §3 para. 3 says. Also, by the ratio the marginal inflation of the strong correlated drivers grows with mtry. The §3 statement “most pronounced for small values of mtry” fits the weak and null correlated variables: the marginal are near at mtry = 1 and fall back at mtry = 8. See ^thm-fig3-readings.

Examples

Reading the result as a validation check (generic)

Before trusting a driver ranking from a tree ensemble on correlated marketing inputs, rerun this kind of matched-pair simulation with your own correlation structure, sample size (e.g. 104 weekly rows) and mtry. Plant a zero-effect variable that correlates with a strong channel. If its marginal importance clearly exceeds that of an independent small-effect channel, rank comparisons between correlated and uncorrelated channels are not reliable in that setting.

Connections

See Also