arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.06921cs.LG
收藏

基于Gumbel-Max方程学习器网络的符号表达式学习

Learning Symbolic Expressions via Gumbel-Max Equation Learner Networks

School of Engineering and Computer Science(工程与计算机科学学院);Victoria University of Wellington(惠灵顿维多利亚大学)

Gang Chen

摘要:

大多数通过最先进的机器学习技术学习的神经网络(NN)都是黑箱模型。为了使机器学习在科学和工程领域取得广泛成功,开发新的NN架构以有效地从复杂数据集中提取高级数学知识是非常重要的。基于这种理解,本文开发了一种新的NN架构,称为Gumbel-Max方程学习器(GMEQL)网络。与以往提出的EQL网络不同,GMEQL通过Gumbel-Max技巧对网络结构进行连续松弛,并引入了两种类型的可训练参数:结构参数和回归参数。本文还提出了一个两阶段的训练过程与新技术,训练结构参数在线和离线设置的基础上的精英库。在8个基准符号回归问题上,实验表明GMEQL优于几种尖端的机器学习方法。

英文摘要:

Most of the neural networks (NNs) learned via state-of-the-art machine learning techniques are black-box models. For a widespread success of machine learning in science and engineering, it is important to develop new NN architectures to effectively extract high-level mathematical knowledge from complex datasets. Motivated by this understanding, this paper develops a new NN architecture called the Gumbel-Max Equation Learner (GMEQL) network. Different from previously proposed Equation Learner (EQL) networks, GMEQL applies continuous relaxation to the network structure via the Gumbel-Max trick and introduces two types of trainable parameters: structure parameters and regression parameters. This paper also proposes a two-stage training process with new techniques to train structure parameters in both online and offline settings based on an elite repository. On 8 benchmark symbolic regression problems, GMEQL is experimentally shown to outperform several cutting-edge machine learning approaches.

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

↑