Multivariate Time-series Transformer Embeddings for Light Curves

TL;DR

This paper introduces a multiband transformer model for astronomical light curves that effectively fuses data across multiple wavelengths, improving classification performance over single-band models.

Contribution

The paper proposes a novel fusion mechanism for multiband light curves within transformer models, enabling better cross-wavelength information integration for astronomical data analysis.

Findings

Multiband models outperform single-band models by ~10% in F1-score.

Joint pre-training of multiband encoders yields further performance gains.

Asynchronous sampling of bands has minimal impact on multiband model performance.

Abstract

Astronomical surveys produce time-series data by observing stellar objects across multiple wavelength bands. Foundational transformer-based models, such as Astromer, encode each time-series as a sequence of embeddings of uniform dimensions. However, such models operate independently on each band at a single time and do not natively leverage information across telescope filters. We extend this framework by introducing a fusion mechanism that maps the collection of single-band embeddings to a unified sequence representation, enabling multiband modeling for downstream tasks. The challenge lies in devising a mechanism within the encoder to coordinate between data from different wavelengths, which are often recorded at asynchronous times. We pre-train multiband models on a subset of 600000 high signal-to-noise light curves from the MACHO survey and fine-tune them using the Alcock and ATLAS survey datasets. Experimental results show that both our proposed multiband architectures outperform the single-band models by approximately 10% in F1-score, with jointly pre-trained multiband encoders further improving performance over a collection of independently pre-trained single-band encoders. Furthermore, our experiments show that there are minimal differences in multiband performance when sampling individual band data asynchronously versus sampling all individual bands on the same set of time-steps. However, jointly pre-trained models can take more than twice the time to pre-train. These results demonstrate the trade-offs of the multiband approach where multivariate data are available.