Generalizing Orthogonalization for Models with Non-Linearities