中文

PINNACLE:基于 PINN 自适应点选取方法

机器学习 2024-04-15 v1 人工智能 计算物理 数据分析、统计与概率 机器学习

摘要

物理信息神经网络 (PINN) 将偏微分方程作为软约束进行训练,包含多种训练点类型:用于强制执行每个偏微分方程及初始/边界条件的不同类型的 collocation 点,以及通常通过实验或仿真获取成本较高的实验点。使用包含不同训练点类型的复合损失函数训练 PINN 具有挑战性,因为这通常需要选择大量不同类型的点,每种点的训练动力学都不同。与过去关注 collocation 点或实验点选择的工作不同,本文引入 PINN Adaptive ColLocation and Experimental points selection (PINNACLE),首个能够联合优化所有训练点类型选择的算法,并在训练过程中自动调整 collocation 点类型的比例。PINNACLE 基于对 PINN 训练动力学的神经切向核 (NTK) 分析,考虑了训练点类型之间的相互作用——此前未被考虑。我们理论上表明,PINNACLE 所使用的准则与 PINN 概化误差相关,并在实证实验中演示 PINNACLE 能为前向、逆向和迁移学习问题超越现有点选择方法。

关键词

引用

@article{arxiv.2404.07662,
  title  = {PINNACLE: PINN Adaptive ColLocation and Experimental points selection},
  author = {Gregory Kang Ruey Lau and Apivich Hemachandra and See-Kiong Ng and Bryan Kian Hsiang Low},
  journal= {arXiv preprint arXiv:2404.07662},
  year   = {2024}
}

备注

Accepted to 12th International Conference on Learning Representations (ICLR 2024), 36 pages