用低秩适配融合不同红移数据,提升星系图像模型泛化能力
Combining datasets with different ground truths using Low-Rank Adaptation to generalize image-based CNN models for photometric redshift prediction
- 用低秩适配(LoRA)微调预训练模型,融合光谱与测光红移数据
- 相比传统迁移学习,偏差降低2.5倍,散度降低2.2倍
- 适合数据稀疏的天体物理任务,节省计算成本
本文展示如何利用低秩适配(LoRA)整合不同红移标注的星系成像数据集,以提升用于宇宙学光谱红移预测的卷积神经网络模型性能。先在包含宽泛星系类型但精度较低的测光红移数据集上训练基础模型,再使用更精确但仅限明亮星系且获取耗时极长的光谱红移数据集通过LoRA进行微调。理想情况下,结合两者可获得更准确且泛化能力更强的模型。结果表明,LoRA方法优于传统迁移学习,偏差降低约2.5倍,散度降低约2.2倍;而直接在合并数据集上重新训练虽表现更优,但计算开销更大。本工作证明,LoRA可作为天体物理回归任务中高效微调的中间方案,有助于复用已有预训练模型,尤其适用于数据稀缺场景。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。