Information preserving space warping functions to prevent affine collapse in neural networks