arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.02152cs.LGphysics.comp-ph
收藏

L-HYDRA:多头物理信息神经网络

L-HYDRA: Multi-Head Physics-Informed Neural Networks

Brown University(布朗大学)

Zongren Zou, George Em Karniadakis

AI总结:

L-HYDRA通过多头物理信息神经网络实现多任务学习、生成建模和少量样本学习,利用共享主体和分布进行协同学习,提升科学机器学习中的性能。

摘要:

我们引入多头神经网络(MH-NNs)到物理信息机器学习中,这是一种具有所有非线性隐藏层作为主体和多个线性输出层作为多头的神经网络。因此,我们构建了多头物理信息神经网络(MH-PINNs)作为一种强大的工具,用于多任务学习(MTL)、生成建模和少量样本学习,以解决科学机器学习(SciML)中的各种问题。MH-PINNs通过共享的主体作为基函数以及共享的分布作为头部来连接多个功能/任务。前者是通过使用MH-PINNs解决多个任务,每个头部独立对应每个任务来实现的,而后者则是通过使用归一化流(NFs)进行密度估计和生成建模来实现的。为此,我们的方法是一种两阶段方法,且两个阶段都可以使用标准深度学习工具进行处理,从而在实践中易于实现。MH-PINNs可以用于各种目的,例如近似随机过程、协同解决多个任务、为下游少量样本学习任务(如元学习和迁移学习)提供信息性先验知识、学习代表性基函数以及不确定性量化。我们在五个基准测试中展示了MH-PINNs的有效性,并研究了回归分析中的协同学习可能性。我们命名开源代码为“Lernaean Hydra”(L-HYDRA),因为这种神话生物拥有许多头来执行重要的多项任务,正如所提出的方法一样。

英文摘要:

We introduce multi-head neural networks (MH-NNs) to physics-informed machine learning, which is a type of neural networks (NNs) with all nonlinear hidden layers as the body and multiple linear output layers as multi-head. Hence, we construct multi-head physics-informed neural networks (MH-PINNs) as a potent tool for multi-task learning (MTL), generative modeling, and few-shot learning for diverse problems in scientific machine learning (SciML). MH-PINNs connect multiple functions/tasks via a shared body as the basis functions as well as a shared distribution for the head. The former is accomplished by solving multiple tasks with MH-PINNs with each head independently corresponding to each task, while the latter by employing normalizing flows (NFs) for density estimate and generative modeling. To this end, our method is a two-stage method, and both stages can be tackled with standard deep learning tools of NNs, enabling easy implementation in practice. MH-PINNs can be used for various purposes, such as approximating stochastic processes, solving multiple tasks synergistically, providing informative prior knowledge for downstream few-shot learning tasks such as meta-learning and transfer learning, learning representative basis functions, and uncertainty quantification. We demonstrate the effectiveness of MH-PINNs in five benchmarks, investigating also the possibility of synergistic learning in regression analysis. We name the open-source code "Lernaean Hydra" (L-HYDRA), since this mythical creature possessed many heads for performing important multiple tasks, as in the proposed method.

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

↑