Surya: Foundation Model for Heliophysics

  • 2025-08-21 16:53:36
  • Sujit Roy, Johannes Schmude, Rohit Lal, Vishal Gaur, Marcus Freitag, Julian Kuehnert, Theodore van Kessel, Dinesha V. Hegde, Andrés Muñoz-Jaramillo, Johannes Jakubik, Etienne Vos, Kshitiz Mandal, Ata Akbari Asanjan, Joao Lucas de Sousa Almeida, Amy Lin, Talwinder Singh, Kang Yang, Chetraj Pandey, Jinsu Hong, Berkay Aydin, Thorsten Kurth, Ryan McGranaghan, Spiridon Kasapis, Vishal Upendran, Shah Bahauddin, Daniel da Silva, Nikolai V. Pogorelov, Anne Spalding, Campbell Watson, Manil Maskey, Madhulika Guhathakurta, Juan Bernabe-Moreno, Rahul Ramachandran
  • 0

Abstract

Heliophysics is central to understanding and forecasting space weather eventsand solar activity. Despite decades of high-resolution observations from theSolar Dynamics Observatory (SDO), most models remain task-specific andconstrained by scarce labeled data, limiting their capacity to generalizeacross solar phenomena. We introduce Surya, a 366M parameter foundation modelfor heliophysics designed to learn general-purpose solar representations frommulti-instrument SDO observations, including eight Atmospheric Imaging Assembly(AIA) channels and five Helioseismic and Magnetic Imager (HMI) products. Suryaemploys a spatiotemporal transformer architecture with spectral gating andlong--short range attention, pretrained on high-resolution solar imageforecasting tasks and further optimized through autoregressive rollout tuning.Zero-shot evaluations demonstrate its ability to forecast solar dynamics andflare events, while downstream fine-tuning with parameter-efficient Low-RankAdaptation (LoRA) shows strong performance on solar wind forecasting, activeregion segmentation, solar flare forecasting, and EUV spectra. Surya is thefirst foundation model in heliophysics that uses time advancement as a pretexttask on full-resolution SDO data. Its novel architecture and performancesuggest that the model is able to learn the underlying physics behind solarevolution.

 

Quick Read (beta)

loading the full paper ...