Subgroup identification in clinical trials via the predicted individual treatment effect