Improving Generalization in Survival Models