Combining datasets with different ground truths using Low-Rank Adaptation to generalize image-based CNN models for photometric redshift prediction

TL;DR

This paper demonstrates how Low-Rank Adaptation (LoRA) can effectively combine galaxy imaging datasets with different ground truths to improve photometric redshift estimation in CNN models, offering a computationally efficient fine-tuning approach.

Contribution

The study introduces LoRA as a novel method for combining datasets with different ground truths in astrophysics, improving model accuracy without full retraining.

Findings

LoRA reduces bias by approximately 2.5 times compared to traditional transfer learning.

LoRA decreases scatter by about 2.2 times.

Combined dataset retraining yields better generalization but requires more computation.

Abstract

In this work, we demonstrate how Low-Rank Adaptation (LoRA) can be used to combine different galaxy imaging datasets to improve redshift estimation with CNN models for cosmology. LoRA is an established technique for large language models that adds adapter networks to adjust model weights and biases to efficiently fine-tune large base models without retraining. We train a base model using a photometric redshift ground truth dataset, which contains broad galaxy types but is less accurate. We then fine-tune using LoRA on a spectroscopic redshift ground truth dataset. These redshifts are more accurate but limited to bright galaxies and take orders of magnitude more time to obtain, so are less available for large surveys. Ideally, the combination of the two datasets would yield more accurate models that generalize well. The LoRA model performs better than a traditional transfer learning method, with $\sim2.5\times$ less bias and $\sim$2.2$\times$ less scatter. Retraining the model on a combined dataset yields a model that generalizes better than LoRA but at a cost of greater computation time. Our work shows that LoRA is useful for fine-tuning regression models in astrophysics by providing a middle ground between full retraining and no retraining. LoRA shows potential in allowing us to leverage existing pretrained astrophysical models, especially for data sparse tasks.