Generative multi-scale modeling via spatial autoregressive transport maps

Abstract

Spatial fields in the Earth and environmental sciences are often available at multiple scales or resolutions. While coarse-scale data (e.g., from global circulation models) are often abundant, they lack the local detail provided by fine-scale data (e.g., from regional climate models), which are typically computationally expensive to generate. Statistical downscaling and multi-scale data fusion address this challenge by predicting high-resolution fields from low-resolution or related inputs. We propose a highly scalable Bayesian approach that can learn the joint non-Gaussian distribution and nonlinear dependence structure of nonstationary spatial fields across multiple scales from a small number of training samples. Our method employs scale-aware autoregressive Gaussian processes with suitably chosen regularization-inducing priors to model the conditional distribution of fine-scale fields given coarse-scale data. Exploiting conjugacy, the integrated likelihood is available in closed form, enabling efficient parameter optimization via stochastic gradient descent. Once trained, the method provides a closed-form characterization of the posterior distribution of fine-scale fields given coarse-scale inputs. In numerical comparisons, we demonstrate that our approach substantially outperforms existing methods and effectively characterizes and simulates fine-scale climate behavior based on output from coarse global circulation models.

23 pages, 8 figures

References in corpus (8)

• Variational Inference: A Review for Statisticians
• Permutation and Grouping Methods for Sharpening Gaussian Process Approximations
• An introduction to sampling via measure transport
• When does the screening effect hold?
• On the representation and learning of monotone triangular transport maps
• Bi-fidelity Variational Auto-encoder for Uncertainty Quantification
• Multi-fidelity Hierarchical Neural Processes
• Bayesian nonparametric generative modeling of large multivariate non-Gaussian spatial fields