Saved in:
Bibliographic Details
Main Authors: Bora, Aniruddha, Zhang, Shixuan, Shukla, Khemraj, Harrop, Bryce, Karniadakis, George Em., Leung, L. Ruby
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.03309
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911299550576640
author Bora, Aniruddha
Zhang, Shixuan
Shukla, Khemraj
Harrop, Bryce
Karniadakis, George Em.
Leung, L. Ruby
author_facet Bora, Aniruddha
Zhang, Shixuan
Shukla, Khemraj
Harrop, Bryce
Karniadakis, George Em.
Leung, L. Ruby
contents Coarse resolution, imperfect parameterizations, and uncertain initial states and forcings limit Earth-system model (ESM) predictions. Traditional bias correction via data assimilation improves constrained simulations but offers limited benefit once models run freely. We introduce an operator-learning framework that maps instantaneous model states to bias-correction tendencies and applies them online during integration. Building on a U-Net backbone, we develop two operator architectures Inception U-Net (IUNet) and a multi-scale network (M\&M) that combine diverse upsampling and receptive fields to capture multiscale nonlinear features under Energy Exascale Earth System Model (E3SM) runtime constraints. Trained on two years E3SM simulations nudged toward ERA5 reanalysis, the operators generalize across height levels and seasons. Both architectures outperform standard U-Net baselines in offline tests, indicating that functional richness rather than parameter count drives performance. In online hybrid E3SM runs, M\&M delivers the most consistent bias reductions across variables and vertical levels. The ML-augmented configurations remain stable and computationally feasible in multi-year simulations, providing a practical pathway for scalable hybrid modeling. Our framework emphasizes long-term stability, portability, and cadence-limited updates, demonstrating the utility of expressive ML operators for learning structured, cross-scale relationships and retrofitting legacy ESMs.
format Preprint
id arxiv_https___arxiv_org_abs_2512_03309
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Retrofitting Earth System Models with Cadence-Limited Neural Operator Updates
Bora, Aniruddha
Zhang, Shixuan
Shukla, Khemraj
Harrop, Bryce
Karniadakis, George Em.
Leung, L. Ruby
Machine Learning
Artificial Intelligence
Mathematical Physics
Coarse resolution, imperfect parameterizations, and uncertain initial states and forcings limit Earth-system model (ESM) predictions. Traditional bias correction via data assimilation improves constrained simulations but offers limited benefit once models run freely. We introduce an operator-learning framework that maps instantaneous model states to bias-correction tendencies and applies them online during integration. Building on a U-Net backbone, we develop two operator architectures Inception U-Net (IUNet) and a multi-scale network (M\&M) that combine diverse upsampling and receptive fields to capture multiscale nonlinear features under Energy Exascale Earth System Model (E3SM) runtime constraints. Trained on two years E3SM simulations nudged toward ERA5 reanalysis, the operators generalize across height levels and seasons. Both architectures outperform standard U-Net baselines in offline tests, indicating that functional richness rather than parameter count drives performance. In online hybrid E3SM runs, M\&M delivers the most consistent bias reductions across variables and vertical levels. The ML-augmented configurations remain stable and computationally feasible in multi-year simulations, providing a practical pathway for scalable hybrid modeling. Our framework emphasizes long-term stability, portability, and cadence-limited updates, demonstrating the utility of expressive ML operators for learning structured, cross-scale relationships and retrofitting legacy ESMs.
title Retrofitting Earth System Models with Cadence-Limited Neural Operator Updates
topic Machine Learning
Artificial Intelligence
Mathematical Physics
url https://arxiv.org/abs/2512.03309