edit: yep - predicting the original based on the augmented version (as a hack to generalize for unseen inputs)