Maven: a multimodal foundation model for supernova science
Abstract A common setting in astronomy is the availability of a small number of high-quality observations, and larger amounts of either lower-quality observations or synthetic data from simplified models. Time-domain astrophysics is a canonical example of this imbalance, with the number of supernovae observed photometrically outpacing the number observed spectroscopically by multiple orders of magnitude. At the same time, no data-driven models exist to understand these photometric and spectroscopic observables in a common context. Contrastive learning objectives, which have grown in popularity for aligning distinct data modalities in a shared embedding space, provide a potential solution to extract information from these modalities. We present Maven, the first foundation model for supernova science. To construct Maven, we first pre-train our model to align photometry and spectroscopy from 0.5 M synthetic supernovae using a contrastive objective. We then fine-tune the model on 4702 observed supernovae from the Zwicky transient facility. Maven reaches state-of-the-art performance on both classification and redshift estimation, despite the embeddings not being explicitly optimized for these tasks. Through ablation studies, we show that pre-training with synthetic data improves overall performance. In the upcoming era of the Vera C. Rubin observatory, Maven will serve as a valuable tool for leveraging large, unlabeled and multimodal time-domain datasets.
Authors
- Thomas Helfer (ORCID: https://orcid.org/0000-0001-6880-1005)
- Gemma Zhang (ORCID: https://orcid.org/0000-0002-8019-8082)
- V. Ashley Villar (ORCID: https://orcid.org/0000-0002-5814-4061)
- Alexander Gagliano (ORCID: https://orcid.org/0000-0003-4906-8447)
- Siddharth Mishra-Sharma (ORCID: https://orcid.org/0000-0001-9088-7845)
Institutions
- Harvard University (US)
- Center for Astrophysics Harvard & Smithsonian (US)
- The NSF AI Institute for Artificial Intelligence and Fundamental Interactions (US)
- Stony Brook University (US)
- Massachusetts Institute of Technology (US)
Publication Details
- Journal
- Machine Learning Science and Technology
- Published
- 2024-12-01
- DOI
- https://doi.org/10.1088/2632-2153/ad990d
- Citations
- 9
- Primary Topic
- Gamma-ray bursts and supernovae
- Type
- article
- Field-Weighted Citation Impact
- 2.98
Funders
- National Science Foundation
- High Energy Physics