LLM2D
使用偏微分方程生成时空图上的机器学习数据集
Synthetic Datasets for Machine Learning on Spatio-Temporal Graphs using PDEs
作者: Jost Arndt, Utku Isil, Michael Detzel, Wojciech Samek, Jackie Ma
发布日期: 2/7/2025
arXiv ID: oai:arXiv.org:2502.04140v1

摘要

arXiv:2502.04140v1 交叉公告类型:cross 摘要:许多物理过程可以通过偏微分方程(PDEs)来表达。此类过程的现实世界测量数据通常在空间上分布在不规则的点上,可以有效地表示为图;然而,目前现有的数据集还很少。我们的工作旨在通过创建和利用基于PDEs的数据集,使PDE建模领域的进展能够被时间图机器学习社区所利用,同时解决数据稀缺性问题。在本文中,我们创建并使用基于PDEs的合成数据集来支持机器学习中的时空图建模,以供不同应用使用。更具体地说,我们展示了三个方程来建模流行病学、大气颗粒和海啸波三大领域的不同类型的灾难和危害。此外,我们展示了如何通过在流行病学数据集上基准测试几种机器学习模型来使用这些创建的数据集。我们还展示了对这个数据集进行预训练如何提高在真实世界流行病学数据上的模型性能。所介绍的方法使其他人能够根据个人需求创建数据集和基准测试。我们的方法和创建的三个数据集的源代码可以在https://github.com/github-usr-ano/Temporal_Graph_Data_PDEs找到。