Generalizable Data-Driven Model Augmentations Using Learning And Inference Assisted By Feature-Space Engineering