arXiv:2502.04140cs.LGcs.AI2025-02中稿 · DMLR, see https://…被引 2

用偏微分方程生成时空图数据,解决灾害建模数据稀缺问题。

Synthetic Datasets for Machine Learning on Spatio-Temporal Graphs using PDEs

  • 基于PDE构建三类灾害的合成时空图数据
  • 在流行病数据上验证模型,预训练提升真实数据表现
  • 方法可定制化,适合灾害模拟与图学习研究者

许多物理过程可用偏微分方程(PDE)描述。现实世界中此类过程的测量常分布在不规则空间点上,可有效表示为图结构;然而现有相关数据集极少。本文旨在让PDE建模进展惠及时序图机器学习领域,同时缓解数据稀缺问题,通过构建和使用基于PDE的合成数据集支持不同应用场景的时空图建模。具体展示三个方程分别模拟流行病、大气颗粒物与海啸波三种灾害。进一步,在流行病数据集上对多个机器学习模型进行基准测试,并证明在该数据集上预训练可提升模型在真实流行病数据上的性能。所提方法使他人能按需创建定制化数据集与基准测试。方法代码及三个数据集见 https://github.com/github-usr-ano/Temporal_Graph_Data_PDEs。

原文摘要 · Abstract (English)

Many physical processes can be expressed through partial differential equations (PDEs). Real-world measurements of such processes are often collected at irregularly distributed points in space, which can be effectively represented as graphs; however, there are currently only a few existing datasets. Our work aims to make advancements in the field of PDE-modeling accessible to the temporal graph machine learning community, while addressing the data scarcity problem, by creating and utilizing datasets based on PDEs. In this work, we create and use synthetic datasets based on PDEs to support spatio-temporal graph modeling in machine learning for different applications. More precisely, we showcase three equations to model different types of disasters and hazards in the fields of epidemiology, atmospheric particles, and tsunami waves. Further, we show how such created datasets can be used by benchmarking several machine learning models on the epidemiological dataset. Additionally, we show how pre-training on this dataset can improve model performance on real-world epidemiological data. The presented methods enable others to create datasets and benchmarks customized to individual requirements. The source code for our methodology and the three created datasets can be found on https://github.com/github-usr-ano/Temporal_Graph_Data_PDEs.

PDE建模时空图合成数据灾害模拟

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。