摘要
图神经网络已成为大规模基于网格的物理模拟的强大工具。现有方法主要采用分层、多尺度消息传递来捕获图中的长程依赖关系。然而,这些图层次结构通常是固定的,并且是手动设计的,它们不能适应复杂物理系统中存在的不断变化的动力学。在本文中,我们介绍了一种名为 DHMP 的新型神经网络,它通过可微节点选择方法学习消息传递网络的动态层次结构。关键组成部分是各向异性消息传递机制,它在层内和层间交互中运行。与现有方法不同,它首先支持在每个图层次结构中相邻节点之间动态特征的方向性非均匀聚合。其次,它根据不同的物理环境确定下一个层次结构的节点选择概率,从而为学习远程节点关系创建更灵活的消息捷径。我们的实验表明了 DHMP 的有效性,与最近的固定层次结构消息传递网络相比,在五个经典物理模拟数据集上平均提高了 22.7%。