Collaborative Research: OAC Core: Distributed Graph Learning Cyberinfrastructure for Large-scale Spatiotemporal Prediction

图形神经网络(GNN)将深度神经网络的成功从独立数据点扩展到相关数据点，例如从环境传感器(如湿度、温度、PM2.5等)现场收集的观测数据。广泛分布在不同的空间位置。虽然大多数现有的工作集中在相对较小的、经过良好管理的数据上的概念验证，但在离线环境下，现实世界的科学研究和应用程序需要更强大的GNN模型，该模型可以有效地学习大规模、实时、地理分布(地理分布)和不同(异质)数据。该项目旨在绘制一种全新的网络基础设施解决方案，用于培训大空间GNN以填补这一空白。该项目的成功将提供一个网络基础设施，克服依赖大规模时空预测的广泛领域科学应用的基本计算和通信瓶颈。拟议的算法和系统将非常适合培养对在地理分布的规模上设计大型机器学习系统的更深层次的理解，教学和培训学生和同行，并为研究生和本科生提供新的课程、研究和实习机会。该项目旨在开发一套全面的图形构建和划分方法、分布式学习算法和网络基础设施设计，以支持大规模GNN在地理空间科学研究和应用中用于真实世界的时空数据。该项目将解决重大的研究挑战，包括(1)在受地理启发的图形深度学习框架内制定时空预测，(2)实现跨大量地理分散的数据集的高精度、高效率和高成本效益的时空预测任务，以及(3)整合空间相关性、空间异质性、空间计算并行性和地理通信效率。这项研究围绕几个关键研究主题展开：(1)创建一个从时空数据构建图形的通用框架，确定空间关系，并填充缺失的节点属性。(2)开发了一个利用多个边缘微数据中心进行GNN模型协作学习的集中式时空图学习基础设施。(3)建立一个分散的时空图学习基础设施，支持分散的地理多任务学习，以解决空间异质性。该奖项反映了NSF的法定使命，并通过使用基金会的智力优势和更广泛的影响审查标准进行评估，被认为值得支持。

Graph Neural Networks (GNNs) have extended Deep Neural Networks’ success from independent data points to relational data points, such as observations collected on-site from environmental sensors (e.g., humidity, temperature, PM2.5, etc.) widely distributed in different spatial locations. While most existing works focus on proof-of-concept on relatively small, well-curated data, with offline settings, real-world scientific research, and applications need more capable GNN models, which can effectively learn from large-scale, real-time, geographically distributed (geo-distributed) and diversely different (heterogeneous) data. This project aims to chart a radically new cyberinfrastructure solution for training large-spatial GNNs to fill this gap. The success of this project will provide a cyberinfrastructure that overcomes the fundamental computational and communication bottlenecks for a broad range of domain science applications that rely on massive spatiotemporal prediction. The proposed algorithms and systems will be ideal for cultivating a deeper understanding of designing large machine-learning systems at a geo-distributed scale, teaching and training students and peers, and providing graduate and undergraduate students with new courses, research, and internship opportunities. This project aims to develop a comprehensive set of graph construction and partitioning methods, distributed learning algorithms, and cyberinfrastructure designs to support large-scale GNNs for real-world spatiotemporal data in geospatial scientific research and applications. The project will address significant research challenges, including (1) formulating spatiotemporal prediction within a geographically inspired graph deep learning framework, (2) enabling highly accurate, efficient, and cost-effective spatiotemporal prediction tasks across vast, geographically dispersed datasets, and (3) integrating spatial correlation, spatial heterogeneity, spatial computing parallelism, and geographic communication efficiency. The research is organized around several key research themes: (1) Creating a universal framework for constructing graphs from spatiotemporal data, determining spatial relationships, and filling in missing node attributes. (2) Developing a centralized spatiotemporal graph learning infrastructure that leverages multiple edge micro-datacenters for collaborative GNN model learning. (3) Establishing a decentralized spatiotemporal graph learning infrastructure that supports decentralized geographical multitask learning to address spatial heterogeneity.This award reflects NSF's statutory mission and has been deemed worthy of support through evaluation using the Foundation's intellectual merit and broader impacts review criteria.