Fig. 2

Heterogeneous treatment effects of education attainment (A), smoking status (B), and alcohol use disorder (C) on risk of all-cause dementia. The predicted individualized treatment effect (ITE) was presented as an absolute risk difference in risk of dementia between exposure and non-exposure for each participant (Y-axis). The predicted ITE is divided into 10 groups according to the deciles (X-axis). Different colors represent participants in distinct subgroups identified by the single decision tree model