Linear space transform transformation based on covariance?

Question

I have a linear space of n dimensions with non-overlapping groups characterized by different variation (different covariance matrices). Is there a way to deform non-linearly the space according to an optimization criterion in order to get a new, non-linear space where the variation of these group would be isotropic?

Here is a visual description of the problem

Thank you!