Exploiting Non-Linear Structure in Astronomical Data for Improved Statistical Inference