摘要
arXiv:2502.06848v1 交叉类型
摘要:近年来,基于图神经网络(GNN)的模型在模拟复杂系统的物理现象方面显示出有希望的结果。然而,训练专用的图网络物理模拟器可能是昂贵的,因为大多数模型受限于完全监督的训练,这需要来自传统物理模拟器的大量数据生成。到目前为止,转移学习如何提高模型性能和训练效率仍未被探索。在本文中,我们介绍了一种图网络模拟器的预训练和转移学习范式。我们提出了可扩展的图U-net(SGUNET)。通过采用创新的深度优先搜索(DFS)池化,SGUNET可以适应不同网格大小和分辨率的不同模拟任务。为了在不同配置的SGUNET之间实现转移学习,我们提出了一组映射函数以对齐预训练模型和目标模型之间的参数。还向损失中添加了一个额外的规范化项,以在预训练权重和目标模型权重之间限制差异,从而提高泛化性能。为了预训练我们的物理模拟器,我们创建了一个数据集,其中包括从开源A Big CAD(ABC)数据集随机选择的20,000个3D形状的物理模拟。我们展示了我们提出的转移学习方法即使在使用少量训练数据微调时,也能让模型比从完整扩展数据集重新训练时表现更好。在2D可变形板基准数据集中,我们使用1/16的训练数据预训练的模型在位置RMSE上比从头开始训练的模型提高了11.05%。