{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/scalable-spatiotemporal-prediction-with","title":"Scalable Spatiotemporal Prediction with Bayesian Neural Fields","arxiv_id":"2403.07657","date":"2024-03-12","proceeding":null,"authors":["Feras Saad","Jacob Burnim","Colin Carroll","Brian Patton","Urs Köster","Rif A. Saurous","Matthew Hoffman"],"abstract":"Spatiotemporal datasets, which consist of spatially-referenced time series, are ubiquitous in diverse applications, such as air pollution monitoring, disease tracking, and cloud-demand forecasting. As the scale of modern datasets increases, there is a growing need for statistical methods that are flexible enough to capture complex spatiotemporal dynamics and scalable enough to handle many observations. This article introduces the Bayesian Neural Field (BayesNF), a domain-general statistical model that infers rich spatiotemporal probability distributions for data-analysis tasks including forecasting, interpolation, and variography. BayesNF integrates a deep neural network architecture for high-capacity function estimation with hierarchical Bayesian inference for robust predictive uncertainty quantification. Evaluations against prominent baselines show that BayesNF delivers improvements on prediction problems from climate and public health data containing tens to hundreds of thousands of measurements. Accompanying the paper is an open-source software package (https://github.com/google/bayesnf) that runs on GPU and TPU accelerators through the JAX machine learning platform.","url_abs":"https://arxiv.org/abs/2403.07657v3","url_pdf":"https://arxiv.org/pdf/2403.07657v3.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"scalable-spatiotemporal-prediction-with","repo_url":"https://github.com/google/bayesnf","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"jax","reach":{"status":"ok","spdx":"Apache-2.0"}}],"tasks":[{"task_slug":"bayesian-inference","task_name":"Bayesian Inference"},{"task_slug":"demand-forecasting","task_name":"Demand Forecasting"},{"task_slug":null,"task_name":"GPU"},{"task_slug":"prediction","task_name":"Prediction"},{"task_slug":"uncertainty-quantification","task_name":"Uncertainty Quantification"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2403.07657","atlas_url":"https://app.syntology.ai/?focus=2403.07657","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2403.07657"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/google/bayesnf","reach":{"status":"ok","spdx":"Apache-2.0"}}],"summary":{"ran":7},"by_repo_kind":{"official":{"samples":7,"ran":7,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"ac3ee7b571921e42","entry":"make_fourier_features","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/models.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/models.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"ac3ee7b571921e42"}},{"code_sha256_prefix":"4450f47e0e5068b3","entry":"make_seasonal_features","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/models.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/models.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"4450f47e0e5068b3"}},{"code_sha256_prefix":"f685ef67b5f0a6fa","entry":"make_seasonal_frequencies","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/models.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/models.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f685ef67b5f0a6fa"}},{"code_sha256_prefix":"b7292506d7fdef97","entry":"make_vi_init","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/inference.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/inference.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"b7292506d7fdef97"}},{"code_sha256_prefix":"f505297eac3c1f04","entry":"permute_dataset","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/inference.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/inference.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f505297eac3c1f04"}},{"code_sha256_prefix":"b6103b95126c8450","entry":"seasonalities_to_array","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/spatiotemporal.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/spatiotemporal.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"b6103b95126c8450"}},{"code_sha256_prefix":"785efccdb0e029be","entry":"seasonality_to_float","repo":"google/bayesnf","repo_kind":"official","path":"src/bayesnf/spatiotemporal.py","file_url":"https://github.com/google/bayesnf/blob/HEAD/src/bayesnf/spatiotemporal.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"785efccdb0e029be"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}