摘要
时空数据集,由空间参考的时间序列组成,广泛应用于空气污染监测、疾病追踪和云需求预测等各种应用中。随着现代数据集规模的增长,越来越需要灵活地捕捉复杂时空动态并能够处理大量观测数据的统计方法。本文介绍了贝叶斯神经场 (BayesNF),这是一种领域通用的统计模型,它可以为包括预测、插值和变异函数分析在内的数据分析任务推断丰富的时空概率分布。BayesNF 集成了深度神经网络架构以进行高容量函数估计,并结合了分层贝叶斯推理以进行稳健的预测不确定性量化。与著名基准的评估结果表明,BayesNF 在包含数十万到数百万个测量值的来自气候和公共卫生数据的预测问题上取得了改进。本文附带一个开源软件包 (https://github.com/google/bayesnf),该软件包可通过 JAX 机器学习平台在 GPU 和 TPU 加速器上运行。