摘要
arXiv:2503.23617v1 公告类型:交叉领域
摘要:在各个领域中,能够发现有意义、准确且精炼的数学方程以描述数据集的能力是宝贵的。方程提供了变量之间明确的关系,从而为深入了解潜在数据模式提供了更深层次的见解。目前大多数方程发现方法依赖于遗传程序,该方法会迭代地搜索方程空间,但通常速度较慢,并且容易过拟合。通过将方程表示为有向无环图,我们利用图神经网络来学习方程的潜在语义,并生成新的、以前未见过的方程。尽管图生成模型在许多领域中证明了在发现新类型的图方面是成功的,但它们在发现方程方面的应用仍然大多未被探索。在这项工作中,我们提出Graph-EQ,一种用于高效方程发现的深度图生成模型。Graph-EQ 使用条件变分自编码器(CVAE)以无监督的方式训练大量的方程集,学习方程空间的丰富潜在表示。我们不是直接搜索方程空间,而是采用贝叶斯优化来高效地探索学习到的潜在空间。我们展示了Graph-Eq 的编码器-解码器架构能够准确重构输入方程。此外,我们展示了学习到的潜在表示可以被采样和解码成有效的方程,包括训练数据中的新和以前未见过的方程。最后,我们通过使用贝叶斯优化在潜在空间中探索,评估Graph-Eq 发现最佳匹配数据集方程的能力。潜在空间探索在20个已知真实方程的数据集上进行,结果显示Graph-Eq 成功地在大多数数据集中发现了真实方程。