Tyche: One Step Flow for Efficient Probabilistic Weather Forecasting

• URL: http://arxiv.org/abs/2605.06916v1
• Date: Thu, 07 May 2026 20:25:28 GMT
• Title: Tyche: One Step Flow for Efficient Probabilistic Weather Forecasting
• Abstract summary: We present Tyche, a one-step conditional flow model for efficient probabilistic weather forecasting.<n>Experiments on ERA5 at 1.5$circ$ and 6-hour resolution show that our Tyche, using merely a single NFE, matches or exceeds the forecast skill and calibration of state-of-the-art multi-step generative baselines.
• Score: 48.263481135566316
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Probabilistic weather forecasting requires not only accurate trajectories, but calibrated distributions over plausible atmospheric futures. Recent data-driven systems have achieved remarkable deterministic skill, and diffusion-based ensemble forecasters have substantially improved sample realism and uncertainty quantification. However, their inference cost scales with forecast horizon, ensemble size, and the number of denoising steps required for each transition, making large operational ensembles expensive. To address this, we present Tyche, a one-step conditional flow model for efficient probabilistic weather forecasting. Tyche models the conditional forecast distribution with a destination-aware average-velocity flow that maps Gaussian noise directly to future weather states in a single function evaluation (1-NFE). To make this one-step transport learnable in high-dimensional geophysical fields, we derive a JVP-regularized rectification objective that enforces temporal self-consistency across source and destination flow timesteps without explicitly forming Jacobians. The transport field is parameterized by an isotropic Swin-style transformer that preserves fine-scale spatial structure while remaining scalable on global grids. To improve ensemble reliability under autoregressive forecasting, we further introduce a rollout-based finetuning stage with curriculum CRPS calibration supervision. Experiments on ERA5 at 1.5$^\circ$ and 6-hour resolution show that our Tyche, using merely a single NFE, matches or exceeds the forecast skill and calibration of state-of-the-art multi-step generative baselines and the operational ECMWF IFS ensemble.

Related papers

• FlowATC: Aircraft Trajectory Prediction via Flow Matching [39.046509366232065]
  We present a flow-matching architecture trained exclusively on historical aircraft trajectories, with no route labels or chart supervision.<n>Our model is trained directly on the native, irregular ADS-B sampling interval.<n>We show that the error degrades gracefully with prediction horizon, and the architecture remains effective when retrained on temporally decimated feeds.
  arXiv  Detail & Related papers  (2026-09-15T02:21:06Z)
• Earth System Foundation Model (ESFM): A unified framework for heterogeneous data integration and forecasting [52.53742727060265]
  Foundation models (FMs) for the Earth system learn statistical relationships between physical variables across massive datasets.<n>Here, we introduce Earth System Foundation Model (ESFM), a fully open model building on the 3Dwin UNet backbone of the pioneering model.<n>We present findings using dense gridded data (ERA5, CMIP6), regionally masked dense data, sparse gridded MODIS satellite data, and station data.<n>Results demonstrate competitive or superior performance relative to state-of-the-art benchmarks.
  arXiv  Detail & Related papers  (2026-04-20T11:40:39Z)
• FuXiWeather2: Learning accurate atmospheric state estimation for operational global weather forecasting [15.10103712842682]
  We present FuXiWeather2, a unified end-to-end neural framework for assimilation and forecasting.<n>FuXiWeather2 generates high-resolution ($0.25circ$) global analysis fields and 10-day forecasts within minutes.<n>These high-quality analysis fields drive deterministic forecasts that exceed the skill of the HRES system in 91% of evaluated metrics.
  arXiv  Detail & Related papers  (2026-03-16T14:36:47Z)
• Universal Diffusion-Based Probabilistic Downscaling [3.47037488116399]
  We introduce a universal diffusion-based downscaling framework that lifts deterministic low-resolution weather forecasts into high-resolution predictions.<n>A single conditional diffusion model is trained on paired coarse-resolution inputs and high-resolution regional reanalysis targets.<n>We evaluate probabilistic forecasts against independent in situ station observations over lead times up to 90 h.
  arXiv  Detail & Related papers  (2026-02-12T12:42:46Z)
