用迁移学习让天文瞬变源分类模型跨数据集、跨观测站快速适配,大幅减少标注数据需求。
Transfer Learning for Transient Classification: From Simulations to Real Data and ZTF to LSST
- 从模拟或旧数据集训练的模型出发,通过迁移学习适配新数据。
- 真实ZTF数据标注量只需原需求的5%,性能不变;LSST训练数据只需30%即达94%基准性能。
- 适合即将启动的大型巡天项目(如LSST)快速部署自动化分类系统。
机器学习在自动分类天文瞬变源中已至关重要,但现有方法存在明显局限:基于模拟数据训练的分类器在真实数据上表现不佳,针对某一巡天设计的模型难以迁移到其他巡天,而新巡天又需海量标注数据。随着薇拉·C·鲁宾天文台的遗产空间与时间巡天(LSST)临近,现有分类模型需重新训练。本文证明,迁移学习可克服上述挑战:以模拟的兹威基瞬变设施(ZTF)光变曲线训练的模型为基础,通过迁移学习仅需5%的真实ZTF标注数据即可保持与从头训练相当的性能;将ZTF模型用于适配LSST模拟数据时,仅用30%训练数据即可达到94%的基准性能。这表明,LSST早期运行阶段即可实现可靠自动化分类,无需等待数月甚至数年积累足够训练数据。
原文摘要 · Abstract (English)
Machine learning has become essential for automated classification of astronomical transients, but current approaches face significant limitations: classifiers trained on simulations struggle with real data, models developed for one survey cannot be easily applied to another, and new surveys require prohibitively large amounts of labelled training data. These challenges are particularly pressing as we approach the era of the Vera C. Rubin Observatory's Legacy Survey of Space and Time (LSST), where existing classification models will need to be retrained using LSST observations. We demonstrate that transfer learning can overcome these challenges by repurposing existing models trained on either simulations or data from other surveys. Starting with a model trained on simulated Zwicky Transient Facility (ZTF) light curves, we show that transfer learning reduces the amount of labelled real ZTF transients needed by 95% while maintaining equivalent performance to models trained from scratch. Similarly, when adapting ZTF models for LSST simulations, transfer learning achieves 94% of the baseline performance while requiring only 30% of the training data. These findings have significant implications for the early operations of LSST, suggesting that reliable automated classification will be possible soon after the survey begins, rather than waiting months or years to accumulate sufficient training data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。