Exploiting Heterogeneity For Robust Policy Learning And Prediction Under Interventions