STACI: Spatio-Temporal Aleatoric Conformal Inference
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2025
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866915573148942336 |
|---|---|
| author | Feng, Brandon R. Park, David Keetae Luo, Xihaier Urdangarin, Arantxa Yoo, Shinjae Reich, Brian J. |
| author_facet | Feng, Brandon R. Park, David Keetae Luo, Xihaier Urdangarin, Arantxa Yoo, Shinjae Reich, Brian J. |
| contents | Fitting Gaussian Processes (GPs) provides interpretable aleatoric uncertainty quantification for estimation of spatio-temporal fields. Spatio-temporal deep learning models, while scalable, typically assume a simplistic independent covariance matrix for the response, failing to capture the underlying correlation structure. However, spatio-temporal GPs suffer from issues of scalability and various forms of approximation bias resulting from restrictive assumptions of the covariance kernel function. We propose STACI, a novel framework consisting of a variational Bayesian neural network approximation of non-stationary spatio-temporal GP along with a novel spatio-temporal conformal inference algorithm. STACI is highly scalable, taking advantage of GPU training capabilities for neural network models, and provides statistically valid prediction intervals for uncertainty quantification. STACI outperforms competing GPs and deep methods in accurately approximating spatio-temporal processes and we show it easily scales to datasets with millions of observations. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2505_21658 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | STACI: Spatio-Temporal Aleatoric Conformal Inference Feng, Brandon R. Park, David Keetae Luo, Xihaier Urdangarin, Arantxa Yoo, Shinjae Reich, Brian J. Machine Learning Methodology Fitting Gaussian Processes (GPs) provides interpretable aleatoric uncertainty quantification for estimation of spatio-temporal fields. Spatio-temporal deep learning models, while scalable, typically assume a simplistic independent covariance matrix for the response, failing to capture the underlying correlation structure. However, spatio-temporal GPs suffer from issues of scalability and various forms of approximation bias resulting from restrictive assumptions of the covariance kernel function. We propose STACI, a novel framework consisting of a variational Bayesian neural network approximation of non-stationary spatio-temporal GP along with a novel spatio-temporal conformal inference algorithm. STACI is highly scalable, taking advantage of GPU training capabilities for neural network models, and provides statistically valid prediction intervals for uncertainty quantification. STACI outperforms competing GPs and deep methods in accurately approximating spatio-temporal processes and we show it easily scales to datasets with millions of observations. |
| title | STACI: Spatio-Temporal Aleatoric Conformal Inference |
| topic | Machine Learning Methodology |
| url | https://arxiv.org/abs/2505.21658 |