arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.11287cs.LGastro-ph.COastro-ph.IMphysics.comp-phstat.ML
收藏

从具有归纳偏差的深度学习中发现符号模型

Discovering Symbolic Models from Deep Learning with Inductive Biases

Princeton University(普林斯顿大学);DeepMind(深度思维);New York University(纽约大学);Flatiron Institute(弗莱提荣研究所);Carnegie Mellon University(卡内基梅隆大学)

Miles Cranmer, Alvaro Sanchez-Gonzalez, Peter Battaglia, Rui Xu, Kyle Cranmer, David Spergel, Shirley Ho

摘要:

我们开发了一种通用方法,通过引入强归纳偏差来提取学习的深度模型的符号表示。我们专注于图神经网络(GNNs)。该技术的工作原理如下:当我们在监督环境中训练GNN时,我们首先鼓励稀疏潜在表示,然后我们将符号回归应用于学习模型的组件以提取显式物理关系。我们发现正确的已知方程,包括力的法律和哈密顿量,可以从神经网络提取。然后,我们将我们的方法应用到一个非平凡的宇宙学的例子-一个详细的暗物质模拟-并发现一个新的解析公式,它可以预测附近的宇宙结构的质量分布的暗物质的浓度。使用我们的技术从GNN中提取的符号表达式也比GNN本身更好地推广到分布外数据。我们的方法为解释神经网络和从他们学习的表示中发现新的物理原理提供了替代方向。

英文摘要:

We develop a general approach to distill symbolic representations of a learned deep model by introducing strong inductive biases. We focus on Graph Neural Networks (GNNs). The technique works as follows: we first encourage sparse latent representations when we train a GNN in a supervised setting, then we apply symbolic regression to components of the learned model to extract explicit physical relations. We find the correct known equations, including force laws and Hamiltonians, can be extracted from the neural network. We then apply our method to a non-trivial cosmology example-a detailed dark matter simulation-and discover a new analytic formula which can predict the concentration of dark matter from the mass distribution of nearby cosmic structures. The symbolic expressions extracted from the GNN using our technique also generalized to out-of-distribution data better than the GNN itself. Our approach offers alternative directions for interpreting neural networks and discovering novel physical principles from the representations they learn.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