arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.04495stat.MLcs.LGstat.CO
收藏

基于双精度数据的神经网络迁移学习

On transfer learning of neural networks using bi-fidelity data for uncertainty propagation

University of Colorado, Boulder(科罗拉多大学博尔德分校);University of California, Riverside(加利福尼亚大学河滨分校);National Renewable Energy Laboratory(国家可再生能源实验室)

Subhayan De, Jolene Britton, Matthew Reynolds, Ryan Skinner, Kenneth Jansen, Alireza Doostan

摘要:

由于它们的高度表达性,神经网络最近已经被用作用于将工程系统的输入映射到感兴趣的输出的代理模型。一旦被训练,神经网络在计算上是廉价的,以评估和消除在不确定性量化应用中对计算上昂贵的模型的重复评估的需要。然而,考虑到神经网络,特别是深度神经网络的高度参数化构造,精确训练通常需要大量的仿真数据,而这些仿真数据在计算昂贵的系统的情况下可能是不可用的。在本文中,为了缓解不确定性传播的这个问题,我们探索了迁移学习技术的应用,使用从高保真和低保真模型生成的训练数据。在训练过程中,我们探索了两种耦合这两个数据集的策略,即标准迁移学习和双保真度加权学习。在前一种方法中,基于低保真度数据训练将输入映射到感兴趣的输出的神经网络模型。然后,高保真度数据被用于调整低保真度网络的上层的参数,或者训练更简单的神经网络以将低保真度网络的输出映射到高保真度模型的输出。在后一种方法中,整个低保真网络参数使用通过用小型高保真数据集训练的高斯过程模型生成的数据来更新。参数更新是通过随机梯度下降的一个变种,学习率由高斯过程模型。使用三个数值示例,我们说明了这些双保真度迁移学习方法的实用性,其中我们专注于通过迁移学习在标准训练方法上实现的准确性提高。

英文摘要:

Due to their high degree of expressiveness, neural networks have recently been used as surrogate models for mapping inputs of an engineering system to outputs of interest. Once trained, neural networks are computationally inexpensive to evaluate and remove the need for repeated evaluations of computationally expensive models in uncertainty quantification applications. However, given the highly parameterized construction of neural networks, especially deep neural networks, accurate training often requires large amounts of simulation data that may not be available in the case of computationally expensive systems. In this paper, to alleviate this issue for uncertainty propagation, we explore the application of transfer learning techniques using training data generated from both high- and low-fidelity models. We explore two strategies for coupling these two datasets during the training procedure, namely, the standard transfer learning and the bi-fidelity weighted learning. In the former approach, a neural network model mapping the inputs to the outputs of interest is trained based on the low-fidelity data. The high-fidelity data is then used to adapt the parameters of the upper layer(s) of the low-fidelity network, or train a simpler neural network to map the output of the low-fidelity network to that of the high-fidelity model. In the latter approach, the entire low-fidelity network parameters are updated using data generated via a Gaussian process model trained with a small high-fidelity dataset. The parameter updates are performed via a variant of stochastic gradient descent with learning rates given by the Gaussian process model. Using three numerical examples, we illustrate the utility of these bi-fidelity transfer learning methods where we focus on accuracy improvement achieved by transfer learning over standard training approaches.

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

↑