Starkindler: An Uncertainty Aware Objective for Photometric Redshift Estimation

Abstract: Photometric Redshift is critical for analyzing astronomical objects, but existing ML methods often overlook the aleatoric uncertainties inherent in observed data. We introduce Starkindler, a novel training objective that explicitly incorporates observational errors into the model's objective function, thereby directly accounting for aleatoric uncertainty. Unlike traditional probabilistic models that focus solely on epistemic uncertainty, Starkindler provides uncertainty estimates that are regularised by aleatoric uncertainty, and is designed to be more interpretable. We train a simple convolutional neural network (CNN) using data from Sloan Digital Sky Survey (SDSS) and compare against the Photometric redshift estimates provided by SDSS. We show improvements in accuracy, calibration and reduction in predicted outlier rate. We also conduct an ablation study which confirms that excluding observational errors significantly degrades model performance, underscoring the importance of accounting for aleatoric uncertainty. Our results suggest that Starkindler not only enhances predictive performance but also provides interpretable uncertainty estimates, making it a robust tool for astronomical data analysis.