arXiv:2502.02717astro-ph.IMcs.AI2025-02被引 2

Astromer 2 用自监督学习提升光变曲线嵌入,小样本下表现远超前代。

Astromer 2

  • 基于150万条光变曲线自监督预训练,通过预测掩码观测来学习特征。
  • 在20~500样本/类的小数据集上,F1得分比Astromer 1提升15%。
  • 融合注意力层中间表示的加权嵌入方法,显著增强泛化能力。

基础模型已成为深度学习的重要范式,能够从大规模数据中学习鲁棒表征并广泛应用于分类等下游任务。本文提出专为光变曲线嵌入设计的Astromer 2,是自监督光变曲线分析模型的升级版本。该模型在150万条来自MACHO巡天的单波段光变曲线上进行预训练,采用自监督任务预测序列中的随机掩码观测。通过在小规模标注数据集上微调,评估其在分类任务中的表现。嵌入质量通过基于Astromer生成嵌入的MLP分类器的F1分数衡量。结果表明,Astromer 2在所有测试场景(每类20、100、500样本)中均显著优于Astromer 1。特别地,使用融合注意力块中间表示的加权样本嵌入方法效果突出。在ATLAS数据集上,其F1分数相较先前模型提升15%,展现出对新数据集的强大泛化能力。该性能在极小标注数据条件下尤为显著,凸显Astromer 2在高效、可扩展光变曲线分析中的潜力。

原文摘要 · Abstract (English)

Foundational models have emerged as a powerful paradigm in deep learning field, leveraging their capacity to learn robust representations from large-scale datasets and effectively to diverse downstream applications such as classification. In this paper, we present Astromer 2 a foundational model specifically designed for extracting light curve embeddings. We introduce Astromer 2 as an enhanced iteration of our self-supervised model for light curve analysis. This paper highlights the advantages of its pre-trained embeddings, compares its performance with that of its predecessor, Astromer 1, and provides a detailed empirical analysis of its capabilities, offering deeper insights into the model's representations. Astromer 2 is pretrained on 1.5 million single-band light curves from the MACHO survey using a self-supervised learning task that predicts randomly masked observations within sequences. Fine-tuning on a smaller labeled dataset allows us to assess its performance in classification tasks. The quality of the embeddings is measured by the F1 score of an MLP classifier trained on Astromer-generated embeddings. Our results demonstrate that Astromer 2 significantly outperforms Astromer 1 across all evaluated scenarios, including limited datasets of 20, 100, and 500 samples per class. The use of weighted per-sample embeddings, which integrate intermediate representations from Astromer's attention blocks, is particularly impactful. Notably, Astromer 2 achieves a 15% improvement in F1 score on the ATLAS dataset compared to prior models, showcasing robust generalization to new datasets. This enhanced performance, especially with minimal labeled data, underscores the potential of Astromer 2 for more efficient and scalable light curve analysis.

光变曲线自监督嵌入学习天体物理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。