arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2302.08875stat.MLcs.LG
收藏

均值方差估计神经网络的最优训练

Optimal Training of Mean Variance Estimation Neural Networks

Radboud University(拉德堡德大学)

Laurens Sluijterman, Eric Cator, Tom Heskes

AI总结:

本文研究MVE网络的最优训练,指出使用预热期能解决收敛困难,并提出对均值和方差分别正则化的新方法,在多个数据集上取得显著改进。

摘要:

本文专注于均值方差估计网络(MVE network)(Nix and Weigend, 1994)的最优实现。这种网络通常作为回归环境中不确定性估计方法的构建模块,例如Concrete dropout(Gal et al., 2017)和Deep Ensembles(Lakshminarayanan et al., 2017)。具体而言,MVE network假设数据由具有均值函数和方差函数的正态分布生成。MVE network输出均值和方差估计,并通过最小化负对数似然来优化网络参数。在我们的论文中,我们提出了两个重要见解。首先,近期工作中报告的收敛困难可以通过遵循原作者简单但经常被忽视的建议来相对容易地避免,即应使用预热期。在此期间,仅以固定方差优化均值。我们通过实验证明了这一步骤的有效性,强调它应成为标准做法。作为旁注,我们研究了在预热之后,是固定均值优化方差更有利,还是同时优化两者更有利。在这里,我们没有观察到显著差异。其次,我们引入了MVE network的一项新颖改进:对均值和方差估计分别进行正则化。我们在玩具示例和多个基准UCI回归数据集上证明,遵循原始建议和新颖的独立正则化可以带来显著改进。

英文摘要:

This paper focusses on the optimal implementation of a Mean Variance Estimation network (MVE network) (Nix and Weigend, 1994). This type of network is often used as a building block for uncertainty estimation methods in a regression setting, for instance Concrete dropout (Gal et al., 2017) and Deep Ensembles (Lakshminarayanan et al., 2017). Specifically, an MVE network assumes that the data is produced from a normal distribution with a mean function and variance function. The MVE network outputs a mean and variance estimate and optimizes the network parameters by minimizing the negative loglikelihood. In our paper, we present two significant insights. Firstly, the convergence difficulties reported in recent work can be relatively easily prevented by following the simple yet often overlooked recommendation from the original authors that a warm-up period should be used. During this period, only the mean is optimized with a fixed variance. We demonstrate the effectiveness of this step through experimentation, highlighting that it should be standard practice. As a sidenote, we examine whether, after the warm-up, it is beneficial to fix the mean while optimizing the variance or to optimize both simultaneously. Here, we do not observe a substantial difference. Secondly, we introduce a novel improvement of the MVE network: separate regularization of the mean and the variance estimate. We demonstrate, both on toy examples and on a number of benchmark UCI regression data sets, that following the original recommendations and the novel separate regularization can lead to significant improvements.

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

↑