Reshaping Geostatistical Modeling and Prediction for Extreme-Scale Environmental Applications
- Resource Type
- Conference
- Authors
- Cao, Qinglei; Abdulah, Sameh; Alomairy, Rabab; Pei, Yu; Nag, Pratik; Bosilca, George; Dongarra, Jack; Genton, Marc G.; Keyes, David E.; Ltaief, Hatem; Sun, Ying
- Source
- SC22: International Conference for High Performance Computing, Networking, Storage and Analysis SC High Performance Computing, Networking, Storage and Analysis, SC22: International Conference for. :1-12 Nov, 2022
- Subject
- Communication, Networking and Broadcast Technologies
Analytical models
Runtime
Computational modeling
High performance computing
Predictive models
Programming
Mathematical models
Space-Time Geospatial Statistics
Climate/Weather Prediction
Task-Based Programming Models
Dynamic Runtime Systems
Mixed-Precision Computations
Low-Rank Matrix Approximations
High Performance Computing
- Language
- ISSN
- 2167-4337
We extend the capability of space-time geostatistical modeling using algebraic approximations, illustrating application-expected accuracy worthy of double precision from majority low-precision computations and low-rank matrix approximations. We exploit the mathematical structure of the dense covariance matrix whose inverse action and determinant are repeatedly required in Gaussian log-likelihood optimization. Geostatistics augments first-principles modeling approaches for the prediction of environmental phenomena given the availability of measurements at a large number of locations; however, traditional Cholesky-based approaches grow cubically in complexity, gating practical extension to continental and global datasets now available. We combine the linear algebraic contributions of mixed-precision and low-rank computations within a tile based Cholesky solver with on-demand casting of precisions and dynamic runtime support from PaRSEC to orchestrate tasks and data movement. Our adaptive approach scales on various systems and leverages the Fujitsu A64FX nodes of Fugaku to achieve up to 12X performance speedup against the highly optimized dense Cholesky implementation.