Accelerated Spatio-Temporal Bayesian Modeling for Multivariate Gaussian Processes
Informazioni aggiuntive
Autori
Gaedke-Merzhäuser L.,
Maillou V.,
Avellaneda F. R.,
Schenk O.,
Luisier M.,
Moraga P.,
Ziogas A. N.,
Rue H.
Tipo
Contributo in atti di convegno
Anno
2025
Lingua
Inglese
Sommario
Multivariate Gaussian processes (GPs) offer a powerful probabilistic framework to represent complex interdependent phenomena. They pose, however, significant computational challenges in high-dimensional settings, which frequently arise in spatial-temporal applications. We present PyINLA, a highly scalable framework for performing Bayesian inference tasks on spatio-temporal multivariate GPs, based on the methodology of integrated nested Laplace approximations. Our approach relies on a sparse inverse covariance matrix formulation of the GP, puts forward a GPU-accelerated block-dense approach, and introduces a hierarchical, triple-layer, distributed memory parallel scheme. We showcase weak scaling performance surpassing the state- of-the-art by two orders of magnitude on an 8$\times$ larger model and measure strong scaling speedups of
three orders of magnitude when running on 496 GH200 superchips on the Alps supercomputer. Applying PyINLA to air pollution data from northern Italy over 48 days, we showcase refined spatial resolutions over the aggregated pollutant measurements.
Parole chiave
omputational earth and atmospheric sciences Improved models, algorithms, performance or scalability of specific applications and respective software Use of uncertainty quantification, statistical, and machine-learning techniques to improve a specific HPC application
Titolo atti di convegno
The International Conference for High Performance Computing, Networking, Storage, and Analysis
Number ( Month )
November
Editore
ACM/IEEE
Nome convegno
The International Conference for High Performance Computing, Networking, Storage, and Analysis
Luogo convegno
St. Louis, MO
Data convegno
16–21 Nov 2025
Pagine (o numero dell’articolo)
1-12
