Machine unlearning: bias correction in neural network downscaled storms

• Machine learning models are tested to downscale synthetic storms in controlled settings. • WGANs reproduce fine-scale spatiotemporal structures but misrepresent extreme statistics. • Combining deep learning with post-training bias correction yields optimal performance. Accurate precipitation at fine spatial resolutions is essential for hydrologic modeling and risk assessment, yet most precipitation data are available at coarse scales. Dynamical downscaling can improve spatial resolution but is computationally expensive while statistical downscaling struggles to reproduce high-resolution characteristics. Machine learning has been shown to offer an operational alternative for transforming coarse data into fine-scale fields, especially honoring spatiotemporal precipitation dependencies. Here, we show that combining machine learning with post-processing bias correction approaches—a form of “machine unlearning”— yields improved performance. We evaluate four machine-learning models—Linear Network (LNet), Fully Connected Network (FCNet), Convolutional Neural Network (UNet), and Wasserstein Generative Adversarial Network (WGAN)—for downscaling precipitation using synthetic benchmark storms with known marginal and spatiotemporal properties. Using synthetic fields provides full control over storm characteristics and enables rigorous evaluation. Raw outputs from all models struggle to reproduce wet/dry boundaries, statistics, and extremes, and only WGAN captures the complex spatiotemporal structure of fine-scale storms. We then apply linear and nonlinear bias corrections to enforce zeros, match the mean of positive values, and align full marginal distributions, including tails. This demonstrates that post-processing is crucial for reliable neural network outputs in operational settings. The results highlight WGAN’s potential for operational downscaling while emphasizing the need for systematic post-processing and careful validation before real-world application.

Authors

Institutions

Publication Details

Journal
Journal of Hydrology
Published
2025-11-28
DOI
https://doi.org/10.1016/j.jhydrol.2025.134689
Primary Topic
Climate variability and models
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Machine unlearning: bias correction in neural network downscaled storms

Antonios Mamalakis, Simon Michael Papalexiou
Journal of Hydrology
Climate variability and models
article

Machine unlearning: bias correction in neural network downscaled storms

Antonios Mamalakis, Simon Michael Papalexiou
article en

Abstract

• Machine learning models are tested to downscale synthetic storms in controlled settings. • WGANs reproduce fine-scale spatiotemporal structures but misrepresent extreme statistics. • Combining deep learning with post-training bias correction yields optimal performance. Accurate precipitation at fine spatial resolutions is essential for hydrologic modeling and risk assessment, yet most precipitation data are available at coarse scales. Dynamical downscaling can improve spatial resolution but is computationally expensive while statistical downscaling struggles to reproduce high-resolution characteristics. Machine learning has been shown to offer an operational alternative for transforming coarse data into fine-scale fields, especially honoring spatiotemporal precipitation dependencies. Here, we show that combining machine learning with post-processing bias correction approaches—a form of “machine unlearning”— yields improved performance. We evaluate four machine-learning models—Linear Network (LNet), Fully Connected Network (FCNet), Convolutional Neural Network (UNet), and Wasserstein Generative Adversarial Network (WGAN)—for downscaling precipitation using synthetic benchmark storms with known marginal and spatiotemporal properties. Using synthetic fields provides full control over storm characteristics and enables rigorous evaluation. Raw outputs from all models struggle to reproduce wet/dry boundaries, statistics, and extremes, and only WGAN captures the complex spatiotemporal structure of fine-scale storms. We then apply linear and nonlinear bias corrections to enforce zeros, match the mean of positive values, and align full marginal distributions, including tails. This demonstrates that post-processing is crucial for reliable neural network outputs in operational settings. The results highlight WGAN’s potential for operational downscaling while emphasizing the need for systematic post-processing and careful validation before real-world application.

Journal of HydrologyVol. 665
Universität Hamburg (DE), University of Virginia (US), Hamburg University of Technology (DE)
Openalex Percentile: Top 65%
Climate variability and models
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.