• DistDF: Time-Series Forecasting Needs Joint-Distribution Wasserstein Alignment [92.70019102733453]
  Training time-series forecast models requires aligning the conditional distribution of model forecasts with that of the label sequence.<n>We propose DistDF, which achieves alignment by alternatively minimizing a discrepancy between the conditional forecast and label distributions.
  arXiv  Detail & Related papers  (2025-10-28T16:09:59Z)
• Appa: Bending Weather Dynamics with Latent Diffusion Models for Global Data Assimilation [4.430758443755128]
  Appa is a score-based data assimilation model producing global atmospheric trajectories at 0.25-degree resolution and 1-hour intervals.<n>Our results establish latent score-based data assimilation as a promising foundation for future global atmospheric modeling systems.
  arXiv  Detail & Related papers  (2025-04-25T22:14:29Z)
• Continuous Ensemble Weather Forecasting with Diffusion models [10.730406954385927]
  Continuous Ensemble Forecasting is a novel and flexible method for sampling ensemble forecasts in diffusion models.<n>It can generate temporally consistent ensemble trajectories completely in parallel, with no autoregressive steps.<n>We demonstrate that the method achieves competitive results for global weather forecasting with good probabilistic properties.
  arXiv  Detail & Related papers  (2024-10-07T18:51:23Z)
• GenCast: Diffusion-based ensemble forecasting for medium-range weather [10.845679586464026]
  We introduce GenCast, a probabilistic weather model with greater skill and speed than the top operational medium-range weather forecast in the world. GenCast generates an ensemble of 15-day global forecasts, at 12-hour steps and 0.25 degree latitude-longitude, for over 80 surface and atmospheric variables in 8 minutes. It has greater skill than ENS on 97.4% of 1320 targets we evaluated, and better predicts extreme weather, tropical cyclones, and wind power production.
  arXiv  Detail & Related papers  (2023-12-25T19:30:06Z)
• When Rigidity Hurts: Soft Consistency Regularization for Probabilistic Hierarchical Time Series Forecasting [69.30930115236228]
  Probabilistic hierarchical time-series forecasting is an important variant of time-series forecasting. Most methods focus on point predictions and do not provide well-calibrated probabilistic forecasts distributions. We propose PROFHiT, a fully probabilistic hierarchical forecasting model that jointly models forecast distribution of entire hierarchy.
  arXiv  Detail & Related papers  (2023-10-17T20:30:16Z)
• SEEDS: Emulation of Weather Forecast Ensembles with Diffusion Models [13.331224394143117]
  Uncertainty quantification is crucial to decision-making. dominant approach to representing uncertainty in weather forecasting is to generate an ensemble of forecasts. We propose to amortize the computational cost by emulating these forecasts with deep generative diffusion models learned from historical data.
  arXiv  Detail & Related papers  (2023-06-24T22:00:06Z)
• When Rigidity Hurts: Soft Consistency Regularization for Probabilistic Hierarchical Time Series Forecasting [69.30930115236228]
  Probabilistic hierarchical time-series forecasting is an important variant of time-series forecasting. Most methods focus on point predictions and do not provide well-calibrated probabilistic forecasts distributions. We propose PROFHiT, a fully probabilistic hierarchical forecasting model that jointly models forecast distribution of entire hierarchy.
  arXiv  Detail & Related papers  (2022-06-16T06:13:53Z)
• Learning Interpretable Deep State Space Model for Probabilistic Time Series Forecasting [98.57851612518758]
  Probabilistic time series forecasting involves estimating the distribution of future based on its history. We propose a deep state space model for probabilistic time series forecasting whereby the non-linear emission model and transition model are parameterized by networks. We show in experiments that our model produces accurate and sharp probabilistic forecasts.
  arXiv  Detail & Related papers  (2021-01-31T06:49:33Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.