arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.10784cs.LGcs.NEstat.ML

OccamNet:一种面向大规模符号回归的快速神经模型

OccamNet: A Fast Neural Model for Symbolic Regression at Scale

Massachusetts Institute of Technology(麻省理工学院)

Owen Dugan, Rumen Dangovski, Allan Costa, Samuel Kim, Pawan Goyal, Joseph Jacobson, Marin Soljačić

AI总结:

本文提出OccamNet,一种通过概率分布采样函数并结合强化学习交叉熵匹配训练的神经符号回归模型,能以低内存开销快速找到紧凑可解释的符号拟合,并在多种任务上超越现有方法。

摘要:

神经网络的表达能力以复杂的黑盒模型为代价,这些模型在训练数据集领域之外往往外推能力较差,这与寻找紧凑解析表达式来描述科学数据的目标相冲突。我们提出了OccamNet,这是一种神经网络模型,能够以奥卡姆剃刀的方式找到可解释、紧凑且稀疏的符号拟合。我们的模型定义了一个关于函数的概率分布,并具有高效的采样和函数求值能力。我们通过采样函数并将概率质量偏向于拟合效果更好的解来进行训练,并在强化学习损失中使用交叉熵匹配进行反向传播。OccamNet能够为多种问题识别符号拟合,包括解析函数和非解析函数、隐函数以及简单的图像分类,并且在实际回归数据集上能够超越最先进的符号回归方法。我们的方法仅需极小的内存占用,可在单个CPU上于数分钟内拟合复杂函数,并可在GPU上扩展。

英文摘要:

Neural networks' expressiveness comes at the cost of complex, black-box models that often extrapolate poorly beyond the domain of the training dataset, conflicting with the goal of finding compact analytic expressions to describe scientific data. We introduce OccamNet, a neural network model that finds interpretable, compact, and sparse symbolic fits to data, à la Occam's razor. Our model defines a probability distribution over functions with efficient sampling and function evaluation. We train by sampling functions and biasing the probability mass toward better fitting solutions, backpropagating using cross-entropy matching in a reinforcement-learning loss. OccamNet can identify symbolic fits for a variety of problems, including analytic and non-analytic functions, implicit functions, and simple image classification, and can outperform state-of-the-art symbolic regression methods on real-world regression datasets. Our method requires a minimal memory footprint, fits complicated functions in minutes on a single CPU, and scales on a GPU.

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

↑